107 lines
3 KiB
Python
107 lines
3 KiB
Python
'''
|
|
This file is the gevent launcher for local / development use.
|
|
|
|
Simply run it on the command line:
|
|
python ycdl_flask_dev.py [port]
|
|
'''
|
|
import gevent.monkey; gevent.monkey.patch_all()
|
|
|
|
import logging
|
|
handler = logging.StreamHandler()
|
|
log_format = '{levelname}:ycdl.{module}.{funcName}: {message}'
|
|
handler.setFormatter(logging.Formatter(log_format, style='{'))
|
|
logging.getLogger().addHandler(handler)
|
|
|
|
import argparse
|
|
import gevent.pywsgi
|
|
import os
|
|
import sys
|
|
|
|
from voussoirkit import pathclass
|
|
from voussoirkit import vlogging
|
|
|
|
import bot
|
|
import ycdl
|
|
import backend
|
|
|
|
site = backend.site
|
|
site.debug = True
|
|
|
|
####################################################################################################
|
|
|
|
site = backend.site
|
|
|
|
HTTPS_DIR = pathclass.Path(__file__).parent.with_child('https')
|
|
LOG_LEVEL = vlogging.NOTSET
|
|
|
|
def ycdl_flask_launch(
|
|
*,
|
|
create,
|
|
localhost_only,
|
|
port,
|
|
refresh_rate,
|
|
use_https,
|
|
):
|
|
if use_https is None:
|
|
use_https = port == 443
|
|
|
|
if use_https:
|
|
http = gevent.pywsgi.WSGIServer(
|
|
listener=('0.0.0.0', port),
|
|
application=site,
|
|
keyfile=HTTPS_DIR.with_child('ycdl.key').absolute_path,
|
|
certfile=HTTPS_DIR.with_child('ycdl.crt').absolute_path,
|
|
)
|
|
else:
|
|
http = gevent.pywsgi.WSGIServer(
|
|
listener=('0.0.0.0', port),
|
|
application=site,
|
|
)
|
|
|
|
if localhost_only:
|
|
site.localhost_only = True
|
|
|
|
youtube_core = ycdl.ytapi.Youtube(bot.get_youtube_key())
|
|
backend.common.init_ycdldb(youtube_core, create=create, log_level=LOG_LEVEL)
|
|
ycdl.ytrss.log.setLevel(LOG_LEVEL)
|
|
|
|
if refresh_rate is not None:
|
|
backend.common.start_refresher_thread(refresh_rate)
|
|
|
|
message = f'Starting server on port {port}, pid={os.getpid()}.'
|
|
if use_https:
|
|
message += ' (https)'
|
|
print(message)
|
|
|
|
try:
|
|
http.serve_forever()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
def ycdl_flask_launch_argparse(args):
|
|
return ycdl_flask_launch(
|
|
create=args.create,
|
|
localhost_only=args.localhost_only,
|
|
port=args.port,
|
|
refresh_rate=args.refresh_rate,
|
|
use_https=args.use_https,
|
|
)
|
|
|
|
def main(argv):
|
|
global LOG_LEVEL
|
|
(LOG_LEVEL, argv) = vlogging.get_level_by_argv(argv)
|
|
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
|
|
parser.add_argument('port', nargs='?', type=int, default=5000)
|
|
parser.add_argument('--dont_create', '--dont-create', '--no-create', dest='create', action='store_false', default=True)
|
|
parser.add_argument('--https', dest='use_https', action='store_true', default=None)
|
|
parser.add_argument('--localhost_only', '--localhost-only', dest='localhost_only', action='store_true')
|
|
parser.add_argument('--refresh_rate', '--refresh-rate', dest='refresh_rate', type=int, default=None)
|
|
parser.set_defaults(func=ycdl_flask_launch_argparse)
|
|
|
|
args = parser.parse_args(argv)
|
|
return args.func(args)
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main(sys.argv[1:]))
|