from flask import Flask, render_template, request, redirect, url_for, session, jsonify
from datetime import datetime, timedelta
import os
from flask import Flask, render_template, request, redirect, url_for, session

# 导入用户模块
from user import bp as user_bp, init_user_db
# 导入API模块
from api import api_bp
# 导入数据库工具
from db_utils import get_db_connection, close_db_connection, close_global_connection

app = Flask(__name__)
app.secret_key = 'your_secret_key'  # 生产环境中应该使用更安全的密钥

# 注册用户蓝图
app.register_blueprint(user_bp)
# 注册API蓝图
app.register_blueprint(api_bp)

# 确保模板文件夹存在
templates_dir = os.path.join(os.path.dirname(__file__), 'templates')
if not os.path.exists(templates_dir):
    os.makedirs(templates_dir)

# 初始化数据库
def init_db():
    # 初始化用户相关的数据库
    init_user_db()
    
    conn = get_db_connection()
    cursor = conn.cursor()
    
    # 创建车辆信息表
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS car_owners (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        plate_number TEXT NOT NULL UNIQUE,
        phone TEXT NOT NULL,
        id_card TEXT NOT NULL,
        name TEXT NOT NULL,
        email TEXT,
        address TEXT
    )
    ''')
    
    # 创建查询历史表
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS query_history (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        department TEXT NOT NULL,
        query_date TEXT NOT NULL,
        plate_number TEXT,
        phone TEXT,
        results_count INTEGER NOT NULL,
        query_ip TEXT NOT NULL
    )
    ''')
    
    try:
        conn.commit()
    except Exception as e:
        print(f"提交数据库更改时出错: {e}")
    finally:
        close_db_connection(conn)

# 首页重定向到登录页
@app.route('/')
def index():
    return redirect(url_for('user.login'))

# 查询路由
@app.route('/query', methods=['GET', 'POST'])
def query():
    if 'username' not in session:
        return redirect(url_for('user.login'))
    
    # 获取最近查询历史
    formatted_recent_queries = get_recent_queries()
    
    if request.method == 'POST':
        plate_number = request.form.get('plate_number', '').strip()
        phone = request.form.get('phone', '').strip()
        
        if not plate_number and not phone:
            return render_template('query.html', error='请至少输入车牌号或手机号', recent_queries=formatted_recent_queries)
        
        conn = get_db_connection()
        cursor = conn.cursor()
        
        query_date = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
        
        if plate_number and phone:
            cursor.execute("SELECT * FROM car_owners WHERE plate_number = ? AND phone = ?", (plate_number, phone))
        elif plate_number:
            cursor.execute("SELECT * FROM car_owners WHERE plate_number = ?", (plate_number,))
        else:
            cursor.execute("SELECT * FROM car_owners WHERE phone = ?", (phone,))
        
        results = cursor.fetchall()
        
        # 获取客户端IP地址
        query_ip = request.remote_addr
        
        # 从用户信息表获取用户ID
        user_id = None
        try:
            # 可以复用同一个连接，不需要创建新连接
            cursor.execute("SELECT id FROM users WHERE username = ?", (session['username'],))
            user_info = cursor.fetchone()
            if user_info:
                user_id = user_info[0]
        except Exception as e:
            print(f"获取用户信息时出错: {e}")
        
        # 记录查询历史到数据库
        try:
            # 继续使用同一个连接
            # 检查是否是从"增加"按钮提交的请求
            is_add_query = request.form.get('is_add_query', 'false').lower() == 'true'
            
            # 如果是增加按钮提交的请求，将results_count设置为-1
            if is_add_query:
                results_count = -1
            else:
                results_count = len(results)
            
            cursor.execute(
                "INSERT INTO query_history (query_date, plate_number, phone, results_count, query_ip, user_id) VALUES (?, ?, ?, ?, ?, ?)",
                (query_date, plate_number if plate_number else None, phone if phone else None, results_count, query_ip, user_id)
            )
            conn.commit()
        except Exception as e:
            print(f"记录查询历史时出错: {e}")
        finally:
            close_db_connection(conn)
        
        # 处理查询结果
        car_owners = []
        for result in results:
            car_owners.append({
                'id': result[0],
                'plate_number': result[1],
                'phone': result[2],
                'id_card': result[3],
                'name': result[4],
                'email': result[5],
                'address': result[6]
            })
        
        # 构建查询历史信息
        query_history = {
            'date': query_date,
            'plate_number': plate_number,
            'phone': phone,
            'results_count': len(car_owners)
        }
        
        # 使用辅助函数获取更新后的最近查询历史
        formatted_recent_queries = get_recent_queries()
        
        return render_template('query.html', query_history=query_history, car_owners=car_owners, recent_queries=formatted_recent_queries)
    
    return render_template('query.html', recent_queries=formatted_recent_queries)

# 获取查询历史详情的路由
@app.route('/query_history_detail/<int:history_id>')
def query_history_detail(history_id):
    if 'username' not in session:
        return jsonify({'error': '未登录'}), 401
    
    conn = get_db_connection()
    cursor = conn.cursor()
    
    try:
        # 获取查询历史记录
        cursor.execute("SELECT plate_number, phone FROM query_history WHERE id = ?", (history_id,))
        history = cursor.fetchone()
        
        if not history:
            return jsonify({'error': '查询历史不存在'}), 404
        
        plate_number, phone = history
        
        # 根据车牌号和手机号查询车辆信息
        if plate_number and phone:
            cursor.execute("SELECT * FROM car_owners WHERE plate_number = ? AND phone = ?", (plate_number, phone))
        elif plate_number:
            cursor.execute("SELECT * FROM car_owners WHERE plate_number = ?", (plate_number,))
        else:
            cursor.execute("SELECT * FROM car_owners WHERE phone = ?", (phone,))
        
        results = cursor.fetchall()
        
        # 格式化结果
        car_owners = []
        for result in results:
            car_owners.append({
                'id': result[0],
                'plate_number': result[1],
                'phone': result[2],
                'id_card': result[3],
                'name': result[4],
                'email': result[5],
                'address': result[6]
            })
        
        return jsonify({'car_owners': car_owners})
    except Exception as e:
        print(f"查询历史详情时出错: {e}")
        return jsonify({'error': '查询失败'}), 500
    finally:
        close_db_connection(conn)

# 获取最近查询历史的辅助函数
def get_recent_queries():
    # 获取最近20条查询历史记录，按查询日期降序排列
    conn = get_db_connection()
    cursor = conn.cursor()
    
    try:
        # 使用JOIN操作获取用户姓名和部门信息
        cursor.execute("""
            SELECT qh.id, qh.query_date, qh.plate_number, qh.phone, qh.results_count, qh.query_ip, qh.user_id,
                   u.name AS user_name, u.department AS user_department
            FROM query_history qh
            LEFT JOIN users u ON qh.user_id = u.id
            ORDER BY qh.query_date DESC LIMIT 20
        """)
        recent_queries = cursor.fetchall()
        
        # 格式化查询历史数据
        formatted_recent_queries = []
        for q in recent_queries:
            formatted_recent_queries.append({
                'id': q[0],
                'query_date': q[1],
                'plate_number': q[2] if q[2] else '未提供',
                'phone': q[3] if q[3] else '未提供',
                'results_count': q[4],
                'query_ip': q[5],
                'user_id': q[6] if len(q) > 6 else None,
                'user_name': q[7] if len(q) > 7 else '未知',
                'user_department': q[8] if len(q) > 8 else '未知'
            })
        
        return formatted_recent_queries
    except Exception as e:
        print(f"获取最近查询历史时出错: {e}")
        return []
    finally:
        close_db_connection(conn)

import signal
import sys

def signal_handler(sig, frame):
    """处理终止信号，确保在程序退出时关闭数据库连接"""
    print('\n接收到终止信号，正在关闭数据库连接...')
    close_global_connection()
    print('数据库连接已关闭')
    sys.exit(0)

if __name__ == '__main__':
    # 注册信号处理程序
    signal.signal(signal.SIGINT, signal_handler)  # 处理Ctrl+C
    signal.signal(signal.SIGTERM, signal_handler)  # 处理终止信号
    
    try:
        init_db()
        app.run(debug=True,host="0.0.0.0",port=99)
    finally:
        # 确保在程序退出时关闭数据库连接
        close_global_connection()