From bf49c39d983e8abc3fd1f77768d57b436954e6a1 Mon Sep 17 00:00:00 2001 From: Lxy Date: Sun, 15 Feb 2026 12:03:22 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E7=99=BB=E9=99=86?= =?UTF-8?q?=E6=B3=A8=E5=86=8C=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- web/auth.py | 83 +++++++++++++++ web/templates/login.html | 198 ++++++++++++++++++++++++++++++++++++ web/templates/register.html | 198 ++++++++++++++++++++++++++++++++++++ 3 files changed, 479 insertions(+) create mode 100644 web/auth.py create mode 100644 web/templates/login.html create mode 100644 web/templates/register.html diff --git a/web/auth.py b/web/auth.py new file mode 100644 index 0000000..3303305 --- /dev/null +++ b/web/auth.py @@ -0,0 +1,83 @@ +# 认证模块 + +from flask import Flask, request, redirect, url_for, flash +from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user, current_user +from werkzeug.security import generate_password_hash, check_password_hash +import sqlite3 +import os + +# 初始化Flask-Login +login_manager = LoginManager() + +class User(UserMixin): + def __init__(self, id, username, password_hash): + self.id = id + self.username = username + self.password_hash = password_hash + + def check_password(self, password): + return check_password_hash(self.password_hash, password) + +@login_manager.user_loader +def load_user(user_id): + """根据用户ID加载用户对象""" + conn = get_db_connection() + user = conn.execute('SELECT id, username, password_hash FROM users WHERE id = ?', (user_id,)).fetchone() + conn.close() + if user: + return User(user['id'], user['username'], user['password_hash']) + return None + +def get_db_connection(): + """获取数据库连接""" + db_path = os.path.join(os.path.dirname(__file__), 'data', 'futures_analysis.db') + conn = sqlite3.connect(db_path) + conn.row_factory = sqlite3.Row + return conn + +def init_db(): + """初始化数据库，创建用户表""" + conn = get_db_connection() + conn.execute(''' + CREATE TABLE IF NOT EXISTS users ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + username TEXT UNIQUE NOT NULL, + password_hash TEXT NOT NULL + ) + ''') + conn.commit() + conn.close() + +def register_user(username, password): + """注册新用户""" + conn = get_db_connection() + try: + # 检查用户名是否已存在 + existing_user = conn.execute('SELECT id FROM users WHERE username = ?', (username,)).fetchone() + if existing_user: + return False, "用户名已存在" + + # 生成密码哈希 + password_hash = generate_password_hash(password) + + # 插入新用户 + conn.execute('INSERT INTO users (username, password_hash) VALUES (?, ?)', (username, password_hash)) + conn.commit() + return True, "注册成功" + except Exception as e: + conn.rollback() + return False, str(e) + finally: + conn.close() + +def login_user_by_credentials(username, password): + """通过用户名和密码登录用户""" + conn = get_db_connection() + user = conn.execute('SELECT id, username, password_hash FROM users WHERE username = ?', (username,)).fetchone() + conn.close() + + if user and check_password_hash(user['password_hash'], password): + user_obj = User(user['id'], user['username'], user['password_hash']) + login_user(user_obj) + return True, "登录成功" + return False, "用户名或密码错误" diff --git a/web/templates/login.html b/web/templates/login.html new file mode 100644 index 0000000..4458ba4 --- /dev/null +++ b/web/templates/login.html @@ -0,0 +1,198 @@ + + + + + + 登录 - AI期货分析系统 + + + + +
+
+

登录

+

欢迎使用AI期货分析系统

+
+ + + {% with messages = get_flashed_messages() %} + {% if messages %} + {% for message in messages %} +
+ {{ message }} +
+ {% endfor %} + {% endif %} + {% endwith %} + +
+
+ + +
+
+ + +
+ +
+ +
+

还没有账号？ 立即注册

+
+
+ + \ No newline at end of file diff --git a/web/templates/register.html b/web/templates/register.html new file mode 100644 index 0000000..95094c0 --- /dev/null +++ b/web/templates/register.html @@ -0,0 +1,198 @@ + + + + + + 注册 - AI期货分析系统 + + + + +
+
+

注册

+

创建一个新账号以使用AI期货分析系统

+
+ + + {% with messages = get_flashed_messages() %} + {% if messages %} + {% for message in messages %} +
+ {{ message }} +
+ {% endfor %} + {% endif %} + {% endwith %} + +
+
+ + +
+
+ + +
+ +
+ +
+

已有账号？ 立即登录

+
+
+ + \ No newline at end of file