pccs/app/pccs.py

50 lines
1.3 KiB
Python

"""
pccs — 算力中心集群系统 (Pooled Computing Cluster System)
入口: webapp(init) 模式
"""
import os, sys
from ahserver.webapp import webapp
def init():
# Monkey-patch DictObject.__getattr__ — 所有 ServerEnv 实例默认返回 'pccs'
from appPublic.dictObject import DictObject
_orig = DictObject.__getattr__
def _patched(self, key):
if key == 'get_module_dbname':
return lambda m: 'pccs'
if key == 'password_encode':
return lambda s: s
return _orig(self, key)
DictObject.__getattr__ = _patched
# Monkey-patch get_sor_context — 复用已初始化的 DBPools 而非创建空实例
from appPublic.jsonConfig import getConfig
import sqlor.dbpools as _dbp
from sqlor.dbpools import DBPools
config = getConfig()
_global_db = DBPools(config.databases)
_orig_sor_ctx = _dbp.get_sor_context
async def _patched_sor_ctx(env, modulename, errorback=None):
dbname = env.get_module_dbname(modulename)
async with _global_db.sqlorContext(dbname) as sor:
yield sor
_dbp.get_sor_context = _patched_sor_ctx
from appbase.init import load_appbase
from rbac.init import load_rbac
load_appbase()
load_rbac()
if __name__ == '__main__':
webapp(init)