def test_people_find_people_not_exist_3(self): people = People() expectedMsg = "can't use find people service" try: result = people.findPeopleByCompanyId("fasdf") self.assertFail() except SystemErrorException as inst: self.assertEqual(inst.message, expectedMsg)
def get(self, person_id): '''get faviourite friuts and vegetables by id of person ''' pattern = re.compile("^[0-9]+$") if bool(pattern.match(person_id)) is False: api.abort(400, "input error") people = People() try: items = people.findFavourite(person_id) except: api.abort(500) return items
def get(self, id): '''get person detail''' pattern = re.compile("^[0-9]{1,6}$") if bool(pattern.match(id)) is False: api.abort(400, 'Invalid Input') people = People() try: index = int(id) except: api.abort(400, 'Invalid Input') result_list = people.getDetailById(index) if -1 == result_list: return '', 204 return result_list
def get(person): cnx = db.connect_to_database() cursor = db.create_cursor(cnx) statement, data = PeopleFactory.get_query(person) cursor.execute(statement, data) results = cursor.fetchall() db.close_cursor(cursor) db.disconnect_database(cnx) for result in results: return People(result) return None
def listEmployeeListByName(self, name): campany_name = name.upper() collection_name = 'companies' db = mng_db[collection_name] listOfCompanies = [] try: result_campany = db.find_one({"company": campany_name}) if result_campany != None: companyId = result_campany['index'] people = People() company_list = people.findPeopleByCompanyId(companyId) return company_list except: raise Exception("system error") if len(listOfCompanies) == 0: return -1 return listOfCompanies
def get(self, person_a_id, person_b_id): '''get common friends between people A and B ''' pattern = re.compile("^[0-9]{1,10}$") if bool(pattern.match(person_a_id)) is False: api.abort(400, "input error") if bool(pattern.match(person_b_id)) is False: api.abort(400, "input error") people = People() person_a_id = int(person_a_id) person_b_id = int(person_b_id) person_a_detail = people.getDetailById(person_a_id) person_b_detail = people.getDetailById(person_b_id) if -1 == person_a_detail or -1 == person_a_detail: return '', 204 friends = people.findCommonFriends(person_a_id, person_b_id) details = { "person_a": { "Name": person_a_detail['name'], "Age": person_a_detail['age'], "Address": person_a_detail['address'], "Phone": person_a_detail['phone'], "index": person_a_detail['index'], }, "person_b": { "Name": person_b_detail['name'], "Age": person_b_detail['age'], "Address": person_b_detail['address'], "Phone": person_b_detail['phone'], "index": person_b_detail['index'], }, 'common_friend': friends } return {"data": details}
def insert_all(movie_data): movie = Movie(movie_data["movie"]) cast = movie_data["cast"] crew = movie_data["crew"] companies = movie_data["companies"] # @return movie_id movie_id = MovieFactory.insert(movie) # @return cast_ids cast_ids = [] for actor in cast: actor = People(actor) actor = (PeopleFactory.insert(actor), RoleFactory.insert("Actor")) cast_ids.append(actor) # @retun crew_ids crew_ids = [] for person in crew: technician = People(person) technician = (PeopleFactory.insert(technician), RoleFactory.insert(person["role"])) crew_ids.append(technician) # @return companies_ids companies_ids = [] for stuff in companies: company = Company(stuff) company = (CompanyFactory.insert(company), RoleFactory.insert("production")) companies_ids.append(company) for actor in cast_ids: RoleFactory.insert_mpr(actor, movie_id) for crew in crew_ids: RoleFactory.insert_mpr(crew, movie_id) for company in companies_ids: RoleFactory.insert_mcr(company, movie_id)
def test_people_find_people_not_exist_2(self): people = People() result = people.findPeopleByCompanyId(9999999) assert(result== -1 )
def test_people_find_people_by_cid(self): people = People() result = people.findPeopleByCompanyId(1) assert(result[0]['index']==289)