yumoqing/pccs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add @asynccontextmanager to patched get_sor_context

Browse Source
This commit is contained in:
yumoqing 2026-08-11 17:02:05 +08:00
parent 186ed11d55
commit 8c378f0ef0
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/pccs.py
View File

@ -3,6 +3,7 @@ pccs — 算力中心集群系统 (Pooled Computing Cluster System)
入口: webapp(init) 模式 入口: webapp(init) 模式
""" """
import os, sys import os, sys
from contextlib import asynccontextmanager
# ═══ 最早导入并 patch sqlor, 确保所有后续 import 拿到 patched 版本 ═══ # ═══ 最早导入并 patch sqlor, 确保所有后续 import 拿到 patched 版本 ═══
import sqlor.dbpools as _dbp import sqlor.dbpools as _dbp
@ -16,6 +17,7 @@ _global_db = _dbp.DBPools(_config.databases)
_orig_sor_ctx = _dbp.get_sor_context _orig_sor_ctx = _dbp.get_sor_context
@asynccontextmanager
async def _patched_sor_ctx(env, modulename, errorback=None): async def _patched_sor_ctx(env, modulename, errorback=None):
dbname = env.get_module_dbname(modulename) dbname = env.get_module_dbname(modulename)
async with _global_db.sqlorContext(dbname) as sor: async with _global_db.sqlorContext(dbname) as sor: