e407b78506
* Remove six for PY2 compatability since our dependencies are not, PY2 is legacy. * Removed usages of utils from future/past libraries since they are deprecated. This includes 'from __future__ ...' and 'from past...' statements. * Removed compatibility imports for PY2, switched from six imports to standard library imports. * Removed utils code blocks that handle operations depending on PY2/3 versions. * Removed 'from __future__ ...' lines from templates/code generators * Used PY3 syntaxes in place of PY2 compatible blocks. eg: metaclass
26 lines
838 B
Python
26 lines
838 B
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
import frappe
|
|
import os
|
|
from werkzeug.exceptions import NotFound
|
|
from werkzeug.middleware.shared_data import SharedDataMiddleware
|
|
from frappe.utils import get_site_name, cstr
|
|
|
|
|
|
class StaticDataMiddleware(SharedDataMiddleware):
|
|
def __call__(self, environ, start_response):
|
|
self.environ = environ
|
|
return super(StaticDataMiddleware, self).__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
|