59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
"""
|
|
pccs — 算力中心集群系统 (Pooled Computing Cluster System)
|
|
入口: webapp(init) 模式
|
|
"""
|
|
import os, sys
|
|
|
|
from ahserver.webapp import webapp
|
|
|
|
# ═══ 模块级 Monkey Patches (必须在任何 rbac/sqlor 使用前执行) ═══
|
|
|
|
# 1. DictObject.__getattr__ — 所有 ServerEnv 实例默认 get_module_dbname
|
|
from appPublic.dictObject import DictObject
|
|
|
|
_orig_getattr = DictObject.__getattr__
|
|
|
|
|
|
def _patched_getattr(self, key):
|
|
if key == 'get_module_dbname':
|
|
return lambda m: 'pccs'
|
|
if key == 'password_encode':
|
|
return lambda s: s
|
|
return _orig_getattr(self, key)
|
|
|
|
|
|
DictObject.__getattr__ = _patched_getattr
|
|
|
|
# 2. get_sor_context — 复用全局 DBPools 而非创建空实例
|
|
import os as _os
|
|
from appPublic.jsonConfig import getConfig as _getConfig
|
|
import sqlor.dbpools as _dbp
|
|
from sqlor.dbpools import DBPools
|
|
|
|
_workdir = _os.path.dirname(_os.path.dirname(_os.path.abspath(__file__)))
|
|
_config = _getConfig(_workdir, NS={'workdir': _workdir})
|
|
_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
|
|
print('[pccs] patched get_sor_context', flush=True)
|
|
|
|
|
|
def init():
|
|
from appbase.init import load_appbase
|
|
from rbac.init import load_rbac
|
|
load_appbase()
|
|
load_rbac()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
webapp(init)
|