110 lines
4.9 KiB
Python
110 lines
4.9 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
from app.core.database import get_db
|
|
from app.core.security import get_current_user
|
|
from app.models import User, Song, SongDance, DanceAlternative
|
|
from app.schemas import (
|
|
SongCreate, SongOut,
|
|
SongDanceCreate, SongDanceOut,
|
|
DanceAlternativeCreate, DanceAlternativeOut,
|
|
)
|
|
|
|
router = APIRouter(prefix="/songs", tags=["songs"])
|
|
|
|
|
|
# ── Sange ─────────────────────────────────────────────────────────────────────
|
|
|
|
@router.get("/", response_model=list[SongOut])
|
|
def list_songs(db: Session = Depends(get_db), me: User = Depends(get_current_user)):
|
|
return db.query(Song).filter(Song.owner_id == me.id).all()
|
|
|
|
|
|
@router.post("/", response_model=SongOut, status_code=201)
|
|
def create_song(data: SongCreate, db: Session = Depends(get_db), me: User = Depends(get_current_user)):
|
|
song = Song(owner_id=me.id, **data.model_dump())
|
|
db.add(song)
|
|
db.commit()
|
|
db.refresh(song)
|
|
return song
|
|
|
|
|
|
@router.get("/{song_id}", response_model=SongOut)
|
|
def get_song(song_id: str, db: Session = Depends(get_db), me: User = Depends(get_current_user)):
|
|
song = db.query(Song).filter(Song.id == song_id, Song.owner_id == me.id).first()
|
|
if not song:
|
|
raise HTTPException(404, "Sang ikke fundet")
|
|
return song
|
|
|
|
|
|
@router.delete("/{song_id}", status_code=204)
|
|
def delete_song(song_id: str, db: Session = Depends(get_db), me: User = Depends(get_current_user)):
|
|
song = db.query(Song).filter(Song.id == song_id, Song.owner_id == me.id).first()
|
|
if not song:
|
|
raise HTTPException(404, "Sang ikke fundet")
|
|
db.delete(song)
|
|
db.commit()
|
|
|
|
|
|
# ── Danse på en sang ──────────────────────────────────────────────────────────
|
|
|
|
@router.post("/{song_id}/dances", response_model=SongDanceOut, status_code=201)
|
|
def add_dance(song_id: str, data: SongDanceCreate, db: Session = Depends(get_db), me: User = Depends(get_current_user)):
|
|
song = db.query(Song).filter(Song.id == song_id, Song.owner_id == me.id).first()
|
|
if not song:
|
|
raise HTTPException(404, "Sang ikke fundet")
|
|
dance = SongDance(song_id=song_id, **data.model_dump())
|
|
db.add(dance)
|
|
db.commit()
|
|
db.refresh(dance)
|
|
return dance
|
|
|
|
|
|
@router.delete("/{song_id}/dances/{dance_id}", status_code=204)
|
|
def remove_dance(song_id: str, dance_id: str, db: Session = Depends(get_db), me: User = Depends(get_current_user)):
|
|
song = db.query(Song).filter(Song.id == song_id, Song.owner_id == me.id).first()
|
|
if not song:
|
|
raise HTTPException(404, "Sang ikke fundet")
|
|
dance = db.query(SongDance).filter(SongDance.id == dance_id, SongDance.song_id == song_id).first()
|
|
if not dance:
|
|
raise HTTPException(404, "Dans ikke fundet")
|
|
db.delete(dance)
|
|
db.commit()
|
|
|
|
|
|
# ── Alternativ-danse ──────────────────────────────────────────────────────────
|
|
|
|
@router.post("/{song_id}/dances/{dance_id}/alternatives", response_model=DanceAlternativeOut, status_code=201)
|
|
def add_alternative(song_id: str, dance_id: str, data: DanceAlternativeCreate, db: Session = Depends(get_db), me: User = Depends(get_current_user)):
|
|
song = db.query(Song).filter(Song.id == song_id, Song.owner_id == me.id).first()
|
|
if not song:
|
|
raise HTTPException(404, "Sang ikke fundet")
|
|
dance = db.query(SongDance).filter(SongDance.id == dance_id, SongDance.song_id == song_id).first()
|
|
if not dance:
|
|
raise HTTPException(404, "Dans ikke fundet")
|
|
alt_dance = db.query(SongDance).filter(SongDance.id == data.alt_song_dance_id).first()
|
|
if not alt_dance:
|
|
raise HTTPException(404, "Alternativ-dans ikke fundet")
|
|
|
|
alt = DanceAlternative(song_dance_id=dance_id, **data.model_dump())
|
|
db.add(alt)
|
|
db.commit()
|
|
db.refresh(alt)
|
|
return alt
|
|
|
|
|
|
@router.get("/{song_id}/dances/{dance_id}/alternatives", response_model=list[DanceAlternativeOut])
|
|
def list_alternatives(song_id: str, dance_id: str, db: Session = Depends(get_db), me: User = Depends(get_current_user)):
|
|
dance = db.query(SongDance).filter(SongDance.id == dance_id, SongDance.song_id == song_id).first()
|
|
if not dance:
|
|
raise HTTPException(404, "Dans ikke fundet")
|
|
return dance.alternatives
|
|
|
|
|
|
@router.delete("/{song_id}/dances/{dance_id}/alternatives/{alt_id}", status_code=204)
|
|
def remove_alternative(song_id: str, dance_id: str, alt_id: str, db: Session = Depends(get_db), me: User = Depends(get_current_user)):
|
|
alt = db.query(DanceAlternative).filter(DanceAlternative.id == alt_id, DanceAlternative.song_dance_id == dance_id).first()
|
|
if not alt:
|
|
raise HTTPException(404, "Alternativ ikke fundet")
|
|
db.delete(alt)
|
|
db.commit()
|