many improvements

2022-07-11 19:46:40 +01:00
parent 0cebbfddf6
commit dc975b991c
25 changed files with 1949 additions and 863 deletions
@@ -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)
@@ -0,0 +1,31 @@
+from sqlalchemy import Column, DateTime, ForeignKey, Integer, String, Float, Boolean
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import relationship
+from sqlalchemy.sql import func
+
+Base = declarative_base()
+
+class SingleShift(Base):
+    __tablename__ = 'single_shift'
+    id  = Column(Integer, primary_key=True, index=True)
+    name = Column(String)
+    length = Column(Float)
+    balance_offset = Column(Float, nullable=True)
+    balance_weighting = Column(Float)
+    workers_required = Column(Float)
+    assign_as_block = Column(Boolean)
+    force_as_block = Column(Boolean)
+    hard_constrain_shift = Column(Boolean)
+    bank_holidays_only = Column(Boolean)
+    constraint_id = Column(Integer, ForeignKey('shift_constraint.id'))
+
+    constraint = relationship('ShiftConstraint')
+
+
+class ShiftConstraint(Base):
+    __tablename__ = 'shift_constraint'
+    id = Column(Integer, primary_key=True)
+    name = Column(String)
+    options = Column(Integer, nullable=True)
+
+
@@ -0,0 +1,54 @@
+from datetime import datetime, time, timedelta
+from uuid import UUID
+
+from fastapi import Body, FastAPI,Form, File, UploadFile
+from fastapi.responses import HTMLResponse
+
+app = FastAPI()
+
+
+@app.put("/items/{item_id}")
+async def read_items(
+    item_id: UUID,
+    start_datetime: datetime | None = Body(default=None),
+    end_datetime: datetime | None = Body(default=None),
+    repeat_at: time | None = Body(default=None),
+    process_after: timedelta | None = Body(default=None),
+):
+    start_process = start_datetime + process_after
+    duration = end_datetime - start_process
+    return {
+        "item_id": item_id,
+        "start_datetime": start_datetime,
+        "end_datetime": end_datetime,
+        "repeat_at": repeat_at,
+        "process_after": process_after,
+        "start_process": start_process,
+        "duration": duration,
+    }
+
+@app.post("/files/")
+async def create_files(files: list[bytes] = File()):
+    return {"file_sizes": [len(file) for file in files]}
+
+
+@app.post("/uploadfiles/")
+async def create_upload_files(files: list[UploadFile]):
+    return {"filenames": [file.filename for file in files]}
+
+
+@app.get("/")
+async def main():
+    content = """
+<body>
+<form action="/files/" enctype="multipart/form-data" method="post">
+<input name="files" type="file" multiple>
+<input type="submit">
+</form>
+<form action="/uploadfiles/" enctype="multipart/form-data" method="post">
+<input name="files" type="file" multiple>
+<input type="submit">
+</form>
+</body>
+    """
+    return HTMLResponse(content=content)