main.py

# -*- coding: UTF-8 -*-
# @Time    : 2021/7/27 17:21
# @Author  : Hui-Shao
# %% imports

import os
import shutil
import sys
import traceback

import toml

from utils.avalon import Avalon
from utils.forest import Forest
from utils.user import User


# %% 建立用于保存用户文件的目录
def makedir():
    if not os.path.exists("_user_files"):
        os.mkdir("_user_files")
    else:
        pass


# %% 检查 config.toml 格式是否需要更新
def is_config_need_update():
    if os.path.exists("config_need_update"):  # config_need_update 是手动放置的标记文件
        Avalon.error("\n    config.toml 需要重新配置\n    配置格式请参照 config.toml.example\n    程序退出")
        exit(0)
    else:
        pass


# %% 读取配置文件
def read_config():
    Avalon.info("读取配置文件中……", front="\n")
    global config, username, passwd, uid, remember_token, server
    config_file = sys.argv[1] if len(sys.argv) > 1 else "config.toml"
    try:
        with open(config_file, "r", encoding="utf-8") as f:
            config = toml.load(f)
    except UnicodeEncodeError:
        with open(config_file, "r", encoding="gbk") as f:
            config = toml.load(f)
    except IOError:
        Avalon.error(f"无法加载{config_file}, 请检查文件是否存在, 文件名是否正确")
        return False
    except toml.TomlDecodeError as decode_err_info:
        Avalon.error(f"载入{config_file}错误, 请检查配置文件内容是否规范无误")
        Avalon.error(decode_err_info)
        return False
    except Exception:
        Avalon.error(f"无法加载{config_file}, 其他错误\n")
        Avalon.error(traceback.format_exc(3))
        return False
    else:
        username = config["user"]["username"]
        passwd = config["user"]["password"]
        uid = config["user"]["uid"]
        remember_token = config["user"]["remember_token"]
        server = config["user"]["server"]
        Avalon.info("配置文件读取成功")
        return True


# %% 写配置文件
def write_config():
    config_file = sys.argv[1] if len(sys.argv) > 1 else "config.toml"
    config_2 = config.copy()
    config_2["user"].update({"uid": user.uid, "remember_token": user.remember_token, "server": user.server})
    try:
        f = open(config_file, "w+", encoding="utf-8")
        toml.dump(config_2, f)
        f.close()
    except IOError:
        Avalon.error("IOError")
        return False
    else:
        return True


def logout():
    if F.logout():
        user.uid = 0
        user.remember_token = ""
        write_config()
    try:
        shutil.rmtree("_user_files")
    except IOError:
        Avalon.warning("删除_user_files目录失败, 在切换用户前务必手动删除！")


def login():
    if user.uid * len(user.remember_token) == 0:  # 若未读取到保存的 uid 和 remember_token 则调用登录
        login_info = F.login()
        if len(login_info):
            user.uid = login_info["uid"]
            user.remember_token = login_info["remember_token"]
            user.server = login_info["server"]
            write_config()
        else:
            exit(0)
    else:
        Avalon.info(f"当前用户信息 -> {user.username} ({user.uid}) ({user.server})", front="\n")


def common_settings():
    # url setting
    if (not config["common"]["enable_network_accelerator"]) and config["user"]["server"] == "global":
        F.select_api_url(0)  # 若未启用 network_accelerator 且 user.server 为 global , 更改 api_url 为 官方默认无加速链接
    # app_version setting
    F.app_version = str(config["common"]["app_version"])


def run():
    login()
    if config["remove_plants_by_rewarded_ad"]["enable"]:
        F.remove_plants_by_rewarded_ad()
    if config["auto_plant"]["enable"]:
        F.auto_plant(_boost_by_ad=config["boost_plant_by_rewarded_ad"]["enable"],
                     _mode=config["auto_plant"]["mode"],
                     _total_n=config["auto_plant"]["number"],
                     _short_sleep_time=config["auto_plant"]["short_sleep_time"],
                     _customize_plant_time=config["auto_plant"]["customize_plant_time"]
                     )
    if config["manually_plant"]["enable"]:
        F.manually_plant(_boost_by_ad=config["boost_plant_by_rewarded_ad"]["enable"])
    if config["create_room"]["enable"]:
        F.create_room(_boost_by_ad=config["boost_plant_by_rewarded_ad"]["enable"])
    if config["auto_logout"]["enable"]:
        logout()
    Avalon.info("所有任务执行完毕~\n", front="\n")


if __name__ == '__main__':
    try:
        os.chdir(sys.path[0])
        config = {}
        username = ""
        passwd = ""
        uid = 0
        remember_token = ""
        server = "auto"
        is_config_need_update()
        makedir()
        if read_config():
            user = User(username, passwd, uid, remember_token, server)
            F = Forest(user)
            common_settings()  # 再次对 F 中的类变量进行设置
            run()
        else:
            sys.exit(0)
    except KeyboardInterrupt:
        Avalon.warning("捕获到 KeyboardInterrupt, 程序终止\n", front="\n")
        sys.exit(0)
    except KeyError:
        Avalon.warning("配置文件 config.toml 可能存在问题, 请检查!\n", front="\n")
        Avalon.warning(traceback.format_exc(3))
        sys.exit(1)
    except Exception:
        Avalon.error("其他错误! 程序终止!\n", front="\n")
        Avalon.error(traceback.format_exc(3))
        sys.exit(2)