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