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)