gitscope
/
← marzneshin / app/marzneshin.py / blame
Files Commits 446 Branches 1 Tags 43 Markers 4 README License
https://dev.kronoz.win/r/marzneshin.git
app/marzneshin.py · v0.6.1
Source
1 f7af2b3 Dawsh 2y ago import asyncio
2 f7af2b3 Dawsh 2y ago import logging
3 d377c4c Dawsh 2y ago from contextlib import asynccontextmanager
4 f7af2b3 Dawsh 2y ago from datetime import datetime as dt
5 f7af2b3 Dawsh 2y ago from datetime import timedelta as td
6 f4fecee Dawsh 2y ago from typing import AsyncGenerator
7 f4fecee Dawsh 2y ago
8 f7af2b3 Dawsh 2y ago from apscheduler.schedulers.asyncio import AsyncIOScheduler
9 f7af2b3 Dawsh 2y ago from fastapi import FastAPI, Request, status
10 f7af2b3 Dawsh 2y ago from fastapi.encoders import jsonable_encoder
11 f7af2b3 Dawsh 2y ago from fastapi.exceptions import RequestValidationError
12 f7af2b3 Dawsh 2y ago from fastapi.middleware.cors import CORSMiddleware
13 eafb4a4 Mrdn 2y ago from fastapi.responses import HTMLResponse, JSONResponse
14 d4ec019 Dawsh 2y ago from fastapi_pagination import add_pagination
15 f7af2b3 Dawsh 2y ago from starlette.staticfiles import StaticFiles
16 eafb4a4 Mrdn 2y ago from uvicorn import Config, Server
17 f7af2b3 Dawsh 2y ago
18 8111eb7 Dawsh 2y ago from app.config.env import (
19 01cdf2e Dawsh 2y ago DEBUG,
20 01cdf2e Dawsh 2y ago DOCS,
21 01cdf2e Dawsh 2y ago HOME_PAGE_TEMPLATE,
22 01cdf2e Dawsh 2y ago UVICORN_HOST,
23 01cdf2e Dawsh 2y ago UVICORN_PORT,
24 01cdf2e Dawsh 2y ago UVICORN_SSL_CERTFILE,
25 01cdf2e Dawsh 2y ago UVICORN_SSL_KEYFILE,
26 01cdf2e Dawsh 2y ago UVICORN_UDS,
27 a0e778c Mrdn 2y ago DASHBOARD_PATH,
28 8111eb7 Dawsh 2y ago WEBHOOK_ADDRESS,
29 01cdf2e Dawsh 2y ago )
30 8111eb7 Dawsh 2y ago from app.templates import render_template
31 f7af2b3 Dawsh 2y ago from . import __version__, telegram
32 2de9d72 Dawsh 2y ago from .routes import api_router
33 01cdf2e Dawsh 2y ago from .tasks import (
34 01cdf2e Dawsh 2y ago delete_expired_reminders,
35 01cdf2e Dawsh 2y ago nodes_startup,
36 01cdf2e Dawsh 2y ago record_user_usages,
37 01cdf2e Dawsh 2y ago reset_user_data_usage,
38 01cdf2e Dawsh 2y ago review_users,
39 01cdf2e Dawsh 2y ago send_notifications,
40 01cdf2e Dawsh 2y ago )
41 f7af2b3 Dawsh 2y ago
42 f7af2b3 Dawsh 2y ago logger = logging.getLogger(__name__)
43 f7af2b3 Dawsh 2y ago
44 f4fecee Dawsh 2y ago
45 f4fecee Dawsh 2y ago @asynccontextmanager
46 f4fecee Dawsh 2y ago async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
47 f4fecee Dawsh 2y ago await nodes_startup()
48 f4fecee Dawsh 2y ago yield
49 f4fecee Dawsh 2y ago scheduler.shutdown()
50 f4fecee Dawsh 2y ago logger.info("Sending pending notifications before shutdown...")
51 f4fecee Dawsh 2y ago await send_notifications()
52 f4fecee Dawsh 2y ago
53 f4fecee Dawsh 2y ago
54 f7af2b3 Dawsh 2y ago app = FastAPI(
55 f7af2b3 Dawsh 2y ago title="MarzneshinAPI",
56 f7af2b3 Dawsh 2y ago description="Unified GUI Censorship Resistant Solution Powered by Xray",
57 f7af2b3 Dawsh 2y ago version=__version__,
58 f4fecee Dawsh 2y ago lifespan=lifespan,
59 f7af2b3 Dawsh 2y ago docs_url="/docs" if DOCS else None,
60 f7af2b3 Dawsh 2y ago redoc_url="/redoc" if DOCS else None,
61 f7af2b3 Dawsh 2y ago )
62 f7af2b3 Dawsh 2y ago
63 2de9d72 Dawsh 2y ago app.include_router(api_router)
64 d4ec019 Dawsh 2y ago add_pagination(app)
65 f7af2b3 Dawsh 2y ago
66 f7af2b3 Dawsh 2y ago
67 f7af2b3 Dawsh 2y ago @app.get("/", response_class=HTMLResponse)
68 2de9d72 Dawsh 2y ago def home_page():
69 f7af2b3 Dawsh 2y ago return render_template(HOME_PAGE_TEMPLATE)
70 f7af2b3 Dawsh 2y ago
71 f7af2b3 Dawsh 2y ago
72 f7af2b3 Dawsh 2y ago app.add_middleware(
73 f7af2b3 Dawsh 2y ago CORSMiddleware,
74 f7af2b3 Dawsh 2y ago allow_origins=["*"],
75 f7af2b3 Dawsh 2y ago allow_credentials=True,
76 f7af2b3 Dawsh 2y ago allow_methods=["*"],
77 f7af2b3 Dawsh 2y ago allow_headers=["*"],
78 f7af2b3 Dawsh 2y ago )
79 f7af2b3 Dawsh 2y ago
80 f7af2b3 Dawsh 2y ago scheduler = AsyncIOScheduler(timezone="UTC")
81 eafb4a4 Mrdn 2y ago scheduler.add_job(record_user_usages, "interval", coalesce=True, seconds=30)
82 f1ab16c Dawsh 2y ago scheduler.add_job(
83 d377c4c Dawsh 2y ago review_users, "interval", seconds=30, coalesce=True, max_instances=1
84 f1ab16c Dawsh 2y ago )
85 eafb4a4 Mrdn 2y ago scheduler.add_job(reset_user_data_usage, "interval", coalesce=True, hours=1)
86 f7af2b3 Dawsh 2y ago
87 8111eb7 Dawsh 2y ago if WEBHOOK_ADDRESS:
88 f1ab16c Dawsh 2y ago scheduler.add_job(
89 f1ab16c Dawsh 2y ago send_notifications, "interval", seconds=30, replace_existing=True
90 f1ab16c Dawsh 2y ago )
91 eafb4a4 Mrdn 2y ago scheduler.add_job(
92 eafb4a4 Mrdn 2y ago delete_expired_reminders,
93 eafb4a4 Mrdn 2y ago "interval",
94 eafb4a4 Mrdn 2y ago hours=2,
95 eafb4a4 Mrdn 2y ago start_date=dt.utcnow() + td(minutes=1),
96 eafb4a4 Mrdn 2y ago )
97 f7af2b3 Dawsh 2y ago
98 f7af2b3 Dawsh 2y ago
99 f7af2b3 Dawsh 2y ago @app.exception_handler(RequestValidationError)
100 f1ab16c Dawsh 2y ago def validation_exception_handler(
101 f1ab16c Dawsh 2y ago request: Request, exc: RequestValidationError
102 f1ab16c Dawsh 2y ago ):
103 f7af2b3 Dawsh 2y ago details = {}
104 f7af2b3 Dawsh 2y ago for error in exc.errors():
105 f7af2b3 Dawsh 2y ago details[error["loc"][-1]] = error.get("msg")
106 f7af2b3 Dawsh 2y ago return JSONResponse(
107 f7af2b3 Dawsh 2y ago status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
108 f7af2b3 Dawsh 2y ago content=jsonable_encoder({"detail": details}),
109 f7af2b3 Dawsh 2y ago )
110 f7af2b3 Dawsh 2y ago
111 f7af2b3 Dawsh 2y ago
112 f7af2b3 Dawsh 2y ago async def main():
113 eafb4a4 Mrdn 2y ago if not DEBUG:
114 f7af2b3 Dawsh 2y ago app.mount(
115 a0e778c Mrdn 2y ago DASHBOARD_PATH,
116 4466347 Mrdn 2y ago StaticFiles(directory="dashboard/dist", html=True),
117 4466347 Mrdn 2y ago name="dashboard",
118 f7af2b3 Dawsh 2y ago )
119 a0e778c Mrdn 2y ago app.mount(
120 a0e778c Mrdn 2y ago "/static/",
121 a0e778c Mrdn 2y ago StaticFiles(directory="dashboard/dist/static"),
122 a0e778c Mrdn 2y ago name="static",
123 a0e778c Mrdn 2y ago )
124 a0e778c Mrdn 2y ago app.mount(
125 a0e778c Mrdn 2y ago "/locales/",
126 a0e778c Mrdn 2y ago StaticFiles(directory="dashboard/dist/locales"),
127 a0e778c Mrdn 2y ago name="locales",
128 a0e778c Mrdn 2y ago )
129 f7af2b3 Dawsh 2y ago scheduler.start()
130 f7af2b3 Dawsh 2y ago asyncio.create_task(telegram.start_bot())
131 f7af2b3 Dawsh 2y ago cfg = Config(
132 f7af2b3 Dawsh 2y ago app=app,
133 f7af2b3 Dawsh 2y ago host=UVICORN_HOST,
134 f7af2b3 Dawsh 2y ago port=UVICORN_PORT,
135 f7af2b3 Dawsh 2y ago uds=(None if DEBUG else UVICORN_UDS),
136 f7af2b3 Dawsh 2y ago ssl_certfile=UVICORN_SSL_CERTFILE,
137 f7af2b3 Dawsh 2y ago ssl_keyfile=UVICORN_SSL_KEYFILE,
138 f7af2b3 Dawsh 2y ago workers=1,
139 f7af2b3 Dawsh 2y ago reload=DEBUG,
140 f7af2b3 Dawsh 2y ago log_level=logging.DEBUG if DEBUG else logging.INFO,
141 f7af2b3 Dawsh 2y ago )
142 f7af2b3 Dawsh 2y ago server = Server(cfg)
143 f7af2b3 Dawsh 2y ago await server.serve()