Plugins can now bundle custom templates, closes #224

Refs #14
This commit is contained in:
Simon Willison 2018-04-18 22:50:27 -07:00
commit b52171db1e
No known key found for this signature in database
GPG key ID: 17E2DEA2588B7F52
4 changed files with 41 additions and 11 deletions

View file

@ -1303,16 +1303,22 @@ class Datasette:
def app(self):
app = Sanic(__name__)
default_templates = str(app_root / 'datasette' / 'templates')
template_paths = []
if self.template_dir:
template_loader = ChoiceLoader([
FileSystemLoader([self.template_dir, default_templates]),
# Support {% extends "default:table.html" %}:
PrefixLoader({
'default': FileSystemLoader(default_templates),
}, delimiter=':')
])
else:
template_loader = FileSystemLoader(default_templates)
template_paths.append(self.template_dir)
template_paths.extend([
plugin['templates_path']
for plugin in get_plugins(pm)
if plugin['templates_path']
])
template_paths.append(default_templates)
template_loader = ChoiceLoader([
FileSystemLoader(template_paths),
# Support {% extends "default:table.html" %}:
PrefixLoader({
'default': FileSystemLoader(default_templates),
}, delimiter=':')
])
self.jinja_env = Environment(
loader=template_loader,
autoescape=True,
@ -1344,7 +1350,8 @@ class Datasette:
app.add_route(
JsonDataView.as_view(self, 'plugins.json', lambda: [{
'name': p['name'],
'static': p['static_path'] is not None
'static': p['static_path'] is not None,
'templates': p['templates_path'] is not None,
} for p in get_plugins(pm)]),
'/-/plugins<as_json:(\.json)?$>'
)

View file

@ -690,14 +690,18 @@ def get_plugins(pm):
plugins = []
for plugin in pm.get_plugins():
static_path = None
templates_path = None
try:
if pkg_resources.resource_isdir(plugin.__name__, 'static'):
static_path = pkg_resources.resource_filename(plugin.__name__, 'static')
if pkg_resources.resource_isdir(plugin.__name__, 'templates'):
templates_path = pkg_resources.resource_filename(plugin.__name__, 'templates')
except (KeyError, ImportError):
# Caused by --plugins_dir= plugins - KeyError/ImportError thrown in Py3.5
pass
plugins.append({
'name': plugin.__name__,
'static_path': static_path,
'templates_path': templates_path,
})
return plugins