GH-1: Extract Jinja loader function

acteng · Sep 18, 2023 · ffb8fe8 · ffb8fe8
1 parent e8d6c8a
commit ffb8fe8
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/schemes/__init__.py b/schemes/__init__.py
@@ -1,20 +1,23 @@
 import os
 from flask import Flask, render_template
-from jinja2 import ChoiceLoader, FileSystemLoader, PackageLoader, PrefixLoader
+from jinja2 import BaseLoader, ChoiceLoader, FileSystemLoader, PackageLoader, PrefixLoader
 
 
 def create_app() -> Flask:
  app = Flask(__name__, static_url_path="/")
-
- app.jinja_loader = ChoiceLoader( # type: ignore
- [
- FileSystemLoader(os.path.join(app.root_path, str(app.template_folder))),
- PrefixLoader({"govuk_frontend_jinja": PackageLoader("govuk_frontend_jinja")})
- ]
- )
+ app.jinja_loader = create_jinja_loader(app) # type: ignore
 
  @app.route("/")
  def index() -> str:
  return render_template("index.html")
 
  return app
+
+
+def create_jinja_loader(app: Flask) -> BaseLoader:
+ return ChoiceLoader(
+ [
+ FileSystemLoader(os.path.join(app.root_path, str(app.template_folder))),
+ PrefixLoader({"govuk_frontend_jinja": PackageLoader("govuk_frontend_jinja")})
+ ]
+ )