menu.py

import json
from multiprocessing import Pipe, Process, freeze_support
import time
import PySimpleGUI as sg
import os
from ruamel.yaml import YAML

from arknights_mower.utils.conf import load_conf, save_conf, load_plan, write_plan
from arknights_mower.utils.log import logger
from arknights_mower.data import agent_list
from arknights_mower.__main__ import main
from arknights_mower.__init__ import __version__
from arknights_mower.utils.update import compere_version , update_version,download_version

yaml = YAML()
# confUrl = './conf.yml'
conf = {}
plan = {}
current_plan = {}
operators = {}
global window
buffer = ''
line = 0
half_line_index = 0


def build_plan(url):
    global plan
    global current_plan
    try:
        plan = load_plan(url)
        current_plan = plan[plan['default']]
        conf['planFile'] = url
        for i in range(1, 4):
            for j in range(1, 4):
                window[f'btn_room_{str(i)}_{str(j)}'].update('待建造', button_color=('white', '#4f4945'))
        for key in current_plan:
            if type(current_plan[key]).__name__ == 'list':  # 兼容旧版格式
                current_plan[key] = {'plans': current_plan[key], 'name': ''}
            elif current_plan[key]['name'] == '贸易站':
                window['btn_' + key].update('贸易站', button_color=('#4f4945', '#33ccff'))
            elif current_plan[key]['name'] == '制造站':
                window['btn_' + key].update('制造站', button_color=('#4f4945', '#ffcc00'))
            elif current_plan[key]['name'] == '发电站':
                window['btn_' + key].update('发电站', button_color=('#4f4945', '#ccff66'))
        window['plan_radio_ling_xi_' + str(plan['conf']['ling_xi'])].update(True)
        window['plan_int_max_resting_count'].update(plan['conf']['max_resting_count'])
        window['plan_conf_exhaust_require'].update(plan['conf']['exhaust_require'])
        window['plan_conf_workaholic'].update(plan['conf']['workaholic'])
        window['plan_conf_rest_in_full'].update(plan['conf']['rest_in_full'])
        window['plan_conf_resting_priority'].update(plan['conf']['resting_priority'])
    except Exception as e:
        logger.error(e)
        println('json格式错误！')


# 主页面
def menu():
    global window
    global buffer
    global conf
    global plan
    conf = load_conf()
    plan = load_plan(conf['planFile'])
    sg.theme('LightBlue2')
    # --------主页
    package_type_title = sg.Text('服务器:', size=10)
    package_type_1 = sg.Radio('官服', 'package_type', default=conf['package_type'] == 1,
                              key='radio_package_type_1', enable_events=True)
    package_type_2 = sg.Radio('BiliBili服', 'package_type', default=conf['package_type'] == 2,
                              key='radio_package_type_2', enable_events=True)
    adb_title = sg.Text('adb连接地址:', size=10)
    adb = sg.InputText(conf['adb'], size=60, key='conf_adb', enable_events=True)
    # 黑名单
    free_blacklist_title = sg.Text('宿舍黑名单:', size=10)
    free_blacklist = sg.InputText(conf['free_blacklist'], size=60, key='conf_free_blacklist', enable_events=True)
    # 排班表json
    plan_title = sg.Text('排班表:', size=10)
    plan_file = sg.InputText(conf['planFile'], readonly=True, size=60, key='planFile', enable_events=True)
    plan_select = sg.FileBrowse('...', size=(3, 1), file_types=(("JSON files", "*.json"),))
    # 总开关
    on_btn = sg.Button('开始执行', key='on')
    off_btn = sg.Button('立即停止', key='off', visible=False, button_color='red')
    # 日志栏
    output = sg.Output(size=(150, 25), key='log', text_color='#808069', font=('微软雅黑', 9))

    # --------排班表设置页面
    # 宿舍区
    central = sg.Button('控制中枢', key='btn_central', size=(18, 3), button_color='#303030')
    dormitory_1 = sg.Button('宿舍', key='btn_dormitory_1', size=(18, 2), button_color='#303030')
    dormitory_2 = sg.Button('宿舍', key='btn_dormitory_2', size=(18, 2), button_color='#303030')
    dormitory_3 = sg.Button('宿舍', key='btn_dormitory_3', size=(18, 2), button_color='#303030')
    dormitory_4 = sg.Button('宿舍', key='btn_dormitory_4', size=(18, 2), button_color='#303030')
    central_area = sg.Column([[central], [dormitory_1], [dormitory_2], [dormitory_3], [dormitory_4]])
    # 制造站区
    room_1_1 = sg.Button('待建造', key='btn_room_1_1', size=(12, 2), button_color='#4f4945')
    room_1_2 = sg.Button('待建造', key='btn_room_1_2', size=(12, 2), button_color='#4f4945')
    room_1_3 = sg.Button('待建造', key='btn_room_1_3', size=(12, 2), button_color='#4f4945')
    room_2_1 = sg.Button('待建造', key='btn_room_2_1', size=(12, 2), button_color='#4f4945')
    room_2_2 = sg.Button('待建造', key='btn_room_2_2', size=(12, 2), button_color='#4f4945')
    room_2_3 = sg.Button('待建造', key='btn_room_2_3', size=(12, 2), button_color='#4f4945')
    room_3_1 = sg.Button('待建造', key='btn_room_3_1', size=(12, 2), button_color='#4f4945')
    room_3_2 = sg.Button('待建造', key='btn_room_3_2', size=(12, 2), button_color='#4f4945')
    room_3_3 = sg.Button('待建造', key='btn_room_3_3', size=(12, 2), button_color='#4f4945')
    left_area = sg.Column([[room_1_1, room_1_2, room_1_3],
                           [room_2_1, room_2_2, room_2_3],
                           [room_3_1, room_3_2, room_3_3]])
    # 功能区
    meeting = sg.Button('会客室', key='btn_meeting', size=(24, 2), button_color='#303030')
    factory = sg.Button('加工站', key='btn_factory', size=(24, 2), button_color='#303030')
    contact = sg.Button('办公室', key='btn_contact', size=(24, 2), button_color='#303030')
    right_area = sg.Column([[meeting], [factory], [contact]])

    setting_layout = [
        [sg.Column([[sg.Text('设施类别:'), sg.InputCombo(['贸易站', '制造站', '发电站'], size=12, key='station_type')]],
                   key='station_type_col', visible=False)]]
    # 排班表设置标签
    for i in range(1, 6):
        set_area = sg.Column([[sg.Text('干员：'),
                               sg.Combo(['Free'] + agent_list, size=20, key='agent' + str(i), change_submits=True),
                               sg.Text('组：'),
                               sg.InputText('', size=15, key='group' + str(i)),
                               sg.Text('替换：'),
                               sg.InputText('', size=30, key='replacement' + str(i))
                               ]], key='setArea' + str(i), visible=False)
        setting_layout.append([set_area])
    setting_layout.append(
        [sg.Button('保存', key='savePlan', visible=False), sg.Button('清空', key='clearPlan', visible=False)])
    setting_area = sg.Column(setting_layout, element_justification="center",
                             vertical_alignment="bottom",
                             expand_x=True)

    # --------高级设置页面
    current_version = sg.Text('当前版本：' + __version__, size=25)
    btn_check_update = sg.Button('检测更新', key='check_update')
    update_msg = sg.Text('', key='update_msg')
    run_mode_title = sg.Text('运行模式：', size=25)
    run_mode_1 = sg.Radio('换班模式', 'run_mode', default=conf['run_mode'] == 1,
                          key='radio_run_mode_1', enable_events=True)
    run_mode_2 = sg.Radio('仅跑单模式', 'run_mode', default=conf['run_mode'] == 2,
                          key='radio_run_mode_2', enable_events=True)
    ling_xi_title = sg.Text('令夕模式（令夕上班时起作用）：', size=25)
    ling_xi_1 = sg.Radio('感知信息', 'ling_xi', default=plan['conf']['ling_xi'] == 1,
                         key='plan_radio_ling_xi_1', enable_events=True)
    ling_xi_2 = sg.Radio('人间烟火', 'ling_xi', default=plan['conf']['ling_xi'] == 2,
                         key='plan_radio_ling_xi_2', enable_events=True)
    ling_xi_3 = sg.Radio('均衡模式', 'ling_xi', default=plan['conf']['ling_xi'] == 3,
                         key='plan_radio_ling_xi_3', enable_events=True)
    enable_party_title = sg.Text('线索收集：', size=25)
    enable_party_1 = sg.Radio('启用', 'enable_party', default=conf['enable_party'] == 1,
                              key='radio_enable_party_1', enable_events=True)
    enable_party_0 = sg.Radio('禁用', 'enable_party', default=conf['enable_party'] == 0,
                              key='radio_enable_party_0', enable_events=True)
    max_resting_count_title = sg.Text('最大组人数：', size=25, key='max_resting_count_title')
    max_resting_count = sg.InputText(plan['conf']['max_resting_count'], size=5,
                                     key='plan_int_max_resting_count', enable_events=True)
    drone_count_limit_title = sg.Text('无人机使用阈值：', size=25, key='drone_count_limit_title')
    drone_count_limit = sg.InputText(conf['drone_count_limit'], size=5,
                                     key='int_drone_count_limit', enable_events=True)
    run_order_delay_title = sg.Text('跑单前置延时(分钟)：', size=25, key='run_order_delay_title')
    run_order_delay = sg.InputText(conf['run_order_delay'], size=5,
                                   key='float_run_order_delay', enable_events=True)
    drone_room_title = sg.Text('无人机使用房间（room_X_X）：', size=25, key='drone_room_title')
    reload_room_title = sg.Text('搓玉补货房间（逗号分隔房间名）：', size=25, key='reload_room_title')
    drone_room = sg.InputText(conf['drone_room'], size=15,
                              key='conf_drone_room', enable_events=True)
    reload_room = sg.InputText(conf['reload_room'], size=30,
                               key='conf_reload_room', enable_events=True)
    rest_in_full_title = sg.Text('需要回满心情的干员：', size=25)
    rest_in_full = sg.InputText(plan['conf']['rest_in_full'], size=60,
                                key='plan_conf_rest_in_full', enable_events=True)

    exhaust_require_title = sg.Text('需用尽心情的干员：', size=25)
    exhaust_require = sg.InputText(plan['conf']['exhaust_require'], size=60,
                                   key='plan_conf_exhaust_require', enable_events=True)

    workaholic_title = sg.Text('0心情工作的干员：', size=25)
    workaholic = sg.InputText(plan['conf']['workaholic'], size=60,
                              key='plan_conf_workaholic', enable_events=True)

    resting_priority_title = sg.Text('宿舍低优先级干员：', size=25)
    resting_priority = sg.InputText(plan['conf']['resting_priority'], size=60,
                                    key='plan_conf_resting_priority', enable_events=True)

    start_automatically = sg.Checkbox('启动mower时自动开始任务', default=conf['start_automatically'],
                                      key='conf_start_automatically', enable_events=True)
    # --------外部调用设置页面
    # mail
    mail_enable_1 = sg.Radio('启用', 'mail_enable', default=conf['mail_enable'] == 1,
                             key='radio_mail_enable_1', enable_events=True)
    mail_enable_0 = sg.Radio('禁用', 'mail_enable', default=conf['mail_enable'] == 0,
                             key='radio_mail_enable_0', enable_events=True)
    account_title = sg.Text('QQ邮箱', size=25)
    account = sg.InputText(conf['account'], size=60, key='conf_account', enable_events=True)
    pass_code_title = sg.Text('授权码', size=25)
    pass_code = sg.Input(conf['pass_code'], size=60, key='conf_pass_code', enable_events=True, password_char='*')
    mail_frame = sg.Frame('邮件提醒',
                          [[mail_enable_1, mail_enable_0], [account_title, account], [pass_code_title, pass_code]])
    # maa

    maa_enable_1 = sg.Radio('启用', 'maa_enable', default=conf['maa_enable'] == 1,
                            key='radio_maa_enable_1', enable_events=True)
    maa_enable_0 = sg.Radio('禁用', 'maa_enable', default=conf['maa_enable'] == 0,
                            key='radio_maa_enable_0', enable_events=True)
    maa_gap_title = sg.Text('MAA启动间隔(小时)：', size=15)
    maa_gap = sg.InputText(conf['maa_gap'], size=5, key='float_maa_gap', enable_events=True)
    maa_mall_buy_title = sg.Text('信用商店优先购买（逗号分隔）：', size=25, key='mall_buy_title')
    maa_mall_buy = sg.InputText(conf['maa_mall_buy'], size=30,
                               key='conf_maa_mall_buy', enable_events=True)
    maa_recruitment_time = sg.Checkbox('公招三星设置7:40而非9:00', default=conf['maa_recruitment_time'],
                                      key='conf_maa_recruitment_time', enable_events=True)
    maa_recruit_only_4 = sg.Checkbox('仅公招四星', default=conf['maa_recruit_only_4'],
                                       key='conf_maa_recruit_only_4', enable_events=True)
    maa_mall_blacklist_title = sg.Text('信用商店黑名单（逗号分隔）：', size=25, key='mall_blacklist_title')
    maa_mall_blacklist = sg.InputText(conf['maa_mall_blacklist'], size=30,
                                key='conf_maa_mall_blacklist', enable_events=True)
    maa_rg_title = sg.Text('肉鸽：', size=10)
    maa_rg_enable_1 = sg.Radio('启用', 'maa_rg_enable', default=conf['maa_rg_enable'] == 1,
                               key='radio_maa_rg_enable_1', enable_events=True)
    maa_rg_enable_0 = sg.Radio('禁用', 'maa_rg_enable', default=conf['maa_rg_enable'] == 0,
                               key='radio_maa_rg_enable_0', enable_events=True)
    maa_rg_sleep = sg.Text('肉鸽任务休眠时间(如8:30-23:30)', size=25)
    maa_rg_sleep_min = sg.InputText(conf['maa_rg_sleep_min'], size=5, key='conf_maa_rg_sleep_min', enable_events=True)
    maa_rg_sleep_max = sg.InputText(conf['maa_rg_sleep_max'], size=5, key='conf_maa_rg_sleep_max', enable_events=True)
    maa_path_title = sg.Text('MAA地址', size=25)
    maa_path = sg.InputText(conf['maa_path'], size=60, key='conf_maa_path', enable_events=True)
    maa_adb_path_title = sg.Text('adb地址', size=25)
    maa_adb_path = sg.InputText(conf['maa_adb_path'], size=60, key='conf_maa_adb_path', enable_events=True)
    maa_weekly_plan_title = sg.Text('周计划', size=25)
    maa_layout = [[maa_enable_1, maa_enable_0, maa_gap_title, maa_gap, maa_recruitment_time, maa_recruit_only_4],
                  [maa_mall_buy_title, maa_mall_buy, maa_mall_blacklist_title, maa_mall_blacklist],
                  [maa_rg_title, maa_rg_enable_1, maa_rg_enable_0, maa_rg_sleep, maa_rg_sleep_min, maa_rg_sleep_max],
                  [maa_path_title, maa_path], [maa_adb_path_title, maa_adb_path],
                  [maa_weekly_plan_title]]
    for i, v in enumerate(conf['maa_weekly_plan']):
        maa_layout.append([
            sg.Text(f"-- {v['weekday']}:", size=15),
            sg.Text('关卡:', size=5),
            sg.InputText(",".join(v['stage']), size=15, key='maa_weekly_plan_stage_' + str(i), enable_events=True),
            sg.Text('理智药:', size=10),
            sg.Spin([l for l in range(0, 999)], initial_value=v['medicine'], size=5,
                    key='maa_weekly_plan_medicine_' + str(i), enable_events=True, readonly=True)
        ])

    maa_frame = sg.Frame('MAA', maa_layout)
    # --------组装页面
    main_tab = sg.Tab('  主页  ', [[package_type_title, package_type_1, package_type_2],
                                 [adb_title, adb],
                                 [free_blacklist_title, free_blacklist],
                                 [plan_title, plan_file, plan_select],
                                 [output],
                                 [on_btn, off_btn]])

    plan_tab = sg.Tab('  排班表 ', [[left_area, central_area, right_area], [setting_area]], element_justification="center")
    setting_tab = sg.Tab('  高级设置 ',
                         [
                             [current_version, btn_check_update, update_msg],
                             [run_mode_title, run_mode_1, run_mode_2], [ling_xi_title, ling_xi_1, ling_xi_2, ling_xi_3],
                             [enable_party_title, enable_party_1, enable_party_0],
                             [max_resting_count_title, max_resting_count, sg.Text('', size=16), run_order_delay_title,
                              run_order_delay],
                             [drone_room_title, drone_room, sg.Text('', size=7), drone_count_limit_title,
                              drone_count_limit],
                             [reload_room_title, reload_room],
                             [rest_in_full_title, rest_in_full],
                             [exhaust_require_title, exhaust_require],
                             [workaholic_title, workaholic],
                             [resting_priority_title, resting_priority],
                             [start_automatically],
                         ], pad=((10, 10), (10, 10)))

    other_tab = sg.Tab('  外部调用 ',
                       [[mail_frame], [maa_frame]], pad=((10, 10), (10, 10)))
    window = sg.Window('Mower', [[sg.TabGroup([[main_tab, plan_tab, setting_tab, other_tab]], border_width=0,
                                              tab_border_width=0, focus_color='#bcc8e5',
                                              selected_background_color='#d4dae8', background_color='#aab6d3',
                                              tab_background_color='#aab6d3')]], font='微软雅黑', finalize=True,
                       resizable=False)
    build_plan(conf['planFile'])
    btn = None
    bind_scirpt()  # 为基建布局左边的站点排序绑定事件
    drag_task = DragTask()
    if conf['start_automatically']:
        start()
    while True:
        event, value = window.Read()
        if event == sg.WIN_CLOSED:
            break
        if event.endswith('-script'):  # 触发事件，进行处理
            run_script(event[:event.rindex('-')], drag_task)
            continue
        drag_task.clear()  # 拖拽事件连续不间断，若未触发事件，则初始化

        if event.startswith('plan_conf_'):  # plan_conf开头，为字符串输入的排班表配置
            key = event[10:]
            plan['conf'][key] = window[event].get().strip()
        elif event.startswith('plan_int_'):  # plan_int开头，为数值型输入的配置
            key = event[9:]
            try:
                plan['conf'][key] = int(window[event].get().strip())
            except ValueError:
                println(f'[{window[key + "_title"].get()}]需为数字')
        elif event.startswith('plan_radio_'):
            v_index = event.rindex('_')
            plan['conf'][event[11:v_index]] = int(event[v_index + 1:])
        elif event.startswith('conf_'):  # conf开头，为字符串输入的配置
            key = event[5:]
            value = window[event].get()
            conf[key] = value.strip() if isinstance(value, str) else value
        elif event.startswith('int_'):  # int开头，为数值型输入的配置
            key = event[4:]
            try:
                conf[key] = int(window[event].get().strip())
            except ValueError:
                println(f'[{window[key + "_title"].get()}]需为数字')
        elif event.startswith('float_'):  # float开头，为数值型输入的配置
            key = event[6:]
            try:
                conf[key] = float(window[event].get().strip())
            except ValueError:
                println(f'[{window[key + "_title"].get()}]需为数字')
        elif event.startswith('radio_'):
            v_index = event.rindex('_')
            conf[event[6:v_index]] = int(event[v_index + 1:])
        elif event == 'planFile' and plan_file.get() != conf['planFile']:  # 排班表
            write_plan(plan, conf['planFile'])
            build_plan(plan_file.get())
            plan_file.update(conf['planFile'])
        elif event.startswith('maa_weekly_plan_stage_'):  # 关卡名
            v_index = event.rindex('_')
            conf['maa_weekly_plan'][int(event[v_index + 1:])]['stage'] = [window[event].get()]
        elif event.startswith('maa_weekly_plan_medicine_'):  # 体力药
            v_index = event.rindex('_')
            conf['maa_weekly_plan'][int(event[v_index + 1:])]['medicine'] = int(window[event].get())
        elif event.startswith('btn_'):  # 设施按钮
            btn = event
            init_btn(event)
        elif event.endswith('-agent_change'):  # 干员填写
            input_agent = window[event[:event.rindex('-')]].get().strip()
            window[event[:event.rindex('-')]].update(value=input_agent, values=list(
                filter(lambda s: input_agent in s, ['Free'] + agent_list)))
        elif event.startswith('agent'):
            input_agent = window[event].get().strip()
            window[event].update(value=input_agent,
                                 values=list(filter(lambda s: input_agent in s, ['Free'] + agent_list)))
        elif event == 'check_update':
            window['update_msg'].update('正在检测更新...', text_color='black')
            window[event].update(disabled=True)
            window.perform_long_operation(check_update, 'check_update_value')
        elif event == 'check_update_value':
            if value[event] :
                b=sg.popup_yes_no("下载成功！是否立刻更新？")
                if b == 'Yes':
                    update_version()

        elif event == 'savePlan':  # 保存设施信息
            save_btn(btn)
        elif event == 'clearPlan':  # 清空当前设施信息
            clear_btn(btn)
        elif event == 'on':
            start()
        elif event == 'off':
            println('停止运行')
            child_conn.close()
            main_thread.terminate()
            on_btn.update(visible=True)
            off_btn.update(visible=False)

    window.close()
    save_conf(conf)
    write_plan(plan, conf['planFile'])


def start():
    global main_thread, child_conn
    window['on'].update(visible=False)
    window['off'].update(visible=True)
    clear()
    parent_conn, child_conn = Pipe()
    main_thread = Process(target=main, args=(conf, plan, operators, child_conn), daemon=True)
    main_thread.start()
    window.perform_long_operation(lambda: recv(parent_conn), 'recv')


def bind_scirpt():
    for i in range(3):
        for j in range(3):
            event = f'btn_room_{str(i + 1)}_{str(j + 1)}'
            window[event].bind("<B1-Motion>", "-motion-script")
            window[event].bind("<ButtonRelease-1>", "-ButtonRelease-script")
            window[event].bind("<Enter>", "-Enter-script")
    for i in range(5):
        event = f'agent{str(i + 1)}'
        window[event].bind("<Key>", "-agent_change")


def run_script(event, drag_task):
    # logger.info(f"{event}:{drag_task}")
    if event.endswith('-motion'):  # 拖拽事件，标志拖拽开始
        if drag_task.step == 0 or drag_task.step == 2:  # 若为2说明拖拽结束未进入其他元素，则初始化
            drag_task.btn = event[:event.rindex('-')]  # 记录初始按钮
            drag_task.step = 1  # 初始化键位，并推进任务步骤
    elif event.endswith('-ButtonRelease'):  # 松开按钮事件，标志着拖拽结束
        if drag_task.step == 1:
            drag_task.step = 2  # 推进任务步骤
    elif event.endswith('-Enter'):  # 进入元素事件，拖拽结束鼠标若在其他元素，会进入此事件
        if drag_task.step == 2:
            drag_task.new_btn = event[:event.rindex('-')]  # 记录需交换的按钮
            switch_plan(drag_task)
            drag_task.clear()
        else:
            drag_task.clear()


def switch_plan(drag_task):
    key1 = drag_task.btn[4:]
    key2 = drag_task.new_btn[4:]
    value1 = current_plan[key1] if key1 in current_plan else None;
    value2 = current_plan[key2] if key2 in current_plan else None;
    if value1 is not None:
        current_plan[key2] = value1
    elif key2 in current_plan:
        current_plan.pop(key2)
    if value2 is not None:
        current_plan[key1] = value2
    elif key1 in current_plan:
        current_plan.pop(key1)
    write_plan(plan, conf['planFile'])
    build_plan(conf['planFile'])


def init_btn(event):
    room_key = event[4:]
    station_name = current_plan[room_key]['name'] if room_key in current_plan.keys() else ''
    plans = current_plan[room_key]['plans'] if room_key in current_plan.keys() else []
    if room_key.startswith('room'):
        window['station_type_col'].update(visible=True)
        window['station_type'].update(station_name)
        visible_cnt = 3  # 设施干员需求数量
    else:
        if room_key == 'meeting':
            visible_cnt = 2
        elif room_key == 'factory' or room_key == 'contact':
            visible_cnt = 1
        else:
            visible_cnt = 5
        window['station_type_col'].update(visible=False)
        window['station_type'].update('')
    window['savePlan'].update(visible=True)
    window['clearPlan'].update(visible=True)
    for i in range(1, 6):
        if i > visible_cnt:
            window['setArea' + str(i)].update(visible=False)
            window['agent' + str(i)].update('')
            window['group' + str(i)].update('')
            window['replacement' + str(i)].update('')
        else:
            window['setArea' + str(i)].update(visible=True)
            window['agent' + str(i)].update(plans[i - 1]['agent'] if len(plans) >= i else '')
            window['group' + str(i)].update(plans[i - 1]['group'] if len(plans) >= i else '')
            window['replacement' + str(i)].update(','.join(plans[i - 1]['replacement']) if len(plans) >= i else '')


def save_btn(btn):
    plan1 = {'name': window['station_type'].get(), 'plans': []}
    for i in range(1, 6):
        agent = window['agent' + str(i)].get()
        group = window['group' + str(i)].get()
        replacement = list(filter(None, window['replacement' + str(i)].get().replace('，', ',').split(',')))
        if agent != '':
            plan1['plans'].append({'agent': agent, 'group': group, 'replacement': replacement})
        elif btn.startswith('btn_dormitory'):  # 宿舍
            plan1['plans'].append({'agent': 'Free', 'group': '', 'replacement': []})
    current_plan[btn[4:]] = plan1
    write_plan(plan, conf['planFile'])
    build_plan(conf['planFile'])


def clear_btn(btn):
    if btn[4:] in current_plan:
        current_plan.pop(btn[4:])
    init_btn(btn)
    write_plan(plan, conf['planFile'])
    build_plan(conf['planFile'])


def check_update():
    try:
        newest_version = compere_version()
        if newest_version:
            window['update_msg'].update('检测到有新版本'+newest_version+',正在下载...',text_color='black')
            download_version(newest_version)
            window['update_msg'].update('下载完毕！',text_color='green')
        else:
            window['update_msg'].update('已是最新版！',text_color='green')
    except Exception as e:
        logger.error(e)
        window['update_msg'].update('更新失败！',text_color='red')
        return None
    window['check_update'].update(disabled=False)
    return newest_version


# 接收推送
def recv(pipe):
    try:
        while True:
            msg = pipe.recv()
            if msg['type'] == 'log':
                println(msg['data'])
            elif msg['type'] == 'operators':
                global operators
                operators = msg['data']
    except EOFError:
        pipe.close()


def println(msg):
    global buffer
    global line
    global half_line_index
    maxLen = 500  # 最大行数
    buffer = f'{buffer}\n{time.strftime("%m-%d %H:%M:%S")} {msg}'.strip()
    window['log'].update(value=buffer)
    if line == maxLen // 2:
        half_line_index = len(buffer)
    if line >= maxLen:
        buffer = buffer[half_line_index:]
        line = maxLen // 2
    else:
        line += 1


# 清空输出栏
def clear():
    global buffer
    global line
    buffer = ''
    window['log'].update(value=buffer)
    line = 0


class DragTask:
    def __init__(self):
        self.btn = None
        self.new_btn = None
        self.step = 0

    def clear(self):
        self.btn = None
        self.new_btn = None
        self.step = 0

    def __repr__(self):
        return f"btn:{self.btn},new_btn:{self.new_btn},step:{self.step}"


if __name__ == '__main__':
    freeze_support()
    menu()