mirror of
https://github.com/BluemediaGER/ScanOS.git
synced 2024-09-20 02:35:29 +02:00
55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
import threading, logging
|
|
from contextlib import asynccontextmanager
|
|
from typing import Annotated
|
|
|
|
from fastapi import FastAPI, Depends
|
|
|
|
from app.data import models
|
|
from app.data.database import SessionLocal, engine
|
|
|
|
from uvicorn.logging import DefaultFormatter
|
|
|
|
from app.scanner.scanner import Scanner
|
|
from app.scanner.scanner import Status as ScannerStatus
|
|
|
|
# Set up logging
|
|
logger = logging.getLogger()
|
|
__syslog = logging.StreamHandler()
|
|
__syslog.setFormatter(DefaultFormatter(fmt="%(levelprefix)s %(message)s", use_colors=True))
|
|
logger.setLevel(logging.INFO)
|
|
logger.addHandler(__syslog)
|
|
|
|
# Create database
|
|
models.Base.metadata.create_all(bind=engine)
|
|
|
|
# Set up scanner instance
|
|
__scanner = Scanner("/var/www/html/img", logger)
|
|
|
|
# Preload scanner after FastAPI start
|
|
@asynccontextmanager
|
|
async def __lifespan(app: FastAPI):
|
|
threading.Thread(target=__scanner.preload).start()
|
|
yield
|
|
|
|
app = FastAPI(lifespan=__lifespan)
|
|
|
|
# SQLAlchemiy session dependency
|
|
def get_db():
|
|
db = SessionLocal()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
def get_scanner():
|
|
return __scanner
|
|
|
|
from app.routers import power
|
|
app.include_router(power.router)
|
|
|
|
from app.routers import scan
|
|
app.include_router(scan.router)
|
|
|
|
@app.get("/api/ready")
|
|
async def __ready(scanner: Annotated[Scanner, Depends(get_scanner)]):
|
|
return scanner.get_status() != ScannerStatus.INITIALIZED |