Browse Source

Merge branch 'lbj' of http://124.70.32.114:3100/hanyuqing/KGPython into hanyuqing

# Conflicts:
#	vue/vue.config.js
#	web_main.py
yangrongze
hanyuqing 4 months ago
parent
commit
84cfa04ed5
  1. 338
      controller/LoginController.py
  2. 182
      controller/RegisterController.py
  3. BIN
      resource/avatar/1_1766477129_1706.jpg
  4. BIN
      resource/avatar/2.jpg
  5. BIN
      resource/avatar/4.png
  6. BIN
      resource/avatar/admin_1766388917_8519.jpg
  7. BIN
      resource/avatar/admin_1766389471_9334.jpg
  8. BIN
      resource/avatar/f8c6732809a04f74bc4f3856b949de5e.jpg
  9. 72
      service/UserService.py
  10. 33
      vue/src/api/register.js
  11. 6
      vue/src/router/index.js
  12. 7
      vue/src/system/Login.vue
  13. 42
      vue/src/system/Profile.vue
  14. 763
      vue/src/system/Register.vue

338
controller/LoginController.py

@ -1,4 +1,4 @@
from robyn import jsonify, Response
from robyn import jsonify, Response, Request
from app import app
from datetime import datetime, timedelta
import uuid
@ -12,11 +12,34 @@ def generate_token() -> str:
"""生成随机token"""
return str(uuid.uuid4())
def validate_token(token: str) -> dict:
"""验证token并返回用户信息"""
if not token or token not in TEMP_TOKENS:
return None
# 检查token是否过期
if datetime.now() > TEMP_TOKENS[token]["expires_at"]:
del TEMP_TOKENS[token]
return None
return TEMP_TOKENS[token]["user"]
@app.post("/api/login")
def login_route(request):
"""登录接口"""
try:
request_data = json.loads(request.body) if request.body else {}
# 解析请求数据
request_data = {}
if request.body:
try:
request_data = json.loads(request.body)
except json.JSONDecodeError:
pass
# 如果JSON解析失败,尝试从form_data获取
if not request_data and hasattr(request, 'form_data'):
request_data = getattr(request, 'form_data', {})
username = request_data.get("username", "").strip()
password = request_data.get("password", "").strip()
remember = request_data.get("remember", False)
@ -83,26 +106,28 @@ def user_info_route(request):
query_params = getattr(request, 'query_params', {})
token = query_params.get("token", "")
# 验证token是否存在
if not token or token not in TEMP_TOKENS:
# 验证token
user = validate_token(token)
if not user:
return Response(
status_code=401,
description=jsonify({"success": False, "message": "未登录或登录已过期"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 检查token是否过期
if datetime.now() > TEMP_TOKENS[token]["expires_at"]:
del TEMP_TOKENS[token]
return Response(
status_code=401,
description=jsonify({"success": False, "message": "登录已过期"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 从数据库获取最新的用户信息
username = user["username"]
db_user = user_service.get_user_info(username)
if db_user:
# 更新TEMP_TOKENS中的用户信息
TEMP_TOKENS[token]["user"] = db_user
user_info = db_user
else:
user_info = user
return Response(
status_code=200,
description=jsonify({"success": True, "user": TEMP_TOKENS[token]["user"]}),
description=jsonify({"success": True, "user": user_info}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
except Exception as e:
@ -113,140 +138,100 @@ def user_info_route(request):
)
@app.post("/api/updateAvatar")
def update_avatar_route(request):
async def update_avatar_route(request: Request):
"""更新用户头像接口"""
try:
# 打印调试信息
print(f"请求类型: {type(request)}")
print(f"请求属性: {dir(request)}")
# 在Robyn中,文件上传的数据存储在request.files中
# 表单字段存储在request.form中
form_data = getattr(request, 'form', {})
files_data = getattr(request, 'files', {})
print(f"表单数据: {form_data}")
print(f"文件数据: {files_data}")
print(f"表单数据类型: {type(form_data)}")
print(f"文件数据类型: {type(files_data)}")
# 获取token - 尝试多种方式获取
# 从files中获取文件和token
avatar_file = request.files.get('avatar') if hasattr(request, 'files') else None
token = None
# 方法1: 从form_data中获取
if isinstance(form_data, dict) and "token" in form_data:
token = form_data["token"]
print(f"从form_data获取token: {token}")
# 方法2: 如果form_data不是字典,尝试其他方式
if not token and hasattr(form_data, 'get'):
token = form_data.get("token", "")
print(f"通过form_data.get()获取token: {token}")
# 从form_data中获取token
if hasattr(request, 'form_data'):
token = request.form_data.get('token')
# 如果files中没有直接找到'avatar',尝试获取第一个文件
if not avatar_file and hasattr(request, 'files') and request.files:
first_key = list(request.files.keys())[0]
avatar_file = request.files[first_key]
# 方法3: 尝试从request的属性中直接获取
# 如果form_data中没有token,尝试从headers获取
if not token:
# 尝试从request中获取所有可能的属性
for attr_name in dir(request):
if 'token' in attr_name.lower() or 'form' in attr_name.lower():
try:
attr_value = getattr(request, attr_name)
print(f"request.{attr_name}: {type(attr_value)} = {attr_value}")
# 如果是字典类型,检查是否包含token
if isinstance(attr_value, dict) and 'token' in attr_value:
token = attr_value['token']
print(f"从request.{attr_name}获取token: {token}")
break
except Exception as e:
print(f"访问request.{attr_name}时出错: {e}")
# 方法4: 从查询参数中获取
if not token:
query_params = getattr(request, 'query_params', {})
if isinstance(query_params, dict) and "token" in query_params:
token = query_params["token"]
print(f"从查询参数获取token: {token}")
# 方法5: 从headers中获取
if not token:
headers = getattr(request, 'headers', {})
if isinstance(headers, dict) and "Authorization" in headers:
auth_header = headers["Authorization"]
if auth_header.startswith("Bearer "):
token = auth_header[7:]
print(f"从Authorization头获取token: {token}")
print(f"最终获取的token: {token}")
print(f"TEMP_TOKENS中的keys: {list(TEMP_TOKENS.keys())}")
headers_dict = {}
if hasattr(request, 'headers'):
try:
headers_dict = dict(request.headers)
except:
pass
token = headers_dict.get('Authorization') or headers_dict.get('authorization')
# 如果还是没有token,尝试从body中解析
if not token and hasattr(request, 'body'):
try:
body_data = json.loads(request.body)
token = body_data.get('token')
except:
pass
# 验证token
if not token or token not in TEMP_TOKENS:
print(f"Token验证失败: {token}")
user = validate_token(token)
if not user:
return Response(
status_code=401,
description=jsonify({"success": False, "message": "未登录或登录已过期"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 检查token是否过期
if datetime.now() > TEMP_TOKENS[token]["expires_at"]:
del TEMP_TOKENS[token]
print("Token已过期")
# 检查文件是否存在
if not avatar_file:
return Response(
status_code=401,
description=jsonify({"success": False, "message": "登录已过期"}),
status_code=400,
description=jsonify({"success": False, "message": "未上传头像文件"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 获取上传的文件
avatar_file = files_data.get("avatar") if isinstance(files_data, dict) else None
# 如果没有通过"avatar"键获取到文件,尝试直接访问files_data的第一个元素
if not avatar_file and isinstance(files_data, dict) and len(files_data) > 0:
# 获取第一个文件作为头像文件
first_key = list(files_data.keys())[0]
avatar_file = files_data[first_key]
print(f"通过备用方法获取文件,键名: {first_key}")
if not avatar_file:
print("未找到avatar文件")
print(f"可用的文件键: {list(files_data.keys()) if isinstance(files_data, dict) else '无法获取键列表'}")
# 获取文件内容和文件名
file_content = None
filename = ""
# Robyn框架中,files字典的值通常是字节数据
if isinstance(avatar_file, bytes):
file_content = avatar_file
filename = list(request.files.keys())[0] if request.files else "avatar.jpg"
elif isinstance(avatar_file, str):
filename = avatar_file
else:
if avatar_file is not None:
# 尝试获取文件内容的其他方式
if hasattr(avatar_file, 'content'):
file_content = avatar_file.content
elif hasattr(avatar_file, 'read'):
file_content = avatar_file.read()
filename = getattr(avatar_file, 'filename', 'avatar.jpg')
# 如果我们有文件内容,继续处理
if not file_content:
return Response(
status_code=400,
description=jsonify({"success": False, "message": "未上传头像文件"}),
description=jsonify({"success": False, "message": "无法读取文件内容"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
print(f"获取的文件: {avatar_file}")
print(f"文件属性: {dir(avatar_file) if avatar_file else '无文件'}")
# 处理文件名
if not filename:
filename = "avatar.jpg"
file_extension = filename.split('.')[-1] if '.' in filename else 'jpg'
# 检查文件类型 - 使用多种方法验证
# 验证文件类型
is_valid_image = False
file_extension = ""
# 方法1: 检查content_type
content_type = getattr(avatar_file, 'content_type', '')
if content_type and content_type.startswith("image/"):
# 通过扩展名验证
if file_extension.lower() in ['jpg', 'jpeg', 'png', 'gif', 'bmp', 'webp', 'svg']:
is_valid_image = True
print(f"通过content_type验证: {content_type}")
# 方法2: 检查文件名和扩展名
filename = getattr(avatar_file, 'filename', '')
if filename:
file_extension = os.path.splitext(filename)[1].lower()
valid_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp', '.svg']
if file_extension in valid_extensions:
is_valid_image = True
print(f"通过文件扩展名验证: {file_extension}")
# 如果两种方法都失败,尝试读取文件头
# 如果扩展名验证失败,尝试读取文件头验证
if not is_valid_image:
try:
# 读取文件前几个字节来检测文件类型
file_content = avatar_file.read(1024)
avatar_file.seek(0) # 重置文件指针
# 检查常见图片格式的文件头
if (file_content.startswith(b'\xFF\xD8\xFF') or # JPEG
file_content.startswith(b'\x89PNG\r\n\x1a\n') or # PNG
file_content.startswith(b'GIF87a') or # GIF
@ -254,64 +239,47 @@ def update_avatar_route(request):
file_content.startswith(b'BM') or # BMP
file_content.startswith(b'RIFF') and b'WEBP' in file_content[:12]): # WebP
is_valid_image = True
print(f"通过文件头验证成功")
else:
print(f"文件头验证失败,文件前16字节: {file_content[:16]}")
except Exception as e:
print(f"读取文件内容进行验证时出错: {e}")
except Exception:
pass
if not is_valid_image:
print(f"文件类型验证失败 - content_type: {content_type}, filename: {filename}, extension: {file_extension}")
return Response(
status_code=400,
description=jsonify({"success": False, "message": f"文件类型必须是图片 (支持的格式: JPG, PNG, GIF, BMP, WebP, SVG)"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 获取文件内容
file_content = avatar_file.file_data
print(f"读取的文件大小: {len(file_content)} 字节")
# 生成唯一的文件名
import time
import random
filename = getattr(avatar_file, 'filename', '')
file_extension = filename.split('.')[-1] if '.' in filename else 'jpg'
timestamp = int(time.time())
random_num = random.randint(1000, 9999)
username = TEMP_TOKENS[token]["user"]["username"]
username = user["username"]
new_filename = f"{username}_{timestamp}_{random_num}.{file_extension}"
# 定义文件保存路径
avatar_dir = "D:/zhishitupu/MedKG/resource/avatar"
file_path = f"{avatar_dir}/{new_filename}"
print(f"保存文件到: {file_path}")
# 定义文件保存路径(使用相对路径)
current_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
avatar_dir = os.path.join(current_dir, "resource", "avatar")
file_path = os.path.join(avatar_dir, new_filename)
# 确保目录存在
import os
os.makedirs(avatar_dir, exist_ok=True)
# 保存文件到磁盘
with open(file_path, "wb") as f:
f.write(file_content)
print(f"文件保存成功")
# 更新用户头像到数据库,存储相对路径
avatar_relative_path = f"/resource/avatar/{new_filename}"
success = user_service.update_user_avatar(username, avatar_relative_path)
if not success:
print("数据库更新失败")
return Response(
status_code=500,
description=jsonify({"success": False, "message": "更新头像失败"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
print("数据库更新成功")
# 更新token中的用户信息
TEMP_TOKENS[token]["user"]["avatar"] = avatar_relative_path
@ -325,9 +293,6 @@ def update_avatar_route(request):
headers={"Content-Type": "application/json; charset=utf-8"}
)
except Exception as e:
print(f"更新头像异常: {str(e)}")
import traceback
traceback.print_exc()
return Response(
status_code=500,
description=jsonify({"success": False, "message": f"更新头像失败: {str(e)}"}),
@ -338,23 +303,15 @@ def update_avatar_route(request):
def update_password_route(request):
"""更新用户密码接口"""
try:
print("开始处理密码更新请求")
# 解析请求数据
request_data = json.loads(request.body) if request.body else {}
print(f"请求数据: {request_data}")
token = request_data.get("token", "")
current_password = request_data.get("currentPassword", "")
new_password = request_data.get("newPassword", "")
print(f"Token: {token}")
print(f"当前密码长度: {len(current_password)}")
print(f"新密码长度: {len(new_password)}")
# 验证输入
if not current_password or not new_password:
print("密码为空")
return Response(
status_code=400,
description=jsonify({"success": False, "message": "当前密码和新密码不能为空"}),
@ -362,46 +319,29 @@ def update_password_route(request):
)
# 验证token
if not token or token not in TEMP_TOKENS:
print(f"Token验证失败: {token}")
user = validate_token(token)
if not user:
return Response(
status_code=401,
description=jsonify({"success": False, "message": "未登录或登录已过期"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 检查token是否过期
if datetime.now() > TEMP_TOKENS[token]["expires_at"]:
del TEMP_TOKENS[token]
print("Token已过期")
return Response(
status_code=401,
description=jsonify({"success": False, "message": "登录已过期"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 获取用户信息
username = TEMP_TOKENS[token]["user"]["username"]
print(f"用户名: {username}")
username = user["username"]
# 验证当前密码
user = user_service.get_user_by_username(username)
if not user:
print("用户不存在")
db_user = user_service.get_user_by_username(username)
if not db_user:
return Response(
status_code=404,
description=jsonify({"success": False, "message": "用户不存在"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
print(f"用户信息: {user}")
# 验证密码
is_password_valid = user_service.verify_password(current_password, user["password"])
print(f"密码验证结果: {is_password_valid}")
is_password_valid = user_service.verify_password(current_password, db_user["password"])
if not is_password_valid:
print("当前密码不正确")
return Response(
status_code=401,
description=jsonify({"success": False, "message": "当前密码不正确"}),
@ -410,69 +350,25 @@ def update_password_route(request):
# 更新密码
success = user_service.update_user_password(username, new_password)
print(f"密码更新结果: {success}")
if not success:
print("密码更新失败")
return Response(
status_code=500,
description=jsonify({"success": False, "message": "密码更新失败"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
print("密码更新成功")
return Response(
status_code=200,
description=jsonify({"success": True, "message": "密码更新成功"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
except Exception as e:
print(f"密码更新异常: {str(e)}")
import traceback
traceback.print_exc()
return Response(
status_code=500,
description=jsonify({"success": False, "message": f"密码更新失败: {str(e)}"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
@app.get("/api/test/db")
def test_db_connection(request):
"""测试数据库连接"""
try:
# 检查数据库连接状态
is_connected = user_service.mysql.is_connected()
print(f"数据库连接状态: {is_connected}")
# 尝试查询用户表
sql = "SELECT COUNT(*) as user_count FROM users"
result = user_service.mysql.execute_query(sql)
print(f"查询结果: {result}")
# 打印当前的token信息用于调试
print(f"当前TEMP_TOKENS: {TEMP_TOKENS}")
return Response(
status_code=200,
description=jsonify({
"success": True,
"message": "数据库连接测试成功",
"is_connected": is_connected,
"user_count": result[0]["user_count"] if result else 0,
"tokens": list(TEMP_TOKENS.keys()) # 返回当前有效的token列表
}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
except Exception as e:
print(f"数据库连接测试失败: {str(e)}")
import traceback
traceback.print_exc()
return Response(
status_code=500,
description=jsonify({"success": False, "message": f"数据库连接测试失败: {str(e)}"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
@app.after_request("/")
def add_cors_headers(response):
"""添加CORS头,支持跨域请求"""

182
controller/RegisterController.py

@ -0,0 +1,182 @@
from robyn import jsonify, Response
from app import app
import os
import uuid
from service.UserService import user_service
# 头像上传目录
AVATAR_UPLOAD_FOLDER = 'resource/avatar'
@app.post("/api/register")
def register_route(request):
"""用户注册接口"""
try:
# 获取表单数据
form_data = getattr(request, 'form_data', {}) if hasattr(request, 'form_data') else {}
# 获取文件上传
files = getattr(request, 'files', {}) if hasattr(request, 'files') else {}
username = form_data.get("username", "").strip()
password = form_data.get("password", "").strip()
# 验证必填字段
if not username or not password:
return Response(
status_code=400,
description=jsonify({"success": False, "message": "用户名和密码不能为空"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 检查用户名是否已存在
existing_user = user_service.get_user_by_username(username)
if existing_user:
return Response(
status_code=409,
description=jsonify({"success": False, "message": "用户名已存在"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 处理头像上传
avatar_path = "/resource/avatar/4.png" # 默认头像
# 获取头像文件
avatar_file = files.get('avatar')
# 如果有头像文件,处理上传
if avatar_file:
# 获取文件内容和文件名
file_content = None
filename = ""
# Robyn框架中,files字典的值通常是字节数据
if isinstance(avatar_file, bytes):
file_content = avatar_file
filename = "avatar.jpg"
elif hasattr(avatar_file, 'content'):
file_content = avatar_file.content
filename = getattr(avatar_file, 'filename', 'avatar.jpg')
elif hasattr(avatar_file, 'read'):
file_content = avatar_file.read()
filename = getattr(avatar_file, 'filename', 'avatar.jpg')
else:
filename = avatar_file if isinstance(avatar_file, str) else 'avatar.jpg'
# 如果我们有文件内容,继续处理
if file_content:
# 处理文件名
file_extension = filename.split('.')[-1] if '.' in filename else 'jpg'
# 验证文件类型
is_valid_image = False
# 通过扩展名验证
if file_extension.lower() in ['jpg', 'jpeg', 'png', 'gif', 'bmp', 'webp', 'svg']:
is_valid_image = True
# 如果扩展名验证失败,尝试读取文件头验证
if not is_valid_image:
try:
if (file_content.startswith(b'\xFF\xD8\xFF') or # JPEG
file_content.startswith(b'\x89PNG\r\n\x1a\n') or # PNG
file_content.startswith(b'GIF87a') or # GIF
file_content.startswith(b'GIF89a')): # GIF
is_valid_image = True
except:
pass
if not is_valid_image:
return Response(
status_code=400,
description=jsonify({"success": False, "message": "不支持的图片格式"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 确保上传目录存在
if not os.path.exists(AVATAR_UPLOAD_FOLDER):
os.makedirs(AVATAR_UPLOAD_FOLDER)
# 生成唯一文件名
unique_filename = f"{uuid.uuid4().hex}.{file_extension}"
file_path = os.path.join(AVATAR_UPLOAD_FOLDER, unique_filename)
# 保存文件
try:
with open(file_path, 'wb') as f:
f.write(file_content)
avatar_path = f"/{file_path}"
except Exception as e:
print(f"保存头像失败: {e}")
return Response(
status_code=500,
description=jsonify({"success": False, "message": "头像保存失败"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 创建用户
user_id = user_service.create_user(username, password, avatar_path)
if user_id:
return Response(
status_code=201,
description=jsonify({
"success": True,
"message": "注册成功",
"user": {
"id": user_id,
"username": username,
"avatar": avatar_path
}
}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
else:
return Response(
status_code=500,
description=jsonify({"success": False, "message": "注册失败,请稍后再试"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
except Exception as e:
print(f"注册异常: {e}")
return Response(
status_code=500,
description=jsonify({"success": False, "message": f"注册失败: {str(e)}"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
@app.get("/api/checkUsername")
def check_username_route(request):
"""检查用户名是否可用"""
try:
# 获取查询参数
username = ""
# 尝试从query_params获取
if hasattr(request, 'query_params') and request.query_params:
username = request.query_params.get("username", "").strip()
if not username:
return Response(
status_code=400,
description=jsonify({"success": False, "message": "用户名不能为空"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
# 检查用户名是否已存在
existing_user = user_service.get_user_by_username(username)
return Response(
status_code=200,
description=jsonify({
"success": True,
"available": existing_user is None,
"message": "用户名可用" if not existing_user else "用户名已存在"
}),
headers={"Content-Type": "application/json; charset=utf-8"}
)
except Exception as e:
print(f"检查用户名异常: {e}")
return Response(
status_code=500,
description=jsonify({"success": False, "message": f"检查失败: {str(e)}"}),
headers={"Content-Type": "application/json; charset=utf-8"}
)

BIN
resource/avatar/1_1766477129_1706.jpg

Binary file not shown.

After

Width:  |  Height:  |  Size: 48 KiB

BIN
resource/avatar/2.jpg

Binary file not shown.

After

Width:  |  Height:  |  Size: 58 KiB

BIN
resource/avatar/4.png

Binary file not shown.

After

Width:  |  Height:  |  Size: 13 KiB

BIN
resource/avatar/admin_1766388917_8519.jpg

Binary file not shown.

After

Width:  |  Height:  |  Size: 58 KiB

BIN
resource/avatar/admin_1766389471_9334.jpg

Binary file not shown.

After

Width:  |  Height:  |  Size: 58 KiB

BIN
resource/avatar/f8c6732809a04f74bc4f3856b949de5e.jpg

Binary file not shown.

After

Width:  |  Height:  |  Size: 48 KiB

72
service/UserService.py

@ -29,10 +29,15 @@ class UserService:
# 验证密码 - 使用bcrypt验证
if bcrypt.checkpw(password.encode('utf-8'), stored_password.encode('utf-8')):
avatar_path = user.get("avatar")
# 如果头像为空,设置默认头像
if not avatar_path:
avatar_path = "/resource/avatar/4.png"
return {
"id": user.get("id"),
"username": user.get("username"),
"avatar": user.get("avatar")
"avatar": avatar_path
}
return None
@ -40,6 +45,27 @@ class UserService:
print(f"验证用户失败: {e}")
return None
def get_user_info(self, username: str) -> Optional[Dict[str, Any]]:
"""获取用户信息(不验证密码)"""
try:
user = self.get_user_by_username(username)
if not user:
return None
avatar_path = user.get("avatar")
# 如果头像为空,设置默认头像
if not avatar_path:
avatar_path = "/resource/avatar/4.png"
return {
"id": user.get("id"),
"username": user.get("username"),
"avatar": avatar_path
}
except Exception as e:
print(f"获取用户信息失败: {e}")
return None
def update_user_avatar(self, username: str, avatar_path: str) -> bool:
"""更新用户头像"""
try:
@ -96,10 +122,50 @@ class UserService:
except Exception as e:
print(f"验证密码失败: {e}")
return False
def create_user(self, username: str, password: str, avatar_path: str = None) -> int:
"""创建新用户"""
try:
# 检查数据库连接状态
if not self.mysql.is_connected():
print("数据库未连接,尝试重新连接...")
if not self.mysql.connect():
print("数据库连接失败")
return 0
# 使用bcrypt加密密码
hashed_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
# 设置默认头像
if not avatar_path:
avatar_path = "/resource/avatar/4.png"
# 插入新用户
sql = "INSERT INTO users (username, password, avatar) VALUES (%s, %s, %s)"
print(f"执行SQL: {sql}")
print(f"参数: {username}, {hashed_password}, {avatar_path}")
# 执行插入
result = self.mysql.execute_update(sql, (username, hashed_password, avatar_path))
if result > 0:
# 获取插入的用户ID
id_sql = "SELECT LAST_INSERT_ID() as user_id"
id_result = self.mysql.execute_query(id_sql)
if id_result:
user_id = id_result[0].get("user_id", 0)
print(f"插入结果,用户ID: {user_id}")
return user_id
return 0
except Exception as e:
print(f"创建用户失败: {e}")
import traceback
traceback.print_exc()
return 0
# 创建全局用户服务实例
user_service = UserService()
# 初始化MySQL连接
def init_mysql_connection():
"""初始化MySQL连接"""
@ -107,4 +173,4 @@ def init_mysql_connection():
return user_service.mysql.connect()
except Exception as e:
print(f"初始化MySQL连接失败: {e}")
return False
return False

33
vue/src/api/register.js

@ -0,0 +1,33 @@
// src/api/register.js
import request from '@/utils/request';
/**
* 用户注册
* 后端接口POST /api/register
* @param {FormData} formData - 包含用户名密码和头像的表单数据
*/
export function register(formData) {
return request({
url: '/api/register',
method: 'post',
data: formData,
headers: {
'Content-Type': 'multipart/form-data'
}
});
}
/**
* 检查用户名是否可用
* 后端接口GET /api/checkUsername
* @param {string} username - 要检查的用户名
*/
export function checkUsername(username) {
return request({
url: '/api/checkUsername',
method: 'get',
params: {
username
}
});
}

6
vue/src/router/index.js

@ -1,5 +1,6 @@
import { createRouter, createWebHistory } from 'vue-router'
import Login from '../system/Login.vue'
import Register from '../system/Register.vue'
import Index from '../system/Index.vue'
import Profile from '../system/Profile.vue'
import Display from '../system/GraphDemo.vue'
@ -17,6 +18,11 @@ const routes = [
component: Login
},
{
path: '/register',
name: 'Register',
component: Register
},
{
path: '/index',
name: 'Index',
component: Index

7
vue/src/system/Login.vue

@ -73,7 +73,7 @@
<!-- </div>-->
<div class="register-link">
<p>还没有账户? <a href="#" class="register"> 立即注册</a></p>
<p>还没有账户? <a href="#" class="register" @click.prevent="goToRegister"> 立即注册</a></p>
</div>
</div>
</div>
@ -152,6 +152,11 @@ const handleLogin = async () => {
loading.value = false;
}
};
//
const goToRegister = () => {
router.push('/register');
};
</script>
<style>

42
vue/src/system/Profile.vue

@ -19,7 +19,12 @@
<!-- 头像区域 -->
<div class="avatar-section">
<div class="avatar-container">
<img :src="userProfile.avatar" alt="用户头像" class="avatar-image">
<img
:src="userProfile.avatar"
alt="用户头像"
class="avatar-image"
@error="handleAvatarError"
>
<div class="avatar-overlay" @click="triggerFileInput">
<span class="camera-icon">📷</span>
<span class="overlay-text">更换头像</span>
@ -36,6 +41,7 @@
<div class="avatar-status">
<p v-if="avatarUploading" class="uploading-text">上传中...</p>
<p v-if="avatarUploadSuccess" class="success-text">头像更新成功</p>
</div>
</div>
@ -113,8 +119,8 @@ const handleSidebarClick = (menuItem) => {
//
const userProfile = ref({
username: 'admin',
avatar: '/resource/avatar/1.jpg'
username: '用户',
avatar: '/resource/avatar/4.png'
});
//
@ -140,11 +146,19 @@ const passwordMismatch = computed(() => {
passwordForm.value.newPassword !== passwordForm.value.confirmPassword;
});
//
const triggerFileInput = () => {
fileInput.value.click();
};
//
const handleAvatarError = (event) => {
console.error('头像加载失败:', userProfile.value.avatar);
//
userProfile.value.avatar = '/resource/avatar/4.png';
};
//
const handleAvatarChange = async (event) => {
const file = event.target.files[0];
@ -185,7 +199,11 @@ const handleAvatarChange = async (event) => {
console.log('Token:', token);
// API
const response = await updateAvatar(formData);
const response = await updateAvatar(formData,{
headers: {
'Content-Type': 'multipart/form-data'
}
});
console.log('收到响应:', response);
@ -314,26 +332,40 @@ onMounted(async () => {
// localStoragetoken
const token = localStorage.getItem('token');
console.log('Profile组件挂载,获取到的token:', token);
if (token) {
// API
const response = await getUserProfile(token);
console.log('获取用户信息响应:', response);
if (response.success) {
//
//
let avatarUrl = response.user.avatar || '/resource/avatar/1.jpg';
let avatarUrl = response.user.avatar || '/resource/avatar/4.png';
if (avatarUrl.startsWith('/resource/')) {
avatarUrl = avatarUrl; // 使
}
console.log('设置头像URL:', avatarUrl);
userProfile.value = {
username: response.user.username,
avatar: avatarUrl
};
} else {
console.error('获取用户信息失败:', response.message);
errorMessage.value = response.message || '获取用户信息失败';
// tokentoken
if (response.message && response.message.includes('登录')) {
localStorage.removeItem('token');
//
}
}
} else {
console.log('用户未登录');
errorMessage.value = '用户未登录,请先登录';
}
} catch (error) {

763
vue/src/system/Register.vue

@ -0,0 +1,763 @@
<template>
<div class="register-container">
<!-- 左上角Logo和标题 -->
<div class="logo-header">
<img src="@/assets/logo.png" alt="Logo" class="logo">
<h1 class="register-title">面向疾病预测的知识图谱应用系统</h1>
</div>
<!-- 左侧注册区域 -->
<div class="register-form-container">
<div class="register-header">
</div>
<div class="register-form">
<h2 class="form-title">注册</h2>
<p class="form-description">创建您的账户以访问系统功能</p>
<form class="form" @submit.prevent="handleRegister">
<!-- 错误信息显示 -->
<div v-if="errorMessage" class="error-message">
{{ errorMessage }}
</div>
<!-- 头像上传区域 -->
<div class="avatar-upload-container">
<div class="avatar-container">
<img
:src="avatarPreview"
alt="用户头像"
class="avatar-image"
@error="handleAvatarError"
>
<div class="avatar-overlay" @click="triggerFileInput">
<span class="camera-icon">📷</span>
<span class="overlay-text">上传头像</span>
</div>
<input
type="file"
ref="fileInput"
@change="handleAvatarChange"
accept="image/*"
class="hidden-input"
>
</div>
<div class="avatar-status">
<p v-if="avatarUploading" class="uploading-text">上传中...</p>
<p v-if="avatarUploadError" class="error-text">{{ avatarUploadError }}</p>
</div>
</div>
<div class="form-group">
<label for="username" class="form-label">用户名</label>
<input
type="text"
id="username"
v-model="registerForm.username"
placeholder="输入您的用户名"
class="form-input"
:class="{ 'error': errors.username }"
>
<p v-if="errors.username" class="error-text">{{ errors.username }}</p>
</div>
<div class="form-group">
<label for="password" class="form-label">密码</label>
<input
type="password"
id="password"
v-model="registerForm.password"
placeholder="输入您的密码"
class="form-input"
:class="{ 'error': errors.password }"
>
<p v-if="errors.password" class="error-text">{{ errors.password }}</p>
</div>
<div class="form-group">
<label for="confirmPassword" class="form-label">确认密码</label>
<input
type="password"
id="confirmPassword"
v-model="registerForm.confirmPassword"
placeholder="再次输入您的密码"
class="form-input"
:class="{ 'error': errors.confirmPassword }"
>
<p v-if="errors.confirmPassword" class="error-text">{{ errors.confirmPassword }}</p>
</div>
<div class="form-checkbox">
<input
type="checkbox"
id="agree"
v-model="registerForm.agree"
class="checkbox"
>
<label for="agree" class="checkbox-label">我同意服务条款和隐私政策</label>
</div>
<button
type="submit"
class="register-button"
:disabled="loading || !isFormValid"
>
<span v-if="!loading">注册</span>
<span v-else>注册中...</span>
</button>
</form>
<div class="social-login">
<p class="social-text">使用其他方式注册</p>
</div>
<div class="login-link">
<p>已有账户? <a href="#" class="login" @click.prevent="goToLogin"> 立即登录</a></p>
</div>
</div>
</div>
<!-- 右侧知识图谱可视化区域 -->
<div class="graph-container">
<!-- 背景装饰 -->
<div class="background-decoration">
<div class="bg-circle circle-1"></div>
<div class="bg-circle circle-2"></div>
</div>
</div>
</div>
</template>
<script setup>
import { ref, computed } from 'vue';
import { useRouter } from 'vue-router';
import { register } from '@/api/register';
const router = useRouter();
//
const registerForm = ref({
username: '',
password: '',
confirmPassword: '',
agree: false,
avatar: null
});
//
const fileInput = ref(null);
const avatarUploading = ref(false);
const avatarUploadError = ref('');
const loading = ref(false);
const errorMessage = ref('');
const errors = ref({
username: '',
password: '',
confirmPassword: ''
});
//
const defaultAvatar = '/resource/avatar/4.png';
const avatarPreview = ref(defaultAvatar);
// -
const isFormValid = computed(() => {
return registerForm.value.username &&
registerForm.value.password &&
registerForm.value.confirmPassword &&
registerForm.value.agree &&
!errors.value.username &&
!errors.value.password &&
!errors.value.confirmPassword &&
registerForm.value.password === registerForm.value.confirmPassword;
});
//
const triggerFileInput = () => {
fileInput.value.click();
};
//
const handleAvatarError = (event) => {
avatarPreview.value = defaultAvatar;
};
//
const handleAvatarChange = (event) => {
const file = event.target.files[0];
if (file) {
//
if (!file.type.startsWith('image/')) {
avatarUploadError.value = '请选择图片文件';
return;
}
// (2MB)
if (file.size > 2 * 1024 * 1024) {
avatarUploadError.value = '图片大小不能超过2MB';
return;
}
avatarUploadError.value = '';
//
const reader = new FileReader();
reader.onload = (e) => {
avatarPreview.value = e.target.result;
registerForm.value.avatar = file;
};
reader.readAsDataURL(file);
}
};
//
const validateForm = () => {
errors.value = {
username: '',
password: '',
confirmPassword: ''
};
//
if (!registerForm.value.username) {
errors.value.username = '用户名不能为空';
}
//
if (!registerForm.value.password) {
errors.value.password = '密码不能为空';
}
//
if (!registerForm.value.confirmPassword) {
errors.value.confirmPassword = '请确认密码';
} else if (registerForm.value.password !== registerForm.value.confirmPassword) {
errors.value.confirmPassword = '两次输入的密码不一致';
}
return !errors.value.username && !errors.value.password && !errors.value.confirmPassword;
};
//
const handleRegister = async () => {
if (!validateForm()) {
return;
}
if (!registerForm.value.agree) {
errorMessage.value = '请同意服务条款和隐私政策';
return;
}
loading.value = true;
errorMessage.value = '';
try {
// FormData
const formData = new FormData();
formData.append('username', registerForm.value.username);
formData.append('password', registerForm.value.password);
// FormData
if (registerForm.value.avatar) {
formData.append('avatar', registerForm.value.avatar);
}
// API
const response = await register(formData);
if (response.success) {
//
router.push('/login');
} else {
errorMessage.value = response.message || '注册失败';
}
} catch (error) {
console.error('注册失败:', error);
errorMessage.value = error.response?.data?.message || '注册过程中发生错误';
} finally {
loading.value = false;
}
};
//
const goToLogin = () => {
router.push('/login');
};
</script>
<style>
/* 全局样式,防止页面滚动 */
body, html {
margin: 0;
padding: 0;
overflow: hidden;
height: 100%;
}
</style>
<style scoped>
/* 基础容器样式 */
.register-container {
display: flex;
height: 100vh;
overflow: hidden;
flex-direction: row;
font-family: 'SimSun', '宋体', serif;
}
/* 左上角Logo和标题样式 */
.logo-header {
position: fixed;
top: 40px;
left: 25px;
display: flex;
align-items: center;
z-index: 10;
}
.logo {
height: 15px;
width: 15px;
margin-right: 7px;
}
.register-title {
font-size: 17px;
font-weight: 900;
font-family: 'SimSun Bold', '宋体', serif;
color: #1f2937;
margin: 0;
white-space: nowrap;
text-shadow: 0.4px 0.4px 0 #1f2937;
}
/* 左侧注册区域样式 */
.register-form-container {
width: 25%;
background-color: #ffffff;
padding: 2rem;
padding-left: 40px;
display: flex;
flex-direction: column;
justify-content: center;
position: relative;
}
.register-header {
margin-bottom: 1.5rem;
width: 100%;
max-width: 24rem;
margin-top: 20px;
}
.register-form {
max-width: 24rem;
width: 100%;
text-align: left;
}
.form-title {
font-size: 18px;
font-weight: 900;
color: #333333;
margin-top: -7px;
margin-bottom: 10px;
margin-left: 13px;
text-shadow: 0.2px 0.2px 0 #1f2937;
text-align: left;
font-family: 'SimSun', '宋体', serif;
}
.form-description {
color: #B5B5B5;
margin-bottom: 2rem;
margin-left: 13px;
text-align: left;
font-size: 11px;
font-weight: bold;
font-family: 'SimSun', '宋体', serif;
}
/* 头像上传区域 */
.avatar-upload-container {
display: flex;
flex-direction: column;
align-items: center;
margin-bottom: 25px;
margin-left: 13px;
}
.avatar-container {
position: relative;
width: 80px;
height: 80px;
margin-bottom: 10px;
}
.avatar-image {
width: 100%;
height: 100%;
border-radius: 50%;
object-fit: cover;
border: 3px solid #e5e7eb;
}
.avatar-overlay {
position: absolute;
top: 0;
left: 0;
width: 100%;
height: 100%;
background-color: rgba(0, 0, 0, 0.5);
border-radius: 50%;
display: flex;
flex-direction: column;
align-items: center;
justify-content: center;
opacity: 0;
transition: opacity 0.2s;
cursor: pointer;
}
.avatar-container:hover .avatar-overlay {
opacity: 1;
}
.camera-icon {
font-size: 1.2rem;
margin-bottom: 0.2rem;
}
.overlay-text {
font-size: 0.6rem;
color: white;
}
.hidden-input {
display: none;
}
.avatar-status {
text-align: center;
}
.uploading-text, .error-text {
font-size: 9px;
margin: 0.25rem 0;
}
.uploading-text {
color: #3b82f6;
}
.form {
display: flex;
flex-direction: column;
gap: 1rem;
width: 100%;
}
.error-message {
color: #ef4444;
font-size: 11px;
padding: 0.5rem;
background-color: #fef2f2;
border: 1px solid #fecaca;
border-radius: 0.375rem;
margin-bottom: 0.5rem;
}
.form-group {
display: flex;
flex-direction: column;
width: 100%;
margin-bottom: 0.5rem;
}
.form-label {
display: block;
font-size: 11px;
font-weight: 700;
color: #374151;
margin-bottom: 0.3rem;
text-align: left;
font-family:'STSong', '宋体', serif;
}
.form-input {
width: 100%;
padding: 0.6rem 0.8rem;
border-radius: 0.5rem;
border: 2px solid #A3A3A3;
transition: all 0.2s;
font-size: 9px;
box-sizing: border-box;
font-family: 'SimSun', '宋体', serif;
background-color: #FFFFFF;
}
.form-input:focus {
outline: none;
border-color: #2563eb;
box-shadow: 0 0 0 2px rgba(37, 99, 235, 0.2);
}
.form-input.error {
border-color: #ef4444;
}
.form-input::placeholder {
color: #9ca3af;
}
.password-header {
display: flex;
justify-content: space-between;
margin-bottom: 0.1rem;
}
.forgot-password {
font-size: 9px;
color: #B5B5B5;
text-decoration: none;
font-family: 'SimSun', '宋体', serif;
}
.forgot-password:hover {
color: #1d4ed8;
}
.form-checkbox {
display: flex;
align-items: center;
margin-top: -5px;
margin-bottom: -5px;
}
.checkbox {
height: 0.8rem;
width: 0.8rem;
color: #2563eb;
border-radius: 0.25rem;
border: 1px solid #d1d5db;
}
.checkbox-label {
margin-left: 0.1rem;
font-size: 11px;
color: #444040ba;
font-weight: bold;
font-family: 'SimSun', '宋体', serif;
}
.register-button {
width: 100%;
background-color: #409EFF;
color: white;
font-weight: 500;
font-size: 11px;
padding: 0.6rem 0.8rem;
border-radius: 0;
border: none;
cursor: pointer;
transition: background-color 0.2s;
display: flex;
align-items: center;
justify-content: center;
box-sizing: border-box;
font-family: 'SimSun', '宋体', 'STSong', '华文宋体', serif;
box-shadow: 3px 3px 5px rgba(0, 0, 0, 0.2);
}
.login-icon {
height: 0.9rem;
width: auto;
margin-right: 8px;
}
.register-button:hover:not(:disabled) {
background-color: #1d4ed8;
}
.register-button:disabled {
background-color: #bdc3c7;
cursor: not-allowed;
}
.arrow-icon {
margin-left: 0.5rem;
}
/* 分割线样式 */
.divider {
display: flex;
align-items: center;
margin: 1.5rem 0;
font-family: 'SimSun', '宋体', serif;
}
.divider::before,
.divider::after {
content: '';
flex: 1;
height: 1px;
background-color: #e5e7eb;
}
.divider span {
padding: 0 1rem;
font-size: 11px;
color: #B5B5B5;
font-family: 'SimSun', '宋体', serif;
}
.social-login {
margin-top: 2rem;
}
.social-text {
text-align: center;
color: #B5B5B5;
margin-bottom: 1rem;
font-size: 11px;
font-weight: bold;
font-family: 'SimSun', '宋体', serif;
}
.social-icons {
display: flex;
justify-content: center;
gap: 1rem;
}
.social-icon {
width: 2.5rem;
height: 2.5rem;
border-radius: 50%;
background-color: #f3f4f6;
border: none;
cursor: pointer;
transition: background-color 0.2s;
display: flex;
align-items: center;
justify-content: center;
color: #6b7280;
font-weight: bold;
}
.social-icon:hover {
background-color: #e5e7eb;
}
.login-link {
position: absolute;
bottom: 7px;
left: 50%;
transform: translateX(-50%);
text-align: center;
}
.login-link p {
color: #B5B5B5;
font-size: 11px;
font-weight: bold;
font-family: 'SimSun', '宋体', serif;
}
.login {
color: #B5B5B5;
font-weight: 500;
text-decoration: none;
font-weight: bold;
font-family: 'SimSun', '宋体', serif;
}
.login:hover {
color: #1d4ed8;
}
/* 右侧知识图谱可视化区域样式 */
.graph-container {
width: 75%;
background: linear-gradient(135deg, #1e3a8a 0%, #1e40af 100%),
url('@/assets/背景.png');
background-size: cover;
background-position: center;
background-blend-mode: overlay;
padding: 2rem;
position: relative;
overflow: hidden;
}
.background-decoration {
position: absolute;
inset: 0;
opacity: 0.2;
}
.bg-circle {
position: absolute;
border-radius: 50%;
filter: blur(3rem);
}
.circle-1 {
top: 25%;
left: 25%;
width: 16rem;
height: 16rem;
background-color: #60a5fa;
}
.circle-2 {
bottom: 33%;
right: 33%;
width: 20rem;
height: 20rem;
background-color: #818cf8;
}
.graph-content {
position: relative;
z-index: 10;
display: flex;
align-items: center;
justify-content: center;
height: 100%;
font-family: 'SimSun', '宋体', serif;
}
.graph-wrapper {
position: relative;
width: 100%;
max-width: 48rem;
aspect-ratio: 1 / 1;
}
.error-text {
color: #ef4444;
font-size: 9px;
margin-top: 5px;
}
/* 响应式设计 */
@media (max-width: 768px) {
.register-container {
flex-direction: column;
}
.register-form-container,
.graph-container {
width: 100%;
}
.register-form-container {
padding: 2rem;
}
.graph-container {
min-height: 400px;
}
}
</style>
Loading…
Cancel
Save