hgweb: open mapfile using templater.open_template()
authorMartin von Zweigbergk <martinvonz@google.com>
Wed, 05 Aug 2020 14:03:14 -0700
changeset 45316 1b983985edd9
parent 45315 2901133ec982
child 45317 c37ab438ff31
hgweb: open mapfile using templater.open_template() This will help us read templates from resources in frozen binaries. Differential Revision: https://phab.mercurial-scm.org/D8899
mercurial/hgweb/hgweb_mod.py
mercurial/templater.py
--- a/mercurial/hgweb/hgweb_mod.py	Wed Aug 05 13:58:30 2020 -0700
+++ b/mercurial/hgweb/hgweb_mod.py	Wed Aug 05 14:03:14 2020 -0700
@@ -82,8 +82,8 @@
             locations = (os.path.join(style, b'map'), b'map-' + style, b'map')
 
             for location in locations:
-                mapfile = os.path.join(path, location)
-                if os.path.isfile(mapfile):
+                mapfile, fp = templater.open_template(location, path)
+                if mapfile:
                     return style, mapfile
 
     raise RuntimeError(b"No hgweb templates found in %r" % path)
--- a/mercurial/templater.py	Wed Aug 05 13:58:30 2020 -0700
+++ b/mercurial/templater.py	Wed Aug 05 14:03:14 2020 -0700
@@ -1084,14 +1084,15 @@
     return path if os.path.isdir(path) else None
 
 
-def open_template(name):
+def open_template(name, templatepath=None):
     '''returns a file-like object for the given template, and its full path
 
     If the name is a relative path and we're in a frozen binary, the template
     will be read from the mercurial.templates package instead. The returned path
     will then be the relative path.
     '''
-    templatepath = templatedir()
+    if templatepath is None:
+        templatepath = templatedir()
     if templatepath is not None or os.path.isabs(name):
         f = os.path.join(templatepath, name)
         try: