demo.py 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. from fastapi_cdn_host import monkey_patch_for_docs_ui
  2. from fastapi import FastAPI
  3. from uvicorn import run
  4. from models.gitModels import Users
  5. from models.aiModels import Scan_Tasks
  6. from api.gitRouter import gitrouter
  7. from api.testapi import testapi
  8. from api.aiRouter import airouter
  9. from tortoise.contrib.fastapi import register_tortoise
  10. from db_config import TORTOISE_ORM
  11. app = FastAPI()
  12. monkey_patch_for_docs_ui(app)
  13. register_tortoise(app=app, config=TORTOISE_ORM)
  14. @app.get("/user/{id}")
  15. async def test(id: int):
  16. user= await Users.get(id=id)
  17. print(type(user))
  18. return user
  19. @app.get("/task/{id}")
  20. async def test(id: int):
  21. 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")
  22. print(type(task))
  23. return task
  24. app.include_router(gitrouter,prefix="/git")
  25. app.include_router(testapi,prefix="/test")
  26. app.include_router(airouter,prefix="/ai")
  27. if __name__ == "__main__":
  28. run(app,host="127.0.0.1",port=8000)