''' Do not execute this file directly. Use etiquette_flask_launch.py to start the server with gevent. ''' import flask; from flask import request import gzip import io import mimetypes import traceback from voussoirkit import bytestring from voussoirkit import pathclass import etiquette from . import caching from . import decorators from . import jinja_filters from . import jsonify from . import sessions # Flask init ####################################################################################### root_dir = pathclass.Path(__file__).parent.parent TEMPLATE_DIR = root_dir.with_child('templates') STATIC_DIR = root_dir.with_child('static') FAVICON_PATH = STATIC_DIR.with_child('favicon.png') BROWSER_CACHE_DURATION = 180 site = flask.Flask( __name__, template_folder=TEMPLATE_DIR.absolute_path, static_folder=STATIC_DIR.absolute_path, ) site.config.update( SEND_FILE_MAX_AGE_DEFAULT=BROWSER_CACHE_DURATION, TEMPLATES_AUTO_RELOAD=True, ) site.jinja_env.add_extension('jinja2.ext.do') site.jinja_env.trim_blocks = True site.jinja_env.lstrip_blocks = True jinja_filters.register_all(site) site.debug = True site.localhost_only = False session_manager = sessions.SessionManager(maxlen=10000) file_cache_manager = caching.FileCacheManager( maxlen=10000, max_filesize=5 * bytestring.MIBIBYTE, max_age=BROWSER_CACHE_DURATION, ) # Response wrappers ################################################################################ # Flask provides decorators for before_request and after_request, but not for # wrapping the whole request. The decorators I am using need to wrap the whole # request, either to catch exceptions (which don't get passed through # after_request) or to maintain some state before running the function and # adding it to the response after. # Instead of copy-pasting my decorators onto every single function and # forgetting to keep up with them in the future, let's just hijack the # decorator I know every endpoint will have: site.route. _original_route = site.route def decorate_and_route(*route_args, **route_kwargs): def wrapper(endpoint): if not hasattr(endpoint, '_fully_decorated'): endpoint = decorators.catch_etiquette_exception(endpoint) endpoint = session_manager.give_token(endpoint) endpoint = _original_route(*route_args, **route_kwargs)(endpoint) endpoint._fully_decorated = True return endpoint return wrapper site.route = decorate_and_route @site.before_request def before_request(): ip = request.remote_addr if site.localhost_only and ip != '127.0.0.1': flask.abort(403) gzip_minimum_size = 500 gzip_maximum_size = 5 * 2**20 gzip_level = 3 @site.after_request def after_request(response): ''' Thank you close.io. https://github.com/closeio/Flask-gzip ''' accept_encoding = request.headers.get('Accept-Encoding', '') bail = False bail = bail or response.status_code < 200 bail = bail or response.status_code >= 300 bail = bail or response.direct_passthrough bail = bail or int(response.headers.get('Content-Length', 0)) > gzip_maximum_size bail = bail or len(response.get_data()) < gzip_minimum_size bail = bail or 'gzip' not in accept_encoding.lower() bail = bail or 'Content-Encoding' in response.headers if bail: return response gzip_buffer = io.BytesIO() gzip_file = gzip.GzipFile(mode='wb', compresslevel=gzip_level, fileobj=gzip_buffer) gzip_file.write(response.get_data()) gzip_file.close() response.set_data(gzip_buffer.getvalue()) response.headers['Content-Encoding'] = 'gzip' response.headers['Content-Length'] = len(response.get_data()) return response # P functions ###################################################################################### def P_wrapper(function): def P_wrapped(thingid, response_type): if response_type not in {'html', 'json'}: raise TypeError(f'response_type should be html or json, not {response_type}.') try: return function(thingid) except etiquette.exceptions.EtiquetteException as exc: if isinstance(exc, etiquette.exceptions.NoSuch): status = 404 else: status = 400 if response_type == 'html': flask.abort(status, exc.error_message) else: response = exc.jsonify() response = jsonify.make_json_response(response, status=status) flask.abort(response) except Exception as exc: traceback.print_exc() if response_type == 'html': flask.abort(500) else: flask.abort(jsonify.make_json_response({}, status=500)) return P_wrapped @P_wrapper def P_album(album_id): return P.get_album(album_id) @P_wrapper def P_albums(album_ids): return P.get_albums_by_id(album_ids) @P_wrapper def P_bookmark(bookmark_id): return P.get_bookmark(bookmark_id) @P_wrapper def P_photo(photo_id): return P.get_photo(photo_id) @P_wrapper def P_photos(photo_ids): return P.get_photos_by_id(photo_ids) @P_wrapper def P_tag(tagname): return P.get_tag(name=tagname) @P_wrapper def P_tag_id(tag_id): return P.get_tag(id=tag_id) @P_wrapper def P_user(username): return P.get_user(username=username) @P_wrapper def P_user_id(user_id): return P.get_user(id=user_id) # Other functions ################################################################################## def back_url(): return request.args.get('goto') or request.referrer or '/' def render_template(request, template_name, **kwargs): session = session_manager.get(request) old_theme = request.cookies.get('etiquette_theme', None) new_theme = request.args.get('theme', None) theme = new_theme or old_theme or 'slate' response = flask.render_template( template_name, session=session, theme=theme, **kwargs, ) if not isinstance(response, sessions.RESPONSE_TYPES): response = flask.Response(response) if new_theme is None: pass elif new_theme == '': response.set_cookie('etiquette_theme', value='', expires=0) elif new_theme != old_theme: response.set_cookie('etiquette_theme', value=new_theme, expires=2147483647) return response def send_file(filepath, override_mimetype=None): ''' Range-enabled file sending. ''' filepath = pathclass.Path(filepath) if not filepath.is_file: flask.abort(404) file_size = filepath.size headers = file_cache_manager.matches(request=request, filepath=filepath) if headers: response = flask.Response(status=304, headers=headers) return response outgoing_headers = {} if override_mimetype is not None: mimetype = override_mimetype else: mimetype = mimetypes.guess_type(filepath.absolute_path)[0] if mimetype is not None: if 'text/' in mimetype: mimetype += '; charset=utf-8' outgoing_headers['Content-Type'] = mimetype if 'range' in request.headers: desired_range = request.headers['range'].lower() desired_range = desired_range.split('bytes=')[-1] int_helper = lambda x: int(x) if x.isdigit() else None if '-' in desired_range: (desired_min, desired_max) = desired_range.split('-') range_min = int_helper(desired_min) range_max = int_helper(desired_max) else: range_min = int_helper(desired_range) range_max = None if range_min is None: range_min = 0 if range_max is None: range_max = file_size # because ranges are 0-indexed range_max = min(range_max, file_size - 1) range_min = max(range_min, 0) range_header = 'bytes {min}-{max}/{outof}'.format( min=range_min, max=range_max, outof=file_size, ) outgoing_headers['Content-Range'] = range_header status = 206 else: range_max = file_size - 1 range_min = 0 status = 200 outgoing_headers['Accept-Ranges'] = 'bytes' outgoing_headers['Content-Length'] = (range_max - range_min) + 1 cache_file = file_cache_manager.get(filepath) if cache_file is not None: outgoing_headers.update(cache_file.get_headers()) if request.method == 'HEAD': outgoing_data = bytes() else: outgoing_data = etiquette.helpers.read_filebytes( filepath.absolute_path, range_min=range_min, range_max=range_max, chunk_size=P.config['file_read_chunk'], ) response = flask.Response( outgoing_data, status=status, headers=outgoing_headers, ) return response #################################################################################################### def init_photodb(*args, **kwargs): global P P = etiquette.photodb.PhotoDB(*args, **kwargs)