# 假设有一个数据库辅助类，你需要根据实际环境实现
import json
import pymysql


class DB:
    def __init__(self, host, user, password, database, port=3306, charset="utf8mb4"):
        """
        初始化数据库连接
        """
        self.connection = pymysql.connect(
            host=host,
            user=user,
            password=password,
            database=database,
            port=port,
            charset=charset,
            cursorclass=pymysql.cursors.DictCursor,
        )
        self.last_insert_id = None

    def _execute(self, sql, params=None):
        """
        内部方法：执行SQL并返回cursor
        """
        try:
            with self.connection.cursor() as cursor:
                cursor.execute(sql, params or ())
                return cursor
        except Exception as e:
            print(f"SQL执行错误: {e}\nSQL: {sql}")
            raise

    def getOne(self, sql):
        # 模拟查询一行
        cursor = self._execute(sql)
        return cursor.fetchone()

    def getone(self, sql):
        # PHP代码中混用了getOne和getone，通常功能一致
        return self.getOne(sql)

    def query(self, sql):
        """
        执行SQL (通常用于无返回结果的操作，如UPDATE, DELETE, CREATE等)
        """
        try:
            with self.connection.cursor() as cursor:
                result = cursor.execute(sql)
                self.connection.commit()
                # 记录最后插入的ID (如果是INSERT操作)
                self.last_insert_id = cursor.lastrowid
                return result
        except Exception as e:
            self.connection.rollback()
            print(f"Query执行失败: {e}\nSQL: {sql}")
            raise

    def query_array(self, sql):
        """
        查询多行
        """
        cursor = self._execute(sql)
        return cursor.fetchall()

    def insert_array(self, data, table):
        """
        插入数据
        :param data: 字典，例如 [{'name': 'Alice', 'age': 25}]
        :param table: 表名
        """
        if not data:
            return False

        columns = ", ".join(data[0].keys())
        # 使用占位符防止SQL注入
        placeholders = ", ".join(["%s"] * len(data[0]))
        values = [tuple(item.values()) for item in data]

        sql = f"INSERT INTO {table} ({columns}) VALUES ({placeholders})"

        try:
            with self.connection.cursor() as cursor:
                cursor.executemany(sql, values)
                self.connection.commit()
                self.last_insert_id = cursor.lastrowid
                return self.last_insert_id
        except Exception as e:
            self.connection.rollback()
            print(f"插入数据失败: {e}")
            raise

    def insert_id(self):
        """
        获取最后插入ID
        """
        return self.last_insert_id

    def close(self):
        """
        关闭数据库连接
        """
        if self.connection:
            self.connection.close()


db = DB(host="localhost", user="fire", password="iTSJSPPZM3LSGAPC", database="fire")

file_name = "simulator_log_super_1776010009.7211218.log"


with open(file_name, "r") as f:
    type = ""
    if "super" in file_name:
        type = "_super"
    elif "normal" in file_name:
        type = ""
    elif "standard" in file_name:
        type = "_free"

    # db.query("truncate table game_sugar_spin{type}".format(type=type))
    # db.query("truncate table game_sugar_step_info{type}".format(type=type))

    spins = []
    steps_data = []
    index = 0
    startGid = 10000
    for line in f.readlines():
        index += 1
        data = json.loads(line)

        data["gid"] = startGid + data['gid']

        steps = data["steps"]
        for step in steps:
            step["gid"] = data["gid"]
            step["symbol_links"] = json.dumps(
                step["symbol_links"], separators=(",", ":"), ensure_ascii=False
            )
            step["multipler"] = json.dumps(
                step["multipler"], separators=(",", ":"), ensure_ascii=False
            )
            steps_data.append(step)

        del data["steps"]
        spins.append(data)

        if len(spins) > 100:
            db.insert_array(spins, f"game_sugar_spin{type}")
            spins = []
            print(f"写入进度: {index}个spin")
        if len(steps_data) > 300:
            db.insert_array(steps_data, f"game_sugar_step_info{type}")
            steps_data = []

    if len(spins) > 0:
        db.insert_array(spins, f"game_sugar_spin{type}")
        spins = []
    if len(steps_data) > 0:
        db.insert_array(steps_data, f"game_sugar_step_info{type}")
        steps_data = []