loan.py

# -*- coding: utf-8 -*-
from project import app, security
from flask import render_template, request, jsonify
from flask.ext.wtf import Form, TextField, validators

@app.route('/loaned')
@security('user')
def loaned(user=None):
	from project.model.Movie import Movie
	from project.model.Library import Library
	library = Library.objects(user=user,name="Borrowed",unit='Movie').first()
	if not library:
		return render_template('404.html',message='Unable to find given Library',user=user),404
	return render_template('library/library.html',library=library,user=user)
	
@app.route('/send-reminder', methods=['POST'])
@security('user')
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")

@app.route('/loan-movie', methods=['POST'])
@security('user')
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 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="Borrowed").first()
	borrowed_lib.addUnit(movie)

	return jsonify(response='success',type="reload")

@app.route('/return-movie', methods=['POST'])
@security('user')
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="Borrowed").first()
	borrowed_lib.removeUnit(movie)
	loan.delete()

	return jsonify(response='success',type="reload")