aswi2022bug-thugs-gitlab/server/sql_app/api/teams.py @ 8fd3d7e5

from typing import List

from fastapi import Depends, FastAPI, HTTPException, APIRouter
from sqlalchemy.orm import Session
from sql_app import crud, models, schemas
from ..database import SessionLocal, engine
from fastapi import FastAPI, Request
from fastapi.responses import HTMLResponse
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates

models.Base.metadata.create_all(bind=engine)
templates = Jinja2Templates(directory="templates/teams")

teams = APIRouter()

# Dependency
def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()


@teams.get("/teams-web/", response_class=HTMLResponse)
async def read_devices(request: Request, skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
    teams = crud.get_teams(db, skip=skip, limit=limit)
    return templates.TemplateResponse("teams.html", {"request": request, "teams": teams})


@teams.post("/team/", response_model=schemas.Team)
def create_device(team: schemas.TeamCreate, db: Session = Depends(get_db)):
    print(crud.create_team(db=db, name=team.name))


@teams.get("/teams/", response_model=List[schemas.Device])
def read_devices(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
    teams = crud.get_teams(db, skip=skip, limit=limit)
    return teams


@teams.get("/team/{team_id}", response_model=schemas.Device)
def read_device(team_id: int, db: Session = Depends(get_db)):
    db_team = crud.get_team(db, team_id=team_id)
    if db_team is None:
        raise HTTPException(status_code=404, detail="Device not found")
    return db_team