36 lines
882 B
Python
36 lines
882 B
Python
import sqlalchemy
|
|
from sqlalchemy.orm import Session
|
|
import models
|
|
|
|
# import schemas
|
|
|
|
|
|
def create_movie(db: Session, name: str, genres: list[str], description: str = ""):
|
|
db_movie = models.Movie(name=name, genres=str(genres), description=description)
|
|
db.add(db_movie)
|
|
db.commit()
|
|
db.refresh(db_movie)
|
|
return db_movie
|
|
|
|
|
|
def get_movie_by_name(db: Session, name: str = ""):
|
|
db_movie = db.query(models.Movie).filter(models.Movie.name == name)
|
|
return db_movie.all()
|
|
|
|
|
|
def get_all_movies(db: Session):
|
|
db_movie = db.query(models.Movie)
|
|
return db_movie.all()
|
|
|
|
|
|
def get_movie_by_id(db: Session, id_: str = ""):
|
|
try:
|
|
id_ = int(id_)
|
|
except ValueError:
|
|
pass
|
|
try:
|
|
db_movie = db.query(models.Movie).filter(models.Movie.id == id_).one()
|
|
except sqlalchemy.exc.NoResultFound:
|
|
raise LookupError
|
|
return db_movie
|