jianf/TrulyMEM-TrueHumanMEM-local
1
0
Fork 0
Code Issues Pull Requests Actions 10 Packages Projects Releases Wiki Activity

fix: web_api.py 使用 get_global_db() 替代全局 graph_db 变量

Browse Source
This commit is contained in:
root
2026-04-29 07:26:25 +08:00
parent 00fd75a0a6
commit 529672a93a
1 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
core/web_api.py
View File

@ -146,9 +146,8 @@ def reload_server_for_user(username: str):
def login_page(): def login_page():
"""登录页面 - 如果没有用户则重定向到设置页""" """登录页面 - 如果没有用户则重定向到设置页"""
# 如果没有用户，重定向到首次设置页 # 如果没有用户，重定向到首次设置页
users_count = 0 g_db = get_global_db()
if graph_db: users_count = g_db.get_web_users_count() if g_db else 0
users_count = graph_db.get_web_users_count()
if users_count == 0: if users_count == 0:
return redirect('/setup') return redirect('/setup')
return render_template('login.html') return render_template('login.html')
@ -157,9 +156,8 @@ def login_page():
@app.route('/setup') @app.route('/setup')
def setup_page(): def setup_page():
"""首次设置页面 - 如果已有用户则跳转到登录页""" """首次设置页面 - 如果已有用户则跳转到登录页"""
has_users = False g_db = get_global_db()
if graph_db: has_users = g_db.get_web_users_count() > 0 if g_db else False
has_users = graph_db.get_web_users_count() > 0
if has_users: if has_users:
return redirect('/login') return redirect('/login')
return render_template('setup.html') return render_template('setup.html')
@ -231,9 +229,8 @@ def userinfo():
@app.route('/api/web-check', methods=['GET']) @app.route('/api/web-check', methods=['GET'])
def web_check(): def web_check():
"""检查是否需要首次设置，返回是否配置完成""" """检查是否需要首次设置，返回是否配置完成"""
users_count = 0 g_db = get_global_db()
if graph_db: users_count = g_db.get_web_users_count() if g_db else 0
users_count = graph_db.get_web_users_count()
return jsonify({ return jsonify({
"needs_setup": users_count == 0, "needs_setup": users_count == 0,