class UserServices():
def __init__(self):
self.db = DatabaseConnection()
self.collection = "users"
def authenticate(self, username, password):
user = self.db.findOne(self.collection, {"username": username})
if user != None:
if user["password"] == password:
return True
else:
return False
else:
return False
def authorize(self, sid):
session = self.db.findOne("sessions", {"sid": sid})
if session:
return session["username"]
else:
return False
def getFirstName(self, username):
user = self.db.findOne(self.collection, {"username": username})
return user["firstName"]
class PropertyService():
def __init__(self):
self.db = DatabaseConnection()
self.collection = "users"
def isUserVendor(self, username):
user = self.db.findOne(self.collection, {"username": username})
if (user['accountType'] == 'vendor'):
return True
else:
return False