diff --git a/CTFd/__init__.py b/CTFd/__init__.py index 4faeda20..5bc5bc0c 100644 --- a/CTFd/__init__.py +++ b/CTFd/__init__.py @@ -1,5 +1,6 @@ import datetime import os +import re import sys import weakref from distutils.version import StrictVersion @@ -116,12 +117,37 @@ class ThemeLoader(FileSystemLoader): if template.startswith("admin/"): template = template[6:] # Strip out admin/ template = "/".join(["admin", "templates", template]) - return super(ThemeLoader, self).get_source(environment, template) + else: + # Load regular theme data + theme = str(utils.get_config("ctf_theme")) + template = "/".join([theme, "templates", template]) + source = super(ThemeLoader, self).get_source(environment, template) - # Load regular theme data - theme = str(utils.get_config("ctf_theme")) - template = "/".join([theme, "templates", template]) - return super(ThemeLoader, self).get_source(environment, template) + html = source[0].split("\n") + lines = [] + start_ignores = r"(