Removed Home and Test models and minor additions to Group Movie and User models

project/model/Group.py

@@ -3,3 +3,19 @@ from project import db
 class Group(db.Document):
 	category = db.StringField(max_length=50, required=True)
 	tags = db.ListField(db.StringField(max_length=50))
+
+	def addTag(self,tag):
+		if tag not in self.tags:
+			self.tags.append(tag)
+		return self
+
+	def removeTag(self,tag):
+		if tag in self.tags:
+			self.tags.remove(tag)
+		return self    
+
+	def __str__(self):
+		return "Category: %s - %s" % (self.category, self.tags)
+
+	def __repr__(self):
+		return "{%s}" % (self.category)
\ No newline at end of file

project/model/Home.py

-# -*- coding: utf-8 -*-
-from flask import flash
-
-
-class Home(object):
-
-    def show_string(self, text):
-        if text == '':
-            flash("You didn't enter any text to flash")
-        else:
-            flash(text + "!!!")

project/model/Movie.py

@@ -8,9 +8,15 @@ class Movie(db.Document):
     tags = db.ListField(db.StringField(max_length=50))
 
     def addTag(self,tag):
+    	if tag not in self.tags:
         	self.tags.append(tag)
         return self
 
+    def removeTag(self,tag):
+    	if tag in self.tags:
+        	self.tags.remove(tag)
+        return self    
+
     def __str__(self):
     	return self.title
 

project/model/Test.py

-# -*- coding: utf-8 -*-
-from flask import flash
-
-
-class Test(object):
-
-    def show_string(self, text):
-        if text == '':
-            flash("You didn't enter any text to flash")
-        else:
-            flash(text + "!!!")
\ No newline at end of file

project/model/User.py

@@ -12,9 +12,20 @@ class User(db.Document):
 		return role in self.roles
 
 	def addRole(self,role):
+		if role not in self.roles:
 			self.roles.append(role)
 		return self
 
+	def removeRole(self,role):
+		if role in self.roles:
+			self.roles.remove(role)
+		return self	
+
+	def changePassword(self,password):
+		self.password =	hashlib.sha224(self.salt.join(password)).hexdigest()
+		self.save()
+		return self
+
 	@staticmethod
 	def createUser(email, password):
 		user = User(email=email)