import pymysql

# MySQL 配置
db_config = {
    'host': '192.168.2.32',
    'user': 'root',
    'password': 'lw123456',
    'database': 'message_archive'
}


# 获取消息列表，按时间倒序
def get_messages(page=1, per_page=10):
    try:
        connection = pymysql.connect(**db_config)
        with connection.cursor() as cursor:
            offset = (page - 1) * per_page
            cursor.execute(
                "SELECT id, group_id, timestamp, sender, content FROM messages ORDER BY timestamp DESC LIMIT %s OFFSET %s",
                (per_page, offset))
            messages = cursor.fetchall()
            return messages
    except pymysql.MySQLError as e:
        print(f"数据库查询失败: {e}")
        return []
    finally:
        connection.close()


# 获取消息总数
def get_total_messages():
    try:
        connection = pymysql.connect(**db_config)
        with connection.cursor() as cursor:
            cursor.execute("SELECT COUNT(*) FROM messages")
            total = cursor.fetchone()[0]
            return total
    except pymysql.MySQLError as e:
        print(f"数据库查询失败: {e}")
        return 0
    finally:
        connection.close()