mirror of
https://github.com/TriM-Organization/LiteyukiBot-TriM.git
synced 2024-11-25 16:45:03 +08:00
78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
import json
|
|
import os.path
|
|
import platform
|
|
import sys
|
|
import time
|
|
|
|
import nonebot
|
|
|
|
__NAME__ = "LiteyukiBot"
|
|
__VERSION__ = "6.3.2" # 60201
|
|
|
|
import requests
|
|
|
|
from liteyuki.utils.base.config import load_from_yaml, config
|
|
from liteyuki.utils.base.log import init_log
|
|
from liteyuki.utils.base.data_manager import TempConfig, auto_migrate, common_db
|
|
|
|
major, minor, patch = map(int, __VERSION__.split("."))
|
|
__VERSION_I__ = major * 10000 + minor * 100 + patch
|
|
|
|
|
|
def register_bot():
|
|
url = "https://api.liteyuki.icu/register"
|
|
data = {
|
|
"name" : __NAME__,
|
|
"version" : __VERSION__,
|
|
"version_i": __VERSION_I__,
|
|
"python" : f"{platform.python_implementation()} {platform.python_version()}",
|
|
"os" : f"{platform.system()} {platform.version()} {platform.machine()}"
|
|
}
|
|
try:
|
|
nonebot.logger.info("Waiting for register to Liteyuki...")
|
|
resp = requests.post(url, json=data, timeout=(10, 15))
|
|
if resp.status_code == 200:
|
|
data = resp.json()
|
|
if liteyuki_id := data.get("liteyuki_id"):
|
|
with open("data/liteyuki/liteyuki.json", "wb") as f:
|
|
f.write(json.dumps(data).encode("utf-8"))
|
|
nonebot.logger.success(f"Register {liteyuki_id} to Liteyuki successfully")
|
|
else:
|
|
raise ValueError(f"Register to Liteyuki failed: {data}")
|
|
|
|
except Exception as e:
|
|
nonebot.logger.warning(f"Register to Liteyuki failed, but it's no matter: {e}")
|
|
|
|
|
|
def init():
|
|
"""
|
|
初始化
|
|
Returns:
|
|
|
|
"""
|
|
# 检测python版本是否高于3.10
|
|
auto_migrate()
|
|
init_log()
|
|
if sys.version_info < (3, 10):
|
|
nonebot.logger.error("Requires Python3.10+ to run, please upgrade your Python Environment.")
|
|
exit(1)
|
|
temp_data: TempConfig = common_db.where_one(TempConfig(), default=TempConfig())
|
|
temp_data.data["start_time"] = time.time()
|
|
common_db.save(temp_data)
|
|
|
|
# 在加载完成语言后再初始化日志
|
|
nonebot.logger.info("Liteyuki is initializing...")
|
|
|
|
if not os.path.exists("data/liteyuki/liteyuki.json"):
|
|
register_bot()
|
|
|
|
if not os.path.exists("pyproject.toml"):
|
|
with open("pyproject.toml", "w", encoding="utf-8") as f:
|
|
f.write("[tool.nonebot]\n")
|
|
|
|
nonebot.logger.info(
|
|
f"Run Liteyuki with Python{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro} "
|
|
f"at {sys.executable}"
|
|
)
|
|
nonebot.logger.info(f"{__NAME__} {__VERSION__}({__VERSION_I__}) is running")
|