ycdl/frontends/ycdl_flask/backend/common.py

77 lines
2.2 KiB
Python
Raw Normal View History

2016-11-29 04:16:16 +00:00
'''
Do not execute this file directly.
Use ycdl_flask_dev.py or ycdl_flask_prod.py.
2016-11-29 04:16:16 +00:00
'''
2020-03-28 23:49:33 +00:00
import flask; from flask import request
import threading
2020-03-28 23:49:33 +00:00
import time
2020-03-29 00:05:43 +00:00
from voussoirkit import flasktools
2020-03-29 00:05:43 +00:00
from voussoirkit import pathclass
2016-11-29 04:16:16 +00:00
import ycdl
from . import jinja_filters
2020-09-22 09:50:24 +00:00
# Flask init #######################################################################################
2017-10-09 04:39:07 +00:00
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')
site = flask.Flask(
__name__,
template_folder=TEMPLATE_DIR.absolute_path,
static_folder=STATIC_DIR.absolute_path,
)
2016-11-29 04:16:16 +00:00
site.config.update(
SEND_FILE_MAX_AGE_DEFAULT=180,
TEMPLATES_AUTO_RELOAD=True,
)
site.jinja_env.add_extension('jinja2.ext.do')
2020-09-22 09:50:24 +00:00
site.jinja_env.trim_blocks = True
site.jinja_env.lstrip_blocks = True
jinja_filters.register_all(site)
2016-11-29 04:16:16 +00:00
site.debug = True
2020-09-30 22:15:30 +00:00
site.localhost_only = False
2016-11-29 04:16:16 +00:00
# Request decorators ###############################################################################
2020-09-22 09:50:24 +00:00
2020-09-30 22:15:30 +00:00
@site.before_request
def before_request():
request.is_localhost = (request.remote_addr == '127.0.0.1')
if site.localhost_only and not request.is_localhost:
2020-09-30 22:15:30 +00:00
flask.abort(403)
2020-08-28 23:43:02 +00:00
@site.after_request
def after_request(response):
response = flasktools.gzip_response(request, response)
2020-08-28 23:43:02 +00:00
return response
2016-11-29 04:16:16 +00:00
####################################################################################################
2021-04-04 18:16:44 +00:00
# These functions will be called by the launcher, flask_dev, flask_prod.
2020-09-22 09:50:24 +00:00
def init_ycdldb(*args, **kwargs):
global ycdldb
ycdldb = ycdl.ycdldb.YCDLDB(*args, **kwargs)
2016-11-29 04:16:16 +00:00
def refresher_thread(rate):
while True:
time.sleep(rate)
print('Starting refresh job.')
thread_kwargs = {'force': False, 'skip_failures': True}
2020-11-10 04:01:47 +00:00
refresh_job = threading.Thread(
target=ycdldb.refresh_all_channels,
kwargs=thread_kwargs,
daemon=True,
)
refresh_job.start()
def start_refresher_thread(rate):
print(f'Starting refresher thread, once per {rate} seconds.')
refresher = threading.Thread(target=refresher_thread, args=[rate], daemon=True)
refresher.start()