jianf/TrulyMEM-TrueHumanMEM
1
0
Fork 0
mirror of https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git synced 2026-09-20 17:08:18 +00:00
Code Issues 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():
"""登录页面 - 如果没有用户则重定向到设置页"""
# 如果没有用户，重定向到首次设置页
users_count = 0
if graph_db:
users_count = graph_db.get_web_users_count()
g_db = get_global_db()
users_count = g_db.get_web_users_count() if g_db else 0
if users_count == 0:
return redirect('/setup')
return render_template('login.html')
@ -157,9 +156,8 @@ def login_page():
@app.route('/setup')
def setup_page():
"""首次设置页面 - 如果已有用户则跳转到登录页"""
has_users = False
if graph_db:
has_users = graph_db.get_web_users_count() > 0
g_db = get_global_db()
has_users = g_db.get_web_users_count() > 0 if g_db else False
if has_users:
return redirect('/login')
return render_template('setup.html')
@ -231,9 +229,8 @@ def userinfo():
@app.route('/api/web-check', methods=['GET'])
def web_check():
"""检查是否需要首次设置，返回是否配置完成"""
users_count = 0
if graph_db:
users_count = graph_db.get_web_users_count()
g_db = get_global_db()
users_count = g_db.get_web_users_count() if g_db else 0
return jsonify({
"needs_setup": users_count == 0,