aswi2022bug-thugs-gitlab/server/sql_app/api/teams_web.py @ 0fcb708f

from typing import List

from fastapi import Depends, FastAPI, HTTPException, APIRouter, Form
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, RedirectResponse
from fastapi_jwt_auth import AuthJWT
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates

models.Base.metadata.create_all(bind=engine)

# Path to html templates used in this file
templates = Jinja2Templates(directory="templates/teams")

# prefix used for all endpoints in this file
teams_web = APIRouter(prefix="")


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


@teams_web.get("/teams-web", response_class=HTMLResponse)
async def read_devices(request: Request, skip: int = 0, limit: int = 100, db: Session = Depends(get_db),
                       Authorize: AuthJWT = Depends()):
    """
    Returns template with all teams currently saved in database
    """
    Authorize.jwt_optional()
    current_user = Authorize.get_jwt_subject()
    teams = crud.get_teams(db, skip=skip, limit=limit)
    if current_user == "admin":
        return templates.TemplateResponse("teams.html", {"request": request, "teams": teams, "user": current_user})
    else:
        current_user = "guest"
        return templates.TemplateResponse("teams_normal.html", {"request": request, "teams": teams, "user": current_user})


@teams_web.get("/team-create", response_class=HTMLResponse)
async def team_create_web(request: Request, Authorize: AuthJWT = Depends()):
    """
    Returns template with form for creating new team
    """
    Authorize.jwt_optional()
    current_user = Authorize.get_jwt_subject()
    if current_user != "admin":
        return RedirectResponse(url=f"/logs-web", status_code=303)
    return templates.TemplateResponse("team_create.html", {"request": request})


@teams_web.post("/teams-web-con")
def create_team(name: str = Form(...), db: Session = Depends(get_db), Authorize: AuthJWT = Depends()):
    """
    Endpoint called from within form for creating new team. Creates new team and redirects to view with all teams
    """
    Authorize.jwt_optional()
    current_user = Authorize.get_jwt_subject()
    if current_user != "admin":
        return RedirectResponse(url=f"/logs-web", status_code=303)
    teams = crud.get_teams(db, 0, 100)
    teams_names = []
    for t in teams:
        teams_names.append(t.name)
    if name not in teams_names:
        team = crud.create_team(db, name)
        if team is None:
            print("something went wrong")
    return RedirectResponse(url=f"/teams-web", status_code=303)


@teams_web.get("/team-change/{team_id}", response_class=HTMLResponse)
async def team_change_web(request: Request, team_id: int, db: Session = Depends(get_db),
                          Authorize: AuthJWT = Depends()):
    """
    Returns template with form for changing teams name
    """
    Authorize.jwt_optional()
    current_user = Authorize.get_jwt_subject()
    if current_user != "admin":
        return RedirectResponse(url=f"/logs-web", status_code=303)
    team = crud.get_team(db, team_id)
    return templates.TemplateResponse("team_change.html", {"request": request, "team": team})

@teams_web.post("/teams-change-process/{team_id}")
async def team_change_process(team_id: int, db:Session = Depends(get_db), name: str = Form(...),
                              Authorize: AuthJWT = Depends()):
    """
    Changes teams name to a new one given by user
    """
    Authorize.jwt_optional()
    current_user = Authorize.get_jwt_subject()
    if current_user != "admin":
        return RedirectResponse(url=f"/logs-web", status_code=303)
    team = crud.change_team(db, team_id, name)
    return RedirectResponse(url=f"/teams-web", status_code=303)