mirror of
https://github.com/EstrellaXD/Auto_Bangumi.git
synced 2026-05-12 19:16:47 +08:00
106 lines
2.5 KiB
Python
106 lines
2.5 KiB
Python
import uvicorn
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.responses import HTMLResponse
|
|
from fastapi.templating import Jinja2Templates
|
|
from pydantic import BaseModel
|
|
import logging
|
|
|
|
from core import RSSAnalyser
|
|
from core import DownloadClient
|
|
from conf import settings, parse
|
|
from utils import json_config
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
# templates = Jinja2Templates(directory="templates")
|
|
|
|
|
|
# @app.get("/", response_class=HTMLResponse)
|
|
# def index(request: Request):
|
|
# context = {"request": request}
|
|
# return templates.TemplateResponse("index.html", context)
|
|
|
|
|
|
class Config(BaseModel):
|
|
rss_link: str
|
|
host: str
|
|
user_name: str
|
|
password: str
|
|
download_path: str
|
|
method: str
|
|
enable_group_tag: bool
|
|
not_contain: str
|
|
debug_mode: bool
|
|
season_one_tag: bool
|
|
remove_bad_torrent: bool
|
|
enable_eps_complete: bool
|
|
|
|
|
|
@app.post("/api/v1/config")
|
|
async def config(config: Config):
|
|
data = {
|
|
"rss_link": config.rss_link,
|
|
"host": config.host,
|
|
"user_name": config.user_name,
|
|
"password": config.password,
|
|
"download_path": config.download_path,
|
|
"method": config.method,
|
|
"enable_group_tag": config.enable_group_tag,
|
|
"not_contain": config.not_contain,
|
|
"debug_mode": config.debug_mode,
|
|
"season_one": config.season_one_tag,
|
|
"remove_bad_torrent": config.remove_bad_torrent,
|
|
"enable_eps_complete": config.enable_eps_complete
|
|
}
|
|
json_config.save("/config/config.json", data)
|
|
return "received"
|
|
|
|
|
|
class RSS(BaseModel):
|
|
link: str
|
|
|
|
|
|
@app.post("/api/v1/subscriptions")
|
|
async def receive(link: RSS):
|
|
client = DownloadClient()
|
|
try:
|
|
data = RSSAnalyser().rss_to_data(link.link)
|
|
client.add_collection_feed(link.link, item_path=data["official_title"])
|
|
client.set_rule(data, link.link)
|
|
return data
|
|
except Exception as e:
|
|
logger.debug(e)
|
|
return "Error"
|
|
|
|
|
|
class Search(BaseModel):
|
|
group: str
|
|
title: str
|
|
subtitle: str
|
|
|
|
|
|
@app.post("/api/v1/search")
|
|
async def search(input: Search):
|
|
return "Nothing Happened"
|
|
|
|
|
|
def run():
|
|
args = parse()
|
|
if args.debug:
|
|
try:
|
|
from conf.const_dev import DEV_SETTINGS
|
|
settings.init(DEV_SETTINGS)
|
|
except ModuleNotFoundError:
|
|
logger.debug("Please copy `const_dev.py` to `const_dev.py` to use custom settings")
|
|
else:
|
|
settings.init()
|
|
uvicorn.run(app, host="0.0.0.0", port=settings.webui_port)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run()
|
|
|