mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 19:37:36 +08:00
70 lines
1.4 KiB
Python
70 lines
1.4 KiB
Python
from __future__ import print_function
|
|
|
|
import traceback
|
|
import asyncio
|
|
import os
|
|
|
|
from . import app
|
|
from ..audit import identifier
|
|
from . import stage
|
|
from ..utils import constants
|
|
|
|
# 引入启动阶段实现以便注册
|
|
from .stages import load_config, setup_logger, build_app, migrate, show_notes
|
|
|
|
|
|
stage_order = [
|
|
"LoadConfigStage",
|
|
"MigrationStage",
|
|
"SetupLoggerStage",
|
|
"BuildAppStage",
|
|
"ShowNotesStage"
|
|
]
|
|
|
|
|
|
async def make_app(loop: asyncio.AbstractEventLoop) -> app.Application:
|
|
|
|
# 生成标识符
|
|
identifier.init()
|
|
|
|
# 确定是否为调试模式
|
|
if "DEBUG" in os.environ and os.environ["DEBUG"] in ["true", "1"]:
|
|
constants.debug_mode = True
|
|
|
|
ap = app.Application()
|
|
|
|
ap.event_loop = loop
|
|
|
|
# 执行启动阶段
|
|
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:
|
|
|
|
# 挂系统信号处理
|
|
import signal
|
|
|
|
ap: app.Application
|
|
|
|
def signal_handler(sig, frame):
|
|
print("[Signal] 程序退出.")
|
|
# ap.shutdown()
|
|
os._exit(0)
|
|
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
|
|
app_inst = await make_app(loop)
|
|
ap = app_inst
|
|
await app_inst.run()
|
|
except Exception as e:
|
|
traceback.print_exc()
|