112 lines
3.0 KiB
Python
112 lines
3.0 KiB
Python
# AUTO_MAA:A MAA Multi Account Management and Automation Tool
|
|
# Copyright © 2024-2025 DLmaster361
|
|
# Copyright © 2025 MoeSnowyFox
|
|
|
|
# This file is part of AUTO_MAA.
|
|
|
|
# AUTO_MAA is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License,
|
|
# or (at your option) any later version.
|
|
|
|
# AUTO_MAA is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty
|
|
# of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
|
|
# the GNU General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with AUTO_MAA. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
# Contact: DLmaster_361@163.com
|
|
|
|
|
|
import os
|
|
import sys
|
|
import ctypes
|
|
|
|
from utils import get_logger
|
|
|
|
logger = get_logger("主程序")
|
|
|
|
|
|
def is_admin() -> bool:
|
|
"""检查当前程序是否以管理员身份运行"""
|
|
try:
|
|
return ctypes.windll.shell32.IsUserAnAdmin()
|
|
except:
|
|
return False
|
|
|
|
|
|
# @logger.catch
|
|
def main():
|
|
|
|
if is_admin():
|
|
|
|
import asyncio
|
|
import uvicorn
|
|
from fastapi import FastAPI
|
|
from contextlib import asynccontextmanager
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
|
|
from core import Config, MainTimer
|
|
|
|
await Config.init_config()
|
|
main_timer = asyncio.create_task(MainTimer.second_task())
|
|
|
|
yield
|
|
|
|
main_timer.cancel()
|
|
try:
|
|
await main_timer
|
|
except asyncio.CancelledError:
|
|
logger.info("主业务定时器已关闭")
|
|
|
|
logger.info("AUTO_MAA 后端程序关闭")
|
|
logger.info("----------------END----------------")
|
|
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from api import (
|
|
info_router,
|
|
scripts_router,
|
|
plan_router,
|
|
queue_router,
|
|
setting_router,
|
|
)
|
|
|
|
app = FastAPI(
|
|
title="AUTO_MAA",
|
|
description="API for managing automation scripts, plans, and tasks",
|
|
version="1.0.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"], # 允许所有域名跨域访问
|
|
allow_credentials=True,
|
|
allow_methods=["*"], # 允许所有请求方法，如 GET、POST、PUT、DELETE
|
|
allow_headers=["*"], # 允许所有请求头
|
|
)
|
|
|
|
app.include_router(info_router)
|
|
app.include_router(scripts_router)
|
|
app.include_router(plan_router)
|
|
app.include_router(queue_router)
|
|
app.include_router(setting_router)
|
|
|
|
uvicorn.run(app, host="0.0.0.0", port=8000)
|
|
|
|
else:
|
|
|
|
ctypes.windll.shell32.ShellExecuteW(
|
|
None, "runas", sys.executable, os.path.realpath(sys.argv[0]), None, 1
|
|
)
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
main()
|