Skip to content

engine.templates

from engine.templates import *

engine.templates

SIGNATURE_PROFILES = [{'name': '{{ your_name }}', 'title': 'Independent Researcher', 'extra': ''}, {'name': '{{ your_name }}', 'title': 'Independent Researcher', 'extra': '\nhttps://scholar.google.com/citations?user=XXXXXXXX'}] module-attribute

SUBJECT_TEMPLATES = ['arXiv Endorsement Request: {{ your_paper_title }}', 'Endorsement Request for arXiv Submission: {{ your_paper_title }}', 'Could You Endorse My arXiv Paper? ({{ your_paper_title }})', 'Request for arXiv Endorsement — {{ your_paper_title }}', 'Seeking Endorsement: {{ your_paper_title }} (arXiv {{ arxiv_category }})'] module-attribute

TemplateEngine

Source code in engine/templates.py 
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
class TemplateEngine:
    def __init__(
        self,
        txt_paths: list[str] | None = None,
        html_paths: list[str] | None = None,
        auto_reload: bool = True,
    ) -> None:
        self.txt_paths = txt_paths or ["template.txt"]
        self.html_paths = html_paths or ["template.html"]
        self.auto_reload = auto_reload
        self._txt_cache: list[str] = []
        self._html_cache: list[str] = []
        self._txt_mtimes: list[float] = []
        self._html_mtimes: list[float] = []
        self._load_templates()

    def _load_templates(self) -> None:
        self._txt_cache.clear()
        self._html_cache.clear()
        self._txt_mtimes.clear()
        self._html_mtimes.clear()

        for path in self.txt_paths:
            if os.path.exists(path):
                with open(path, encoding="utf-8") as f:
                    self._txt_cache.append(f.read())
                self._txt_mtimes.append(os.path.getmtime(path))
            else:
                AppLogger.warn(f"Text template not found: {path}")

        for path in self.html_paths:
            if os.path.exists(path):
                with open(path, encoding="utf-8") as f:
                    self._html_cache.append(f.read())
                self._html_mtimes.append(os.path.getmtime(path))
            else:
                AppLogger.warn(f"HTML template not found: {path}")

        if not self._txt_cache:
            self._txt_cache.append(
                "Dear Dr./Mr./Ms. {{ last_name }},\n\n"
                "Regarding your paper: {{ paper_title }}..."
            )
        if not self._html_cache:
            self._html_cache.append(
                "<p>Dear Dr./Mr./Ms. {{ last_name }},</p>"
                "<p>Regarding your paper: <em>{{ paper_title }}</em>...</p>"
            )

    def _check_reload(self) -> None:
        if not self.auto_reload:
            return
        needs_reload = False
        for i, path in enumerate(self.txt_paths):
            if os.path.exists(path) and i < len(self._txt_mtimes):
                if os.path.getmtime(path) != self._txt_mtimes[i]:
                    needs_reload = True
                    break
        if not needs_reload:
            for i, path in enumerate(self.html_paths):
                if os.path.exists(path) and i < len(self._html_mtimes):
                    if os.path.getmtime(path) != self._html_mtimes[i]:
                        needs_reload = True
                        break
        if needs_reload:
            AppLogger.info("Templates changed on disk — reloading")
            self._load_templates()

    def _interpolate(self, template: str, context: dict[str, Any]) -> str:
        result = template
        for key, val in context.items():
            result = result.replace("{{ " + key + " }}", str(val))
            result = result.replace("{{" + key + "}}", str(val))
            result = result.replace("{ " + key + " }", str(val))
            result = result.replace("{" + key + "}", str(val))
        return result

    def render_subject(self, context: dict[str, Any]) -> str:
        self._check_reload()
        template = random.choice(SUBJECT_TEMPLATES)
        return self._interpolate(template, context)

    def generate_plain_text(self, html_body: str) -> str:
        return strip_html(html_body)

    def get_cache_stats(self) -> dict[str, Any]:
        self._check_reload()
        return {
            "txt_templates": len(self._txt_cache),
            "html_templates": len(self._html_cache),
            "subject_templates": len(SUBJECT_TEMPLATES),
            "signatures": len(SIGNATURE_PROFILES),
        }

    def render_text(self, context: dict[str, Any], variant: int | None = None) -> str:
        self._check_reload()
        if variant is not None and 0 <= variant < len(self._txt_cache):
            idx = variant
        else:
            idx = random.randrange(len(self._txt_cache))
        return self._interpolate(self._txt_cache[idx], context)

    def render_file(self, path: str, context: dict[str, Any]) -> str:
        """Render an arbitrary template file (e.g. follow-up template)."""
        if not os.path.exists(path):
            raise TemplateRenderError(f"Template file not found: {path}")
        with open(path, encoding="utf-8") as f:
            return self._interpolate(f.read(), context)

    def render_html(self, context: dict[str, Any], variant: int | None = None) -> str:
        self._check_reload()
        if variant is not None and 0 <= variant < len(self._html_cache):
            idx = variant
        else:
            idx = random.randrange(len(self._html_cache))
        return self._interpolate(self._html_cache[idx], context)

    def render_signature(self, context: dict[str, Any], variant: int | None = None) -> str:
        if variant is not None and 0 <= variant < len(SIGNATURE_PROFILES):
            idx = variant
        else:
            idx = random.randrange(len(SIGNATURE_PROFILES))
        profile = SIGNATURE_PROFILES[idx]
        sig = self._interpolate(profile["name"], context)
        sig += "\n" + self._interpolate(profile["title"], context)
        if profile["extra"]:
            sig += self._interpolate(profile["extra"], context)
        return sig

    def get_required_vars(self) -> list[str]:
        required: list[str] = []
        for t in self._txt_cache + self._html_cache:
            required.extend(re.findall(r"\{\{\s*(\w+)\s*\}\}", t))
        for s in SUBJECT_TEMPLATES:
            required.extend(re.findall(r"\{\{\s*(\w+)\s*\}\}", s))
        for p in SIGNATURE_PROFILES:
            for v in p.values():
                required.extend(re.findall(r"\{\{\s*(\w+)\s*\}\}", str(v)))
        return sorted(set(required))

    def validate_context(self, context: dict[str, Any]) -> list[str]:
        missing: list[str] = []
        for var in self.get_required_vars():
            if var not in context or context[var] is None:
                missing.append(var)
        return missing

    def render_all(
        self, context: dict[str, Any]
    ) -> dict[str, str]:
        return {
            "subject": self.render_subject(context),
            "text_body": self.render_text(context),
            "html_body": self.render_html(context),
            "signature": self.render_signature(context),
        }

auto_reload = auto_reload instance-attribute

html_paths = html_paths or ['template.html'] instance-attribute

txt_paths = txt_paths or ['template.txt'] instance-attribute

__init__(txt_paths=None, html_paths=None, auto_reload=True)

Source code in engine/templates.py 
49
50
51
52
53
54
55
56
57
58
59
60
61
62
def __init__(
    self,
    txt_paths: list[str] | None = None,
    html_paths: list[str] | None = None,
    auto_reload: bool = True,
) -> None:
    self.txt_paths = txt_paths or ["template.txt"]
    self.html_paths = html_paths or ["template.html"]
    self.auto_reload = auto_reload
    self._txt_cache: list[str] = []
    self._html_cache: list[str] = []
    self._txt_mtimes: list[float] = []
    self._html_mtimes: list[float] = []
    self._load_templates()

generate_plain_text(html_body)

Source code in engine/templates.py 
130
131
def generate_plain_text(self, html_body: str) -> str:
    return strip_html(html_body)

get_cache_stats()

Source code in engine/templates.py 
133
134
135
136
137
138
139
140
def get_cache_stats(self) -> dict[str, Any]:
    self._check_reload()
    return {
        "txt_templates": len(self._txt_cache),
        "html_templates": len(self._html_cache),
        "subject_templates": len(SUBJECT_TEMPLATES),
        "signatures": len(SIGNATURE_PROFILES),
    }

get_required_vars()

Source code in engine/templates.py 
177
178
179
180
181
182
183
184
185
186
def get_required_vars(self) -> list[str]:
    required: list[str] = []
    for t in self._txt_cache + self._html_cache:
        required.extend(re.findall(r"\{\{\s*(\w+)\s*\}\}", t))
    for s in SUBJECT_TEMPLATES:
        required.extend(re.findall(r"\{\{\s*(\w+)\s*\}\}", s))
    for p in SIGNATURE_PROFILES:
        for v in p.values():
            required.extend(re.findall(r"\{\{\s*(\w+)\s*\}\}", str(v)))
    return sorted(set(required))

render_all(context)

Source code in engine/templates.py 
195
196
197
198
199
200
201
202
203
def render_all(
    self, context: dict[str, Any]
) -> dict[str, str]:
    return {
        "subject": self.render_subject(context),
        "text_body": self.render_text(context),
        "html_body": self.render_html(context),
        "signature": self.render_signature(context),
    }

render_file(path, context)

Render an arbitrary template file (e.g. follow-up template).

Source code in engine/templates.py 
150
151
152
153
154
155
def render_file(self, path: str, context: dict[str, Any]) -> str:
    """Render an arbitrary template file (e.g. follow-up template)."""
    if not os.path.exists(path):
        raise TemplateRenderError(f"Template file not found: {path}")
    with open(path, encoding="utf-8") as f:
        return self._interpolate(f.read(), context)

render_html(context, variant=None)

Source code in engine/templates.py 
157
158
159
160
161
162
163
def render_html(self, context: dict[str, Any], variant: int | None = None) -> str:
    self._check_reload()
    if variant is not None and 0 <= variant < len(self._html_cache):
        idx = variant
    else:
        idx = random.randrange(len(self._html_cache))
    return self._interpolate(self._html_cache[idx], context)

render_signature(context, variant=None)

Source code in engine/templates.py 
165
166
167
168
169
170
171
172
173
174
175
def render_signature(self, context: dict[str, Any], variant: int | None = None) -> str:
    if variant is not None and 0 <= variant < len(SIGNATURE_PROFILES):
        idx = variant
    else:
        idx = random.randrange(len(SIGNATURE_PROFILES))
    profile = SIGNATURE_PROFILES[idx]
    sig = self._interpolate(profile["name"], context)
    sig += "\n" + self._interpolate(profile["title"], context)
    if profile["extra"]:
        sig += self._interpolate(profile["extra"], context)
    return sig

render_subject(context)

Source code in engine/templates.py 
125
126
127
128
def render_subject(self, context: dict[str, Any]) -> str:
    self._check_reload()
    template = random.choice(SUBJECT_TEMPLATES)
    return self._interpolate(template, context)

render_text(context, variant=None)

Source code in engine/templates.py 
142
143
144
145
146
147
148
def render_text(self, context: dict[str, Any], variant: int | None = None) -> str:
    self._check_reload()
    if variant is not None and 0 <= variant < len(self._txt_cache):
        idx = variant
    else:
        idx = random.randrange(len(self._txt_cache))
    return self._interpolate(self._txt_cache[idx], context)

validate_context(context)

Source code in engine/templates.py 
188
189
190
191
192
193
def validate_context(self, context: dict[str, Any]) -> list[str]:
    missing: list[str] = []
    for var in self.get_required_vars():
        if var not in context or context[var] is None:
            missing.append(var)
    return missing

strip_html(html)

Source code in engine/templates.py 
17
18
19
20
21
22
23
def strip_html(html: str) -> str:
    def replace_entity(m: re.Match) -> str:
        return _HTML_ENTITY_MAP.get(m.group(0), " ")
    text = _HTML_ENTITY_RE.sub(replace_entity, html)
    text = _HTML_TAG_RE.sub("", text)
    text = re.sub(r"\n\s*\n", "\n\n", text)
    return text.strip()