from fastapi_cdn_host import monkey_patch_for_docs_ui
from fastapi import FastAPI
from uvicorn import run

from models.gitModels import Users
from models.aiModels import Scan_Tasks

from api.gitRouter import gitrouter
from api.testapi import testapi
from api.aiRouter import airouter

from tortoise.contrib.fastapi import register_tortoise
from db_config import TORTOISE_ORM

app = FastAPI()
monkey_patch_for_docs_ui(app)
register_tortoise(app=app, config=TORTOISE_ORM)

@app.get("/user/{id}")
async def test(id: int):
    user= await Users.get(id=id)
    print(type(user))
    return user

@app.get("/task/{id}")
async def test(id: int):
    task = await Scan_Tasks.create(repo_id=1, state=1,result={"a":1}, create_time=1234567890,scan_start_time=1234567890,scan_end_time=1234567890,create_user="admin",repo_hash="1234567890")
    print(type(task))
    return task

app.include_router(gitrouter,prefix="/git")
app.include_router(testapi,prefix="/test")

app.include_router(airouter,prefix="/ai")

if __name__ == "__main__":
    run(app,host="127.0.0.1",port=8000)