269 lines
12 KiB
Python
269 lines
12 KiB
Python
import sqlite3
|
||
import tkinter as tk
|
||
from tkinter import ttk
|
||
import os
|
||
import json
|
||
|
||
|
||
class TeamSwitchPacketGenerator:
|
||
def __init__(self):
|
||
# 检查数据库文件是否存在
|
||
db_file = 'aola.sqlite'
|
||
if not os.path.exists(db_file):
|
||
raise FileNotFoundError(f"数据库文件 {db_file} 未找到，请确保文件在 {os.getcwd()} 目录下！")
|
||
|
||
print(f"数据库文件路径: {os.path.abspath(db_file)}")
|
||
self.conn = sqlite3.connect(db_file)
|
||
self.cursor = self.conn.cursor()
|
||
self.root = tk.Tk()
|
||
self.root.title("阵容切换封包生成器")
|
||
|
||
# 获取所有阵容名称和 spell_code
|
||
self.team_data = self.get_team_data() # 存储 team_name 和 spell_code 的列表
|
||
self.team_names = [item[0] for item in self.team_data] # 仅 team_name 列表用于显示
|
||
|
||
# 创建搜索框（替换下拉框）
|
||
tk.Label(self.root, text="搜索阵容（输入拼音缩写）：").pack(anchor=tk.W, padx=5)
|
||
self.search_var = tk.StringVar()
|
||
self.search_var.trace("w", self.update_combobox) # 监听输入变化
|
||
self.search_entry = tk.Entry(self.root, textvariable=self.search_var)
|
||
self.search_entry.pack(pady=5, padx=5, fill=tk.X)
|
||
|
||
# 创建可搜索的下拉框
|
||
self.team_var = tk.StringVar()
|
||
self.team_dropdown = ttk.Combobox(self.root, textvariable=self.team_var)
|
||
self.team_dropdown['values'] = self.team_names
|
||
self.team_dropdown.pack(pady=10)
|
||
self.team_dropdown.set("请选择阵容")
|
||
|
||
# 创建按钮框架，包含“刷新”和“生成封包”按钮
|
||
button_frame = tk.Frame(self.root)
|
||
button_frame.pack(pady=5)
|
||
|
||
# 刷新按钮
|
||
tk.Button(button_frame, text="刷新", command=self.refresh_dropdown).pack(side=tk.LEFT, padx=5)
|
||
|
||
# 生成按钮
|
||
tk.Button(button_frame, text="生成封包（第一次生成请先解锁二级密码）", command=self.generate_packets).pack(side=tk.LEFT, padx=5)
|
||
|
||
# 创建 Frame 包含封包输出框和复制按钮
|
||
output_frame = tk.Frame(self.root)
|
||
output_frame.pack(fill=tk.BOTH, expand=True, padx=5, pady=5)
|
||
|
||
# 封包输出框
|
||
tk.Label(output_frame, text="封包输出：").pack(anchor=tk.W)
|
||
self.output_text = tk.Text(output_frame, height=15, width=60)
|
||
self.output_text.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
|
||
|
||
# 复制按钮（放在右侧）
|
||
tk.Button(output_frame, text="复制脚本", command=self.copy_output).pack(side=tk.RIGHT, padx=5)
|
||
|
||
# 信息提示框
|
||
tk.Label(self.root, text="提示信息：").pack(anchor=tk.W)
|
||
self.info_text = tk.Text(self.root, height=5, width=80, fg="red")
|
||
self.info_text.pack(pady=5)
|
||
|
||
def get_team_data(self):
|
||
"""从数据库获取所有阵容名称和 spell_code"""
|
||
try:
|
||
self.cursor.execute("SELECT team_name, spell_code FROM aolaer_team ORDER BY team_name")
|
||
return self.cursor.fetchall() # 返回 [(team_name, spell_code), ...]
|
||
except sqlite3.Error as e:
|
||
self.info_text.insert(tk.END, f"获取阵容数据失败: {e}\n")
|
||
return []
|
||
|
||
def update_combobox(self, *args):
|
||
"""根据搜索框输入更新下拉框内容"""
|
||
search_text = self.search_var.get().strip().lower()
|
||
if not search_text:
|
||
# 如果搜索框为空，显示所有阵容
|
||
self.team_dropdown['values'] = self.team_names
|
||
self.team_dropdown.set("请选择阵容")
|
||
else:
|
||
# 筛选 spell_code 包含搜索文本的阵容
|
||
filtered_teams = [
|
||
team_name for team_name, spell_code in self.team_data
|
||
if spell_code and search_text in spell_code.lower()
|
||
]
|
||
self.team_dropdown['values'] = filtered_teams
|
||
if filtered_teams:
|
||
self.team_dropdown.set(filtered_teams[0]) # 默认选中第一个匹配项
|
||
else:
|
||
self.team_dropdown.set("无匹配阵容")
|
||
self.team_dropdown['values'] = []
|
||
|
||
def refresh_dropdown(self):
|
||
"""刷新下拉框数据"""
|
||
# 重新获取阵容数据
|
||
self.team_data = self.get_team_data()
|
||
self.team_names = [item[0] for item in self.team_data]
|
||
# 更新下拉框选项
|
||
self.team_dropdown['values'] = self.team_names
|
||
self.team_dropdown.set("请选择阵容")
|
||
# 清空搜索框
|
||
self.search_var.set("")
|
||
# 在提示框中显示刷新成功
|
||
self.info_text.insert(tk.END, "下拉框数据已刷新！\n")
|
||
|
||
def copy_output(self):
|
||
"""复制脚本输出框的内容到剪贴板"""
|
||
content = self.output_text.get(1.0, tk.END).strip()
|
||
if content:
|
||
self.root.clipboard_clear()
|
||
self.root.clipboard_append(content)
|
||
self.info_text.insert(tk.END, "脚本内容已复制到剪贴板！\n")
|
||
else:
|
||
self.info_text.insert(tk.END, "脚本输出框为空，无法复制！\n")
|
||
|
||
def validate_ids(self, ids_str):
|
||
"""验证 ids 数组，确保格式正确"""
|
||
try:
|
||
ids = json.loads(ids_str)
|
||
if not isinstance(ids, list) or len(ids) != 5:
|
||
return False, "ids 必须是长度为 5 的数组"
|
||
for x in ids:
|
||
if not isinstance(x, int) or x < 0:
|
||
return False, "ids 中的值必须是非负整数"
|
||
return True, ids
|
||
except json.JSONDecodeError:
|
||
return False, "ids 格式不正确"
|
||
|
||
def generate_packets(self):
|
||
"""生成封包代码"""
|
||
self.output_text.delete(1.0, tk.END)
|
||
self.info_text.delete(1.0, tk.END)
|
||
packets = []
|
||
|
||
selected_team = self.team_var.get()
|
||
if not selected_team or selected_team in ["请选择阵容", "无匹配阵容"]:
|
||
self.info_text.insert(tk.END, "请先选择一个有效阵容！\n")
|
||
return
|
||
|
||
try:
|
||
# 获取 team_code 和 team_set
|
||
self.cursor.execute("SELECT team_code, team_set FROM aolaer_team WHERE team_name = ?", (selected_team,))
|
||
result = self.cursor.fetchone()
|
||
if not result:
|
||
self.info_text.insert(tk.END, f"未找到阵容 {selected_team} 的 team_code！\n")
|
||
return
|
||
team_code, team_set = result
|
||
packets.append(f'|#send={{"id":13,"param":{{"pms":"{team_code}"}},"cmd":"1222"}}|')
|
||
|
||
# 处理 is_set = 1 的记录（重置魂卡）
|
||
self.cursor.execute("SELECT id FROM aolaer_id WHERE is_set = 1")
|
||
set_ids = [row[0] for row in self.cursor.fetchall()]
|
||
|
||
for set_id in set_ids:
|
||
packets.append(
|
||
f'|#send={{"id":42,"param":{{"bk":1,"petId":{set_id},"ids":[0,0,0,0,0]}},"cmd":"ASC221104_2"}}|')
|
||
self.cursor.execute("UPDATE aolaer_id SET is_set = 0 WHERE id = ?", (set_id,))
|
||
self.conn.commit()
|
||
|
||
# 处理 team_code 中的 ID
|
||
team_ids = team_code.split('#')
|
||
team_ids = [id for id in team_ids if id]
|
||
|
||
# 处理 team_set
|
||
team_set_codes = None
|
||
if team_set:
|
||
team_set_codes = team_set.split('#')
|
||
if len(team_set_codes) != len(team_ids):
|
||
self.info_text.insert(tk.END,
|
||
f"警告：team_set 的长度 ({len(team_set_codes)}) 与 team_code 的 ID 数量 ({len(team_ids)}) 不匹配！\n")
|
||
return
|
||
|
||
# 查询 aolaer_id 表
|
||
self.cursor.execute(
|
||
"SELECT id, div_weapon, aolaer_typecode FROM aolaer_id WHERE id IN ({})".format(
|
||
','.join('?' for _ in team_ids)
|
||
),
|
||
team_ids
|
||
)
|
||
|
||
id_to_typecode = {row[0]: row for row in self.cursor.fetchall()}
|
||
|
||
# 分组生成封包，确保顺序：子宠物 → 魂艺 → 魂卡
|
||
sub_pet_packets = []
|
||
soul_art_packets = []
|
||
soul_card_packets = []
|
||
|
||
for idx, id in enumerate(team_ids):
|
||
id = int(id)
|
||
if id not in id_to_typecode:
|
||
self.info_text.insert(tk.END, f"警告：ID {id} 在 aolaer_id 表中不存在！\n")
|
||
continue
|
||
|
||
div_weapon = id_to_typecode[id][1]
|
||
typecode = id_to_typecode[id][2]
|
||
if team_set_codes and team_set_codes[idx] != '0':
|
||
typecode = team_set_codes[idx]
|
||
|
||
# 生成子宠物封包（仅当 div_weapon 非空时）
|
||
if div_weapon and str(div_weapon).strip():
|
||
sub_pet_packets.append(
|
||
f'|#send={{"id":42,"param":{{"subPetId":{id},"petId":{div_weapon}}},"cmd":"ASBS230623_con"}}|')
|
||
else:
|
||
self.info_text.insert(tk.END, f"提示：ID {id} 的 div_weapon 为空，跳过生成子宠物封包！\n")
|
||
|
||
# 获取 soul_art 和 soul_card
|
||
self.cursor.execute("SELECT soul_art, soul_card FROM aolaer_type WHERE aolaer_typecode = ?",
|
||
(typecode,))
|
||
result = self.cursor.fetchone()
|
||
if not result:
|
||
self.info_text.insert(tk.END, f"警告：typecode {typecode} 在 aolaer_type 表中不存在！\n")
|
||
continue
|
||
soul_art, soul_card = result
|
||
|
||
# 生成魂艺封包（仅当 soul_art 非空时）
|
||
if soul_art and soul_art.strip():
|
||
try:
|
||
sid1, sid2 = soul_art.split(',')
|
||
soul_art_packets.append(
|
||
f'|#send={{"id":42,"param":{{"ui":{sid1},"petId":{id}}},"cmd":"ATT231229_1"}}|')
|
||
soul_art_packets.append(
|
||
f'|#send={{"id":42,"param":{{"ui":{sid2},"petId":{id}}},"cmd":"ATT231229_1"}}|')
|
||
except ValueError:
|
||
self.info_text.insert(tk.END, f"错误：soul_art 格式不正确 for typecode {typecode}！\n")
|
||
continue
|
||
else:
|
||
self.info_text.insert(tk.END,
|
||
f"提示：typecode {typecode} 的 soul_art 为空，跳过生成魂艺封包！\n")
|
||
|
||
# 生成魂卡封包
|
||
if soul_card and soul_card.strip():
|
||
is_valid, ids_or_error = self.validate_ids(soul_card)
|
||
if is_valid:
|
||
soul_card_packets.append(
|
||
f'|#send={{"id":42,"param":{{"bk":1,"petId":{id},"ids":{soul_card}}},"cmd":"ASC221104_2"}}|')
|
||
self.cursor.execute("UPDATE aolaer_id SET is_set = 1 WHERE id = ?", (id,))
|
||
self.conn.commit()
|
||
else:
|
||
self.info_text.insert(tk.END,
|
||
f"警告：typecode {typecode} 的 soul_card {soul_card} 无效（{ids_or_error}），已跳过处理！\n")
|
||
else:
|
||
self.info_text.insert(tk.END, f"提示：typecode {typecode} 的 soul_card 为空，已跳过处理！\n")
|
||
|
||
# 按顺序添加封包：子宠物 → 魂艺 → 魂卡
|
||
packets.extend(sub_pet_packets)
|
||
packets.extend(soul_art_packets)
|
||
packets.extend(soul_card_packets)
|
||
|
||
if packets:
|
||
self.output_text.insert(tk.END, '\n'.join(packets))
|
||
else:
|
||
self.info_text.insert(tk.END, "未生成任何封包，请检查数据！\n")
|
||
except sqlite3.Error as e:
|
||
self.info_text.insert(tk.END, f"数据库操作失败: {e}\n")
|
||
|
||
def run(self):
|
||
"""运行程序"""
|
||
self.root.mainloop()
|
||
|
||
def __del__(self):
|
||
"""清理数据库连接"""
|
||
self.conn.close()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
app = TeamSwitchPacketGenerator()
|
||
app.run() |