52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
import os
|
|
import multiprocessing
|
|
import logging
|
|
|
|
from WebHost import app
|
|
from waitress import serve
|
|
|
|
from WebHost.models import db
|
|
|
|
DEBUG = False
|
|
port = 80
|
|
|
|
|
|
def autohost(config: dict):
|
|
from WebHost.models import *
|
|
import time
|
|
from datetime import timedelta, datetime
|
|
|
|
def keep_running():
|
|
db.bind(**config["pony"])
|
|
db.generate_mapping(check_tables=False)
|
|
while 1:
|
|
time.sleep(3)
|
|
with db_session:
|
|
select(
|
|
room for room in Room if (datetime.utcnow() - room.last_activity) < timedelta(hours=room.timeout))
|
|
|
|
import threading
|
|
threading.Thread(target=keep_running).start()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
multiprocessing.freeze_support()
|
|
multiprocessing.set_start_method('spawn')
|
|
logging.basicConfig(format='[%(asctime)s] %(message)s', level=logging.INFO)
|
|
|
|
configpath = "config.yaml"
|
|
|
|
if os.path.exists(configpath):
|
|
import yaml
|
|
with open(configpath) as c:
|
|
app.config.update(yaml.safe_load(c))
|
|
|
|
logging.info(f"Updated config from {configpath}")
|
|
db.bind(**app.config["PONY"])
|
|
db.generate_mapping(create_tables=True)
|
|
if DEBUG:
|
|
autohost(config)
|
|
app.run(debug=True, port=port)
|
|
else:
|
|
serve(app, port=port, threads=app.config["WAITRESS_THREADS"])
|