mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-26 03:44:58 +08:00
64 lines
1.3 KiB
Python
64 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
import traceback
|
|
import asyncio
|
|
import os
|
|
|
|
from . import app
|
|
from . import stage
|
|
from ..utils import constants, importutil
|
|
|
|
# Import startup stage implementation to register
|
|
from . import stages
|
|
|
|
importutil.import_modules_in_pkg(stages)
|
|
|
|
|
|
stage_order = [
|
|
'LoadConfigStage',
|
|
'MigrationStage',
|
|
'GenKeysStage',
|
|
'SetupLoggerStage',
|
|
'BuildAppStage',
|
|
'ShowNotesStage',
|
|
]
|
|
|
|
|
|
async def make_app(loop: asyncio.AbstractEventLoop) -> app.Application:
|
|
# Determine if it is debug mode
|
|
if 'DEBUG' in os.environ and os.environ['DEBUG'] in ['true', '1']:
|
|
constants.debug_mode = True
|
|
|
|
ap = app.Application()
|
|
|
|
ap.event_loop = loop
|
|
|
|
# Execute startup stage
|
|
for stage_name in stage_order:
|
|
stage_cls = stage.preregistered_stages[stage_name]
|
|
stage_inst = stage_cls()
|
|
|
|
await stage_inst.run(ap)
|
|
|
|
await ap.initialize()
|
|
|
|
return ap
|
|
|
|
|
|
async def main(loop: asyncio.AbstractEventLoop):
|
|
try:
|
|
# Hang system signal processing
|
|
import signal
|
|
|
|
def signal_handler(sig, frame):
|
|
app_inst.dispose()
|
|
print('[Signal] Program exit.')
|
|
os._exit(0)
|
|
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
|
|
app_inst = await make_app(loop)
|
|
await app_inst.run()
|
|
except Exception:
|
|
traceback.print_exc()
|