From f24eab1895102af7ee7106bef12df1962591b926 Mon Sep 17 00:00:00 2001
From: Brian Wood <legion.the.unforgiven@gmail.com>
Date: Sun, 20 Apr 2014 23:07:58 -0600
Subject: [PATCH] Added Loan model

---
 project/model/Loan.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 project/model/Loan.py

diff --git a/project/model/Loan.py b/project/model/Loan.py
new file mode 100644
index 0000000..7247cad
--- /dev/null
+++ b/project/model/Loan.py
@@ -0,0 +1,28 @@
+from project import db
+import datetime
+from User import User
+from Movie import Movie
+
+class Loan(db.Document):
+	user = db.ReferenceField(User)
+	movie = db.ReferenceField(Movie, unique_with=['user'])
+	lent_date = db.DateTimeField(default=datetime.datetime.now, required=True)
+	expected_return_date = db.DateTimeField(default=datetime.datetime.now, required=True)
+	borrower_email = db.StringField()
+
+	@staticmethod
+	def create(user,movie,email,expected_return_date=None):
+		info = Loan(user=user,movie=movie,email=email)
+		if expected_return_date:
+			info.expected_return_date = expected_return_date
+		else:
+			info.expected_return_date = info.expected_return_date + datetime.timedelta(days=7)
+		info.save()
+		return info
+
+
+	def __str__(self):
+		return "%s due %s" % (self.movie.title, self.expected_return_date.isoformat())
+
+	def __repr__(self):
+		return self.__str__()
\ No newline at end of file
-- 
GitLab