def get_pick_result(group_count, club_member_dispatch_flag = False, gender_dispatch_flag = False):
result = {}
all_staff = staff_db.get_all_staff()
for staff in all_staff:
v = random.random()
if club_member_dispatch_flag == True and staff['is_club_member'] == 1:
v = v + 1000
if staff['is_partner'] == 1:
v = v + 100
if gender_dispatch_flag == True and staff['gender'] == 2:
v = v + 10
staff['random_value'] = v
# sort the staff
all_staff.sort(key=lambda x: x['random_value'], reverse = True)
result['group_names'] = []
result['group_count'] = group_count
for i in range(group_count):
group_name = u'第%s组' % (i + 1)
result['group_names'].append(group_name)
index = 0
max_row = 0
result['matrix'] = {}
for staff in all_staff:
if result['matrix'].get(index) is None:
result['matrix'][index] = []
result['matrix'][index].append(staff['name'])
index = index + 1
if index == group_count:
max_row = max_row + 1
index = 0
if index > 0:
max_row = max_row + 1
while index < group_count:
result['matrix'][index].append(None)
index = index + 1
result['max_row'] = max_row
return result
def get_pair_pick_result():
staffs = staff_db.get_all_staff(True)
pairs = staff_db.get_staff_id_mapping()
min_and_max_info = staff_db.get_min_and_max_pair_id()
min_id = min_and_max_info.get('MIN_ID')
max_id = min_and_max_info.get('MAX_ID')
while True:
result = []
occupied_id_list = []
restart_flag = False
print 'start an round'
for item in pairs:
count = 0
while True:
if count >= len(pairs):
restart_flag = True
break
random_value = randint(min_id, max_id)
if random_value != item['pair_id'] and random_value not in occupied_id_list:
item['gift_id'] = random_value
item['staff_name'] = staffs.get(item['staff_id'], {}).get('name')
occupied_id_list.append(random_value)
break
count += 1
if restart_flag:
break
result.append(item)
if not restart_flag:
break
result.sort(key=lambda x: x['gift_id'], reverse = False)
return result
