28 lines
604 B
Python
28 lines
604 B
Python
from sqlalchemy.orm import Session
|
|
|
|
import models
|
|
|
|
# import schemas
|
|
|
|
|
|
def create_movie(db: Session, name: str = ""):
|
|
db_movie = models.Movie(name=name)
|
|
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_movie_by_id(db: Session, id_: str = ""):
|
|
try:
|
|
id_ = int(id_)
|
|
except ValueError:
|
|
pass
|
|
db_movie = db.query(models.Movie).filter(models.Movie.id == id_)
|
|
return db_movie.one()
|