73 lines
3.2 KiB
Python
73 lines
3.2 KiB
Python
import logging
|
|
logger = logging.getLogger(__name__)
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlmodel import Session, select
|
|
from typing import List
|
|
from sqlalchemy.exc import NoResultFound
|
|
|
|
from ..model.models import Card, CardCreate, CardUpdate, GroupDB
|
|
from ..services.database import engine, get_session, add_and_refresh
|
|
from ..services.auth import auth_is_admin
|
|
import uuid as gen_uuid
|
|
from app.services.scanner import WriteNewCard, DeleteCard
|
|
|
|
card_router = APIRouter(prefix="/api/v1/cards", tags=["Card"])
|
|
|
|
def register_card(cardInput: CardCreate):
|
|
key, uid = WriteNewCard()
|
|
if key == None:
|
|
logger.info("No card registered. Check logs!")
|
|
raise HTTPException(status.HTTP_417_EXPECTATION_FAILED, detail="No card registered. Check logs!")
|
|
card = Card(
|
|
group_id=cardInput.group_id,
|
|
key=key,
|
|
name=cardInput.name,
|
|
card_serial=uid,
|
|
enabled=cardInput.enabled
|
|
)
|
|
return card
|
|
|
|
@card_router.post("/", response_model=Card)
|
|
def add_card(cardInput: CardCreate, db: Session = Depends(get_session), admin: bool = Depends(auth_is_admin)):
|
|
try: assert db.exec(select(Card).where(Card.name == cardInput.name)).one_or_none() == None
|
|
except AssertionError:
|
|
raise HTTPException(status.HTTP_409_CONFLICT, detail="Name already used!")
|
|
try: assert db.exec(select(GroupDB).where(GroupDB.id == cardInput.group_id)).one_or_none() is not None
|
|
except AssertionError:
|
|
raise HTTPException(status.HTTP_404_NOT_FOUND, detail="GroupID not found!")
|
|
card = register_card(cardInput)
|
|
return add_and_refresh(db, card)
|
|
|
|
@card_router.delete("/")
|
|
def del_card(*, db: Session = Depends(get_session), admin: bool = Depends(auth_is_admin)):
|
|
key = DeleteCard()
|
|
logger.info(key)
|
|
try:
|
|
card = db.exec(select(Card).where(Card.key == key)).one()
|
|
except NoResultFound:
|
|
logger.info(f"The key:'{key}' was not found in db!")
|
|
raise HTTPException(status.HTTP_500_INTERNAL_SERVER_ERROR, detail="Key on card not found in DB. Please tell an admin about this. KEY={key}")
|
|
db.delete(card)
|
|
db.commit()
|
|
return {"message": "Card deleted successfully"}
|
|
|
|
@card_router.get("/{group_id}", response_model=List[Card])
|
|
def get_cards(*, db: Session = Depends(get_session), group_id: int, admin: bool = Depends(auth_is_admin)):
|
|
cards = db.exec(select(Card).where(Card.group_id == group_id)).all()
|
|
return cards
|
|
|
|
@card_router.patch("/{card_id}", response_model=Card)
|
|
def update_card(card_id: int, cardInput: CardUpdate, db: Session = Depends(get_session), admin: bool = Depends(auth_is_admin)):
|
|
db_card = db.get(Card, card_id)
|
|
if db_card is None:
|
|
raise HTTPException(status.HTTP_404_NOT_FOUND, detail="Card not found!")
|
|
card_data = cardInput.model_dump(exclude_unset=True, exclude_none=True)
|
|
print(card_data)
|
|
if "group_id" in card_data:
|
|
print("test")
|
|
print(db.exec(select(GroupDB).where(GroupDB.id == cardInput.group_id)).one_or_none())
|
|
try: assert db.exec(select(GroupDB).where(GroupDB.id == cardInput.group_id)).one_or_none() is not None
|
|
except AssertionError:
|
|
raise HTTPException(status.HTTP_404_NOT_FOUND, detail="GroupID not found!")
|
|
db_card.sqlmodel_update(card_data)
|
|
return add_and_refresh(db, db_card) |