12345678910111213141516171819202122232425262728293031323334353637 |
- 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)
|