crud.py

"""CRUD operations."""

from model import db, User, Movie, Rating, connect_to_db
from datetime import datetime


# Functions start here!
if __name__ == '__main__':
    from server import app
    connect_to_db(app)
def create_user(email, password):
    
    """Create and return a new user."""

    user = User(email= email, password = password)

    db.session.add(user)
    db.session.commit()
    return user

def create_movie(title, overview, release_date, poster_path):
    movie = Movie(title=title, overview=overview, release_date=release_date, poster_path=poster_path)

    db.session.add(movie)
    db.session.commit()
    return movie

def create_rating(user,movie,score):
    rating = Rating(user = user, movie = movie, score = score)

    db.session.add(rating)
    db.session.commit()
    return rating

def return_movie():
    return Movie.query.all()

def get_movie_by_id(movie_id):
    return Movie.query.get(movie_id)

def return_user():
    return User.query.all()

def get_user_by_id(user_id):
    return User.query.get(user_id)

def get_user_by_email(email):
    return User.query.filter(User.email == email).first()

def login_user(email, password):
    return User.query.filter((User.email == email) & (User.password == password)).first()

def get_rating(score):
    return Rating.query.get(score)