63 lines
2.4 KiB
Python
63 lines
2.4 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, 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
|
|
|
|
|
|
#TODO:
|
|
# -Split Authorisations + Cards
|
|
# -Deactivation
|
|
# -Deleting |