import json
from flask import Flask, request, jsonify
import sqlalchemy
from sqlalchemy import Table
from sqlalchemy.engine import create_engine
from sqlalchemy.orm import sessionmaker, scoped_session, declarative_base

# 创建 SQLAlchemy 基类，用于映射表结构
Base = declarative_base()

# 创建 Flask 应用实例
app = Flask(__name__)

# 创建数据库连接
engine = create_engine("mysql+pymysql://root:123456@172.27.13.88:3306/mumushouji", echo=True)

# 创建会话工厂
session = sessionmaker(engine)

# 使用 scoped_session 确保线程安全的会话管理
db_session = scoped_session(session)

# 定义 User 类，映射到数据库中的 USER 表
class User(Base):
    __table__ = Table("USER", Base.metadata, autoload_with=engine)  # 自动加载表结构

@app.route("/login", methods=["POST"])  # 建议仅保留 POST 方法
def login():
    """
    处理用户登录请求。
    
    请求方法：POST
    请求体：JSON 格式，包含 "username" 和 "password" 字段
    返回值：JSON 格式的响应，包含成功或错误信息
    """
    try:
        # 获取并解析请求体中的 JSON 数据
        request_data = request.get_json()
        if not request_data:
            return jsonify({"error": "Invalid or empty JSON"}), 400  # 如果 JSON 数据无效或为空，返回 400 错误

        # 检查必填字段是否存在于请求数据中
        required_fields = ["username", "password"]
        for field in required_fields:
            if field not in request_data:
                return jsonify({"error": f"Missing field: {field}"}), 400  # 缺少必填字段，返回 400 错误

        # 提取用户名和密码
        user_input_name = request_data["username"]
        input_password = request_data["password"]
        print(user_input_name, input_password)  # 打印用户名和密码（仅用于调试）

        # 查询用户是否存在
        result = db_session.query(User).filter_by(username=user_input_name).first()
        if not result:
            return jsonify({"error": "User not found"}), 404  # 用户不存在，返回 404 错误

        # 验证密码（假设密码是明文存储，实际应使用哈希）
        if result.password != input_password:
            return jsonify({"error": "Invalid password"}), 401  # 密码不匹配，返回 401 错误

        # 登录成功，返回成功消息
        return jsonify({"message": "登录成功"})

    except Exception as e:
        # 捕获所有异常，并返回服务器内部错误
        return jsonify({"error": f"Internal server error: {str(e)}"}), 500

if __name__ == '__main__':
    app.run(debug=True)  # 启动 Flask 应用，启用调试模式方便开发

from sqlalchemy import and_

# 查询操作
def query_users():
    # 查询 User 表中的所有记录
    users = db_session.query(User).all()
    for user in users:
        print(f"User ID: {user.user_id}, Username: {user.username}")

    # 根据条件查询，例如 username 等于 'test_user'
    user = db_session.query(User).filter_by(username='test_user').first()
    if user:
        print(f"Found user: {user.username}, Nickname: {user.nickname}")

    # 统计 user_id 小于 10 的用户数量
    count = db_session.query(User).filter(User.user_id < 10).count()
    print(f"Number of users with user_id < 10: {count}")

# 插入操作
def insert_user():
    new_user = User(username='new_username', password='secure_password', nickname='Newbie', picture='http://example.com/pic.png')
    db_session.add(new_user)
    db_session.commit()
    print("Inserted new user.")

# 更新操作
def update_user():
    # 使用 filter_by 进行更新
    user = db_session.query(User).filter_by(username='user_to_update').first()
    if user:
        user.nickname = 'UpdatedNickname'
        db_session.commit()
        print("User updated.")
    
    # 或者直接使用 update 方法批量更新
    db_session.query(User).filter_by(username='another_user_to_update').update({"nickname": "AnotherUpdatedNickname"})
    db_session.commit()
    print("Users batch updated.")

# 删除操作
def delete_user():
    # 删除单个实体
    user = db_session.query(User).filter_by(username='username_to_delete').first()
    if user:
        db_session.delete(user)
        db_session.commit()
        print("User deleted.")
    
    # 批量删除
    db_session.query(User).filter(User.username.like('delete_me_%')).delete()
    db_session.commit()
    print("Batch deletion completed.")

if __name__ == "__main__":
    try:
        # 执行查询
        query_users()

        # 执行插入
        insert_user()

        # 执行更新
        update_user()

        # 执行删除
        delete_user()
        
    except Exception as e:
        db_session.rollback()  # 如果发生错误则回滚
        print(f"An error occurred: {e}")
    finally:
        db_session.close()  # 确保关闭数据库会话