162 lines
4.6 KiB
Python
162 lines
4.6 KiB
Python
# 假设有一个数据库辅助类，你需要根据实际环境实现
|
|
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 = []
|