many improvements
This commit is contained in:
+63
@@ -0,0 +1,63 @@
|
||||
import uvicorn
|
||||
from fastapi import FastAPI, Query
|
||||
from pydantic import BaseModel
|
||||
|
||||
from fastapi_sqlalchemy import DBSessionMiddleware, db
|
||||
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
|
||||
from rota.shifts import SingleShift, ShiftConstraint
|
||||
|
||||
from web.models import SingleShift as ModelSingleShift
|
||||
from web.models import ShiftConstraint as ModelShiftConstraint
|
||||
|
||||
#from rota.wokers
|
||||
|
||||
|
||||
load_dotenv('../.env')
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
app.add_middleware(DBSessionMiddleware, db_url="sqlite:///test.db")
|
||||
|
||||
@app.post('/shifts/', response_model=SingleShift)
|
||||
async def shifts(shift: ModelSingleShift):
|
||||
db_shift = ModelSingleShift(name=shift.name, length=shift.length)
|
||||
db.session.add(db_shift)
|
||||
db.session.commit()
|
||||
return db_shift
|
||||
|
||||
|
||||
@app.get("/shifts/")
|
||||
async def shifts():
|
||||
shifts = db.session.query(ModelSingleShift).all()
|
||||
|
||||
return shifts
|
||||
|
||||
|
||||
@app.get("/")
|
||||
async def root():
|
||||
return {"message": "Hello World"}
|
||||
|
||||
@app.get("/files/{file_path:path}")
|
||||
async def read_file(file_path: str):
|
||||
return {"file_path": file_path}
|
||||
|
||||
|
||||
|
||||
class Item(BaseModel):
|
||||
name: str
|
||||
description: str | None = None
|
||||
price: float
|
||||
tax: float | None = None
|
||||
|
||||
|
||||
|
||||
@app.post("/items/")
|
||||
async def create_item(item: Item):
|
||||
item.name + item.price
|
||||
return item
|
||||
|
||||
if __name__ == '__main__':
|
||||
uvicorn.run(app, host='0.0.0.0', port=8008)
|
||||
Reference in New Issue
Block a user