def addlibraryItem(name,user=None): from project.model.Movie import Movie library = Library.objects(user=user,name=name,unit='Movie').first() if not library: return jsonify(response='error',message='Unable to find the given Library'),404 movie_id = request.form['id'] if not movie_id: return jsonify(response='error',message='Invalid Movie given'),404 from project.model.Movie import Movie movie = Movie.objects(tmdb_id=movie_id).first() if movie: if library.name != 'Master': master = Library.objects(user=user,name="Master",unit='Movie').first() master.addUnit(movie) library.addUnit(movie) return jsonify(response='success',type='redirect',path=url_for(endpoint='library',name=name,_external=True)) from tmdb3 import Movie as tmdbMovie movie = tmdbMovie(movie_id) if not movie: return jsonify(response='error',message='Invalid Movie given'),404 from project.model.Movie import Movie movie = Movie.convertMovie(movie) library.addUnit(movie) if library.name != 'Master': master = Library.objects(user=user,name="Master",unit='Movie').first() master.addUnit(movie) return jsonify(response='success',type='redirect',path=url_for(endpoint='library',name=name,_external=True))
def createLoan(user=None):
from datetime import datetime
borrower = request.form["email"]
return_date = request.form["date"] or None
movie_id = request.form["movie"]
if return_date:
return_date = datetime.strptime(return_date, "%m/%d/%Y")
if return_date <= datetime.now():
return jsonify(response="error", message="Return date must be in the future"), 404
if not movie_id:
return jsonify(response="error", message="Invalid Movie given"), 404
from project.model.Movie import Movie
movie = Movie.objects(id=movie_id).first()
if not movie:
return jsonify(response="error", message="Invalid Movie given"), 404
from project.model.Loan import Loan
loan = Loan.objects(user=user, movie=movie).first()
if loan:
return jsonify(response="error", message="A loan already exists for this movie"), 404
loan = Loan.create(user, movie, borrower, return_date)
from project.model.Library import Library
borrowed_lib = Library.objects(user=user, unit="Movie", name="Loaned").first()
borrowed_lib.addUnit(movie)
return jsonify(response="success", type="reload")
def returnMovie(user=None):
movie_id = request.form['id']
if not movie_id:
return jsonify(response='error', message='Invalid Movie given'), 404
from project.model.Movie import Movie
movie = Movie.objects(id=movie_id).first()
if not movie:
return jsonify(response='error', message='Invalid Movie given'), 404
from project.model.Loan import Loan
loan = Loan.objects(user=user, movie=movie).first()
if not loan:
return jsonify(
response='error',
message='This movie has not been loaned out and cannot be returned'
), 404
from project.model.Library import Library
borrowed_lib = Library.objects(user=user, unit="Movie",
name="Loaned").first()
borrowed_lib.removeUnit(movie)
loan.delete()
return jsonify(response='success', type="reload")
def createLoan(user=None):
from datetime import datetime
borrower = request.form['email']
return_date = request.form['date'] or None
movie_id = request.form['movie']
if return_date:
return_date = datetime.strptime(return_date, '%m/%d/%Y')
if return_date <= datetime.now():
return jsonify(response='error',
message='Return date must be in the future'), 404
if not movie_id:
return jsonify(response='error', message='Invalid Movie given'), 404
from project.model.Movie import Movie
movie = Movie.objects(id=movie_id).first()
if not movie:
return jsonify(response='error', message='Invalid Movie given'), 404
from project.model.Loan import Loan
loan = Loan.objects(user=user, movie=movie).first()
if loan:
return jsonify(response='error',
message='A loan already exists for this movie'), 404
loan = Loan.create(user, movie, borrower, return_date)
from project.model.Library import Library
borrowed_lib = Library.objects(user=user, unit="Movie",
name="Loaned").first()
borrowed_lib.addUnit(movie)
return jsonify(response='success', type="reload")
def reminderEmail(user=None):
import smtplib
from_addr = user.email
subject = request.form['subject'] or "Movie Return Reminder"
movie_id = request.form['movie']
if not movie_id:
return jsonify(response='error', message='Invalid Movie given'), 404
from project.model.Movie import Movie
movie = Movie.objects(id=movie_id).first()
if not movie:
return jsonify(response='error', message='Invalid Movie given'), 404
loan = movie.getLoan(user)
if not loan:
return jsonify(response='error', message='Invalid Movie given'), 404
to_addr = loan.borrower_email
message = request.form[
'message'] or "The movie %s, that you borrowed from %s has asked that you return the movie by %s" % (
movie.title, user.email,
loan.expected_return_date.strftime('%m-%d-%Y'))
login = app.config['SMTP_USER']
password = app.config['SMTP_PASSWORD']
smtpserver = app.config['SMTP_SERVER']
header = 'From: %s\n' % from_addr
header += 'To: %s\n' % to_addr
header += 'Subject: %s\n\n' % subject
message = header + message
server = smtplib.SMTP(smtpserver)
server.starttls()
server.login(login, password)
problems = server.sendmail(from_addr, to_addr, message)
server.quit()
return jsonify(response='success', type="reload")
def reminderEmail(user=None):
import smtplib
from_addr = user.email
subject = request.form["subject"] or "Movie Return Reminder"
movie_id = request.form["movie"]
if not movie_id:
return jsonify(response="error", message="Invalid Movie given"), 404
from project.model.Movie import Movie
movie = Movie.objects(id=movie_id).first()
if not movie:
return jsonify(response="error", message="Invalid Movie given"), 404
loan = movie.getLoan(user)
if not loan:
return jsonify(response="error", message="Invalid Movie given"), 404
to_addr = loan.borrower_email
message = request.form[
"message"
] or "The movie %s, that you borrowed from %s has asked that you return the movie by %s" % (
movie.title,
user.email,
loan.expected_return_date.strftime("%m-%d-%Y"),
)
login = app.config["SMTP_USER"]
password = app.config["SMTP_PASSWORD"]
smtpserver = app.config["SMTP_SERVER"]
header = "From: %s\n" % from_addr
header += "To: %s\n" % to_addr
header += "Subject: %s\n\n" % subject
message = header + message
server = smtplib.SMTP(smtpserver)
server.starttls()
server.login(login, password)
problems = server.sendmail(from_addr, to_addr, message)
server.quit()
return jsonify(response="success", type="reload")
def returnMovie(user=None):
movie_id = request.form["id"]
if not movie_id:
return jsonify(response="error", message="Invalid Movie given"), 404
from project.model.Movie import Movie
movie = Movie.objects(id=movie_id).first()
if not movie:
return jsonify(response="error", message="Invalid Movie given"), 404
from project.model.Loan import Loan
loan = Loan.objects(user=user, movie=movie).first()
if not loan:
return jsonify(response="error", message="This movie has not been loaned out and cannot be returned"), 404
from project.model.Library import Library
borrowed_lib = Library.objects(user=user, unit="Movie", name="Loaned").first()
borrowed_lib.removeUnit(movie)
loan.delete()
return jsonify(response="success", type="reload")