Create routes for movies

src/config/__init__.py

@@ -1 +1 @@
-from config.settings import get_settings
+from .settings import get_settings

src/database/__init__.py

@@ -1,8 +1,8 @@
-from database.models import (
+from .models import (
     Base,
     MovieModel
 )
-from database.session import (
+from .session import (
     get_db_contextmanager,
     get_db,
     reset_sqlite_database

src/database/populate.py

@@ -3,8 +3,9 @@
 from sqlalchemy.exc import SQLAlchemyError
 from tqdm import tqdm
 
-from config import get_settings
-from database import MovieModel, get_db_contextmanager
+from src.config.settings import get_settings
+from .models import MovieModel
+from .session import get_db_contextmanager
 
 
 class CSVDatabaseSeeder:

src/database/session.py

@@ -3,8 +3,8 @@
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, Session
 
-from config import get_settings
-from database import Base
+from src.config.settings import get_settings
+from src.database.models import Base
 
 settings = get_settings()
 

src/main.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 
-from routes import movie_router
+from src.routes import movie_router
 
 app = FastAPI(
     title="Movies homework",

src/routes/__init__.py

@@ -1 +1 @@
-from routes.movies import router as movie_router
+from .movies import router as movie_router

src/routes/movies.py

@@ -1,10 +1,46 @@
-from fastapi import APIRouter, Depends, HTTPException, Query
+import math
+
+from fastapi import APIRouter, Depends, HTTPException, Query, Request
 from sqlalchemy.orm import Session
 
-from database import get_db, MovieModel
+from src.database.models import MovieModel
+from src.database.session import get_db
+from src.schemas.movies import MovieDetailResponseSchema, MovieListResponseSchema
 
 
 router = APIRouter()
 
 
-# Write your code here
+@router.get("/movies/", response_model=MovieListResponseSchema)
+def get_movies(
+        page: int = Query(ge=1, default=1),
+        per_page: int = Query(ge=1, le=20, default=10),
+        db: Session = Depends(get_db)
+):
+    movie_query = db.query(MovieModel)
+    total_items = movie_query.count()
+    movies = movie_query.offset(per_page * (page - 1)).limit(per_page).all()
+    if not movies:
+        raise HTTPException(status_code=404, detail="No movies found.")
+
+    total_pages = math.ceil(total_items / per_page)
+
+    prev_page = f"/theater/movies/?page={page - 1}&per_page={per_page}" if page > 1 else None
+    next_page = f"/theater/movies/?page={page + 1}&per_page={per_page}" if page < total_pages else None
+
+    response = {
+        "movies": movies,
+        "prev_page": prev_page,
+        "next_page": next_page,
+        "total_pages": total_pages,
+        "total_items": total_items,
+    }
+    return response
+
+
+@router.get("/movies/{movie_id}/", response_model=MovieDetailResponseSchema)
+def get_movie(movie_id: int, db: Session = Depends(get_db)):
+    movie = db.query(MovieModel).filter(MovieModel.id == movie_id).first()
+    if not movie:
+        raise HTTPException(status_code=404, detail="Movie with the given ID was not found.")
+    return movie

src/schemas/movies.py

@@ -1 +1,32 @@
-# Write your code here
+from datetime import date
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class MovieBase(BaseModel):
+    id: int
+    name: str
+    date: date
+    score: float
+    genre: str
+    overview: str
+    crew: str
+    orig_title: str
+    status: str
+    orig_lang: str
+    budget: float
+    revenue: float
+    country: str
+
+
+class MovieDetailResponseSchema(MovieBase):
+    pass
+
+
+class MovieListResponseSchema(BaseModel):
+    movies: list[MovieDetailResponseSchema]
+    prev_page: Optional[str]
+    next_page: Optional[str]
+    total_pages: int
+    total_items: int

src/tests/conftest.py

@@ -1,13 +1,13 @@
 import pytest
 from fastapi.testclient import TestClient
 
-from config import get_settings
-from database import (
+from src.config import get_settings
+from src.database import (
     reset_sqlite_database,
     get_db_contextmanager,
 )
-from database.populate import CSVDatabaseSeeder
-from main import app
+from src.database.populate import CSVDatabaseSeeder
+from src.main import app
 
 
 @pytest.fixture(scope="function", autouse=True)