mirror of
https://github.com/frappe/frappe.git
synced 2024-06-03 00:20:49 +00:00
81b37cb7d2
refactor: clean up code to py39+ supported syntax - f-strings instead of format - latest typing support instead of pre 3.9 TitleCase - remove UTF-8 declarations. - many more changes Powered by https://github.com/asottile/pyupgrade/ + manual cleanups
29 lines
811 B
Python
29 lines
811 B
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
|
|
import os
|
|
|
|
from werkzeug.exceptions import NotFound
|
|
from werkzeug.middleware.shared_data import SharedDataMiddleware
|
|
|
|
import frappe
|
|
from frappe.utils import cstr, get_site_name
|
|
|
|
|
|
class StaticDataMiddleware(SharedDataMiddleware):
|
|
def __call__(self, environ, start_response):
|
|
self.environ = environ
|
|
return super().__call__(environ, start_response)
|
|
|
|
def get_directory_loader(self, directory):
|
|
def loader(path):
|
|
site = get_site_name(frappe.app._site or self.environ.get("HTTP_HOST"))
|
|
path = os.path.join(directory, site, "public", "files", cstr(path))
|
|
if os.path.isfile(path):
|
|
return os.path.basename(path), self._opener(path)
|
|
else:
|
|
raise NotFound
|
|
# return None, None
|
|
|
|
return loader
|