def manager_register(cls, email, password, username, _id, role, permission, company, facility): data = Database.find_one('facilities', {'company': company}) role = 'Manager' permission = 100 if data is not None: return False, "company already exist" else: user = cls.get_by_email(email) if not cls.check_id(_id): return False, "bad number ID" if user is None: # User dose'nt exist, create new user Facilities.add_company(company, facility) new_user = cls(email, username, password, _id, role, permission, company, facility) try: new_user.save_to_mongodb() session['email'] = email return True, "SUCCESS" except Exception as e: return False, str(e) else: # User already exist return False, "user email already exist"
def manager_register_simulation(cls, email, password, username, _id, role, permission, company, facility): data = Database.find_oneSimulation('facilities', {'company': company}) role = 'Manager' if data is not None: return False, "company already exist" else: Facilities.add_company_simulation(company, facility) user = cls.get_by_email_simulation(email) if user is None: # User dose'nt exist, create new user new_user = cls(email, username, password, _id, role, permission, company, facility) try: new_user.save_to_mongodb_simulation() session['email'] = email return True, "SUCCESS" except Exception as e: return False, str(e) else: # User already exist return False, "user email already exist"
def user_register_simulation(cls, email, password, username, _id, role, permission, company, facility): if not Facilities.is_company_exist_simulation(company): return False, "company dose'nt exist" user = User.get_by_email_simulation(email) print(email) if user is None: # User dose'nt exist, create new user new_user = User(email, username, password, _id, role, permission, company, facility) try: new_user.save_to_mongodb_simulation() except Exception as e: return False, str(e) return True, '' else: # User already exist return False, "user email already exist"
def test_facilities(): Database.initialize() # add company status = Facilities.is_company_exist('google') assert status != Facilities.add_company('google', 'matam') status = Facilities.is_company_exist('yahoo') assert status != Facilities.add_company('yahoo', 'matam') status = Facilities.is_company_exist('microsoft') assert status != Facilities.add_company('microsoft', 'matam') status = Facilities.is_company_exist('vmware') assert status != Facilities.add_company('vmware', 'matam') status = Facilities.is_company_exist('intel') assert status != Facilities.add_company('intel', 'matam') # add facilities to each company status = Facilities.is_facility_exist('google', 'carmel') assert status != Facilities.add_facility('google', 'carmel') status = Facilities.is_facility_exist('yahoo', 'carmel') assert status != Facilities.add_facility('yahoo', 'carmel') status = Facilities.is_facility_exist('microsoft', 'carmel') assert status != Facilities.add_facility('microsoft', 'carmel') status = Facilities.is_facility_exist('vmware', 'carmel') assert status != Facilities.add_facility('vmware', 'carmel') status = Facilities.is_facility_exist('intel', 'carmel') assert status != Facilities.add_facility('intel', 'carmel') assert 'carmel' in Facilities.get_facilities('intel') assert 'matam' in Facilities.get_facilities('intel') assert 'carmel' in Facilities.get_facilities('vmware') assert 'matam' in Facilities.get_facilities('vmware') Facilities.remove_facility('vmware', 'carmel') assert 'carmel' not in Facilities.get_facilities('vmware')
def add_facility_simulation(self, facility): Facilities.add_facility_simulation(self.company, facility)
def add_facility(self, facility): Facilities.add_facility(self.company, facility)
def get_facilities_simulation(self): return Facilities.get_facilities_simulation(self.company)
def get_facilities(self): return Facilities.get_facilities(self.company)