Support instance templates

2 files changed, 8 insertions, 0 deletions

diff --git a/README.rst b/README.rst
index ecaf829..8b6cab5 100644
--- a/README.rst
+++ b/README.rst
@@ -10,6 +10,9 @@ Configuration
 To change settings, modify ``instance/config.py``. For more information on
 instance configuration, see `the Flask documentation <https://flask.palletsprojects.com/en/2.0.x/config/#instance-folders>`_.
 
+To customize the home and error pages, simply create a ``templates`` directory
+in your instance directory and copy any templates you want to modify there.
+
 If you are running nginx, you should use the ``X-Accel-Redirect`` header.
 To make it work, include this in your nginx config’s ``server`` block::
 
diff --git a/fhost.py b/fhost.py
index f939245..8644ade 100755
--- a/fhost.py
+++ b/fhost.py
@@ -23,6 +23,7 @@ from flask import Flask, abort, make_response, redirect, request, send_from_dire
 from flask_sqlalchemy import SQLAlchemy
 from flask_migrate import Migrate
 from jinja2.exceptions import *
+from jinja2 import ChoiceLoader, FileSystemLoader
 from hashlib import sha256
 from magic import Magic
 from mimetypes import guess_extension
@@ -68,6 +69,10 @@ app.config.update(
 
 if not app.config["TESTING"]:
     app.config.from_pyfile("config.py")
+    app.jinja_loader = ChoiceLoader([
+        FileSystemLoader(str(Path(app.instance_path) / "templates")),
+        app.jinja_loader
+    ])
 
     if app.config["DEBUG"]:
         app.config["FHOST_USE_X_ACCEL_REDIRECT"] = False