main.py

from utils.master import Master
from utils.time import getNowTime, getNowTimeWithOffset
from utils.messages import *
from UserInterface import UserInterface
from time import sleep
from datetime import datetime as dt


def getInfo(trial, plan, resp):
    info = plan["seatsInfo"][0]
    print(f"[{getNowTime()}][try={trial}] MSG={resp['MESSAGE']} | {info['roomName']},{info['floorName']},{info['seatNum']}座,{plan['beginTime'].strftime('%Y-%m-%d %H:%M')},{plan['duration']}小时"
)


def run():
    api = UserInterface()
    master = api.master
    if api.run():
        planIndex = 0
        for plan in master.plans:
            planCode = master.planCode[planIndex]
            checkPoint = master.job["checkPoint"]
            
            if not master.job["logDetails"]:
                _planCode = '****'
            else:
                _planCode = planCode

            maxTrials = master.job["maxTrials"]
            delay = master.job["delay"]

            nowTime = getNowTime(precision="second")
            try:
                preExeTime_details = getNowTime().strptime(
                    master.job["preExeTime"], "%H:%M:%S"
                )
            except TypeError as e:
                print(f"[{getNowTime()}]请检查配置文件中的preExeTime是否正确！\n当前preExeTime={master.job['preExeTime']}")
                return
            preExeTime =getNowTimeWithOffset(days=1).replace(
                hour=preExeTime_details.hour,
                minute=preExeTime_details.minute,
                second=preExeTime_details.second,
            )
            
            time_wait = (preExeTime - nowTime).seconds
            if time_wait < 7200:
                # checkpoint A: executeTime seats reservation at 00:00:01
                print(f"[{getNowTime()}][checkpoint A]未到任务执行时间，还差[{time_wait}]s，等待中...")
                if checkPoint is True:
                    sleep(time_wait)
                else:
                    master.job['checkPoint'] = True
                    master.saveConfig()
                    return
            else:
                # checkpoint B: executeTime seats reservation at 20:00:01
                executeTime_details = getNowTime().strptime(
                    master.job["executeTime"], "%H:%M:%S"
                )
                executeTime = getNowTime().replace(
                    hour=executeTime_details.hour,
                    minute=executeTime_details.minute,
                    second=executeTime_details.second,
                )

                if nowTime < executeTime:
                    time_wait = (executeTime - nowTime).seconds
                    print(f"[{getNowTime()}][checkpoint B]未到任务执行时间，还差[{time_wait}]s，等待中...")
                    if checkPoint is True:
                        sleep(time_wait)
                    else:
                        master.job['checkPoint'] = True
                        master.saveConfig()
                        return

            print(f"[{getNowTime()}][plan[{planIndex}]={_planCode}] 开始预约...")
            plan["beginTime"] = getNowTimeWithOffset(days=2, hours=0).replace(
                hour=plan["beginTime"].hour, minute=0
            )
            isSuccess = False
            tryTimes = 0
            _msg = MSG_NO_RESPONSE
            while tryTimes < maxTrials and not isSuccess:
                res = master.run(plan)
                if res is None:
                    isSuccess = False
                    tryTimes += 1
                    continue
                elif api.master.job["logDetails"]:
                    getInfo(tryTimes + 1, plan, res)
                try:
                    if res["DATA"]["result"] != "fail":
                        isSuccess = True
                    elif str(res["MESSAGE"]).startswith(MSG_INVALID_REQUEST):
                        print(f"[{getNowTime()}]plan[{planIndex}]={_planCode}] {res['MESSAGE']}，请更新你的仓库或提交issue！")
                        break
                    elif str(res["MESSAGE"]).startswith(MSG_TIME_OUT_OF_RANGE):
                        sleep(delay)
                    elif str(res["MESSAGE"]).startswith(MSG_DUPLICATE):
                        isSuccess = False
                        break
                    elif str(res["MESSAGE"]).startswith(MSG_SEAT_UNAVAILABLE):
                        isSuccess = False
                        break
                except Exception as e:
                    print(f"[{getNowTime()}]plan[{planIndex}]={_planCode}] 预约失败，原因：{e}")
                finally:
                    tryTimes += 1
                    _msg = res["MESSAGE"]
            print(f"[{getNowTime()}]plan[{planIndex}]={_planCode}] 预约结果：{_msg}")
            planIndex += 1
        

if __name__ == "__main__":
    run()