Develop

src/crud/movies.py

@@ -0,0 +1,59 @@
+from database import get_db
+from fastapi import Depends, Query
+from models import MovieModel
+from schemas import MovieCreateSchema, MovieUpdateSchema
+from sqlalchemy.orm import Session
+
+
+def create_movie(db: Session, movie: MovieCreateSchema):
+    db_movie = MovieModel(**movie.dict())
+    db.add(db_movie)
+    db.commit()
+    db.refresh(db_movie)
+    return db_movie
+
+
+def get_movie(movie_id: int, db: Session = Depends(get_db)):
+    movie = db.query(MovieModel).filter(MovieModel.id == movie_id).first()
+    return movie
+
+
+def get_movies(
+        page: int = Query(default=1, ge=1, description="The page number(>= 1)"),
+        per_page: int = Query(default=10, ge=1, le=20, description="Film quantity at the page (>= 1 и <= 20)"),
+        db: Session = Depends(get_db)
+):
+    total_items = db.query(MovieModel).count()
+    offset = (page - 1) * per_page
+    movies = db.query(MovieModel).order_by(MovieModel.id.desc()).offset(offset).limit(per_page).all()
+    base_url = "/movies/"
+    prev_page = f"{base_url}?page={page - 1}&per_page={per_page}" if page > 1 else None
+    next_page = f"{base_url}?page={page + 1}&per_page={per_page}" if offset + per_page < total_items else None
+    return {
+        "movies": movies,
+        "prev_page": prev_page,
+        "next_page": next_page,
+        "total_pages": (total_items + per_page - 1) // per_page,
+        "total_items": total_items,
+    }
+
+
+def update_movie(db: Session, movie_id: int, movie: MovieUpdateSchema):
+    db_movie = db.query(MovieModel).filter(MovieModel.id == movie_id).first()
+    if not db_movie:
+        return None
+    db_movie.title = movie.title
+    db_movie.genre = movie.genre
+    db_movie.price = movie.price
+    db.commit()
+    db.refresh(db_movie)
+    return db_movie
+
+
+def delete_movie(db: Session, movie_id: int):
+    db_movie = db.query(MovieModel).filter(MovieModel.id == movie_id).first()
+    if not db_movie:
+        return None
+    db.delete(db_movie)
+    db.commit()
+    return db_movie

src/routes/movies.py

@@ -1,12 +1,61 @@
-from fastapi import APIRouter, Depends, HTTPException, Query
-from sqlalchemy.exc import IntegrityError
-from sqlalchemy.orm import Session, joinedload
-
+from src.crud.movies import create_movie, get_movie, get_movies, update_movie, delete_movie
 from database import get_db
-from database.models import MovieModel, CountryModel, GenreModel, ActorModel, LanguageModel
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.orm import Session
 
+from src.database.models import MovieModel
+from src.schemas.movies import MovieReadSchema, MovieCreateSchema, \
+    MovieUpdateSchema
 
 router = APIRouter()
 
 
-# Write your code here
+@router.post("/movies/", response_model=MovieReadSchema)
+def add_movie(movie: MovieCreateSchema, db: Session = Depends(get_db)):
+    existing_movie = (
+        db.query(MovieModel)
+        .filter(MovieModel.name == movie.name, MovieModel.date == movie.date)
+        .first()
+    )
+    if existing_movie:
+        raise HTTPException(
+            status_code=409,
+            detail=f"A movie with the name '{movie.name}' and release date '{movie.date}' already exists."
+        )
+    if existing_movie is None:
+        raise HTTPException(status_code=400)
+    return create_movie(db, movie)
+
+
+@router.get("/movies/", response_model=list[MovieReadSchema], status_code=200)
+def list_movies(db: Session = Depends(get_db)):
+    db_movie = get_movies(db)
+    if not db_movie:
+        raise HTTPException(status_code=404, detail="No movies found for the specified page.")
+    return db_movie
+
+
+@router.get("/movies/{movie_id}", response_model=MovieReadSchema)
+def read_movie(movie_id: int, db: Session = Depends(get_db)):
+    db_movie = get_movie(db, movie_id)
+    if not db_movie:
+        raise HTTPException(status_code=404, detail="Movie with the given ID was not found.")
+    return db_movie
+
+
+@router.put("/movies/{movie_id}", response_model=MovieReadSchema, status_code=200)
+def edit_movie(movie_id: int, movie: MovieUpdateSchema, db: Session = Depends(get_db)):
+    db_movie = update_movie(db, movie_id, movie)
+    if not db_movie:
+        raise HTTPException(status_code=404, detail="Movie with the given ID was not found.")
+    if db_movie is None:
+        raise HTTPException(status_code=400, detail="Invalid input data.")
+    return db_movie
+
+
+@router.delete("/movies/{movie_id}", response_model=MovieReadSchema, status_code=204)
+def remove_movie(movie_id: int, db: Session = Depends(get_db)):
+    db_movie = delete_movie(db, movie_id)
+    if not db_movie:
+        raise HTTPException(status_code=404, detail="Movie with the given ID was not found.")
+    return db_movie

src/schemas/movies.py

@@ -1 +1,103 @@
-# Write your code here
+from datetime import datetime, date, timedelta
+from typing import Optional, List
+
+from pydantic import BaseModel, Field, validator
+
+
+class CountrySchema(BaseModel):
+    name: str
+
+
+class GenreSchema(BaseModel):
+    name: str
+
+
+class ActorSchema(BaseModel):
+    name: str
+
+
+class LanguageSchema(BaseModel):
+    name: str
+
+
+class MovieBaseSchema(BaseModel):
+    id: int
+    name: str
+    date: datetime.date
+    score: Optional[float]
+    genre: Optional[str]
+    overview: Optional[str]
+    crew: Optional[str]
+    orig_title: Optional[str]
+    status: Optional[str]
+    orig_lang: Optional[str]
+    budget: Optional[int]
+    revenue: Optional[float]
+    country: Optional[str]
+
+
+class MovieCreateSchema(MovieBaseSchema):
+    title: str = Field(max_length=255, description="Название фильма (не более 255 символов)")
+    release_date: date = Field(description="Дата выхода фильма")
+    score: int = Field(ge=0, le=100, description="Оценка (0-100)")
+    budget: int = Field(ge=0, description="Бюджет (не может быть отрицательным)")
+    revenue: int = Field(ge=0, description="Доход (не может быть отрицательным)")
+    country: CountrySchema
+    genres: List[GenreSchema]
+    actors: List[ActorSchema]
+    languages: List[LanguageSchema]
+
+    @validator("release_date")
+    def validate_date(cls, value):
+        max_date = date.today() + timedelta(days=365)
+        if value > max_date:
+            raise ValueError("Дата выхода не может превышать текущую дату более чем на 1 год.")
+        return value
+
+
+class MovieUpdateSchema(MovieBaseSchema):
+    score: int = Field(ge=0, le=100, description="Оценка (0-100)")
+    budget: int = Field(ge=0, description="Бюджет (не может быть отрицательным)")
+    revenue: int = Field(ge=0, description="Доход (не может быть отрицательным)")
+
+
+class MovieReadSchema(MovieBaseSchema):
+    id: int = Field(ge=0)
+    genres: List[GenreSchema]
+    actors: List[ActorSchema]
+    languages: List[LanguageSchema]
+
+    class Config:
+        from_attributes = True
+
+    @validator("id")
+    def validate_id(cls, value):
+        if movie_id := value:
+            return movie_id
+
+
+class MovieDetailResponseSchema(BaseModel):
+    id: int
+    name: str
+    date: datetime.date
+    score: Optional[float]
+    genre: Optional[str]
+    overview: Optional[str]
+    crew: Optional[str]
+    orig_title: Optional[str]
+    status: Optional[str]
+    orig_lang: Optional[str]
+    budget: Optional[int]
+    revenue: Optional[float]
+    country: Optional[str]
+
+    class Config:
+        from_attributes: True
+
+
+class MovieListResponseSchema(BaseModel):
+    movies: List[MovieDetailResponseSchema]
+    prev_page: Optional[str]
+    next_page: Optional[str]
+    total_pages: int
+    total_items: int