from TestData.inbox.team_test_data import teamData

team_select_sql = f'select * from `resource_item` where brand_id = {teamData.brandId} and resource_id = 1'
team_select_result = teamData.team_db.select_many_value(sql=team_select_sql)

team_num = len(team_select_result)
team_id = team_select_result[0]['id']
team_id2 = team_select_result[1]['id']
team_name = team_select_result[0]['name']
setattr(teamData, 'team_num', team_num)
setattr(teamData, 'team_id', team_id)
setattr(teamData, 'team_name', team_name)
team_room_list_select_sql = f'select room_id from `resource_room_rel` where item_id = {team_id}'
team_room_list_select_result = teamData.team_db.select_many_value(sql=team_room_list_select_sql)
team_room_list = tuple(i['room_id'] for i in team_room_list_select_result)

if team_room_list != ():
    open_room_select_sql = f'select * from `im_room` where room_id in {team_room_list} and status = 1 and room_type != 2 and is_ignore = 0'
    open_room_select_result = teamData.team_db.select_many_value(sql=open_room_select_sql)
    pending_room_select_sql = f'select * from `im_room` where room_id in {team_room_list} and status = 2 and room_type != 2 and is_ignore = 0'
    pending_room_select_result = teamData.team_db.select_many_value(sql=pending_room_select_sql)
    close_room_select_sql = f'select * from `im_room` where room_id in {team_room_list} and status = 3 and room_type != 2 and is_ignore = 0'
    close_room_select_result = teamData.team_db.select_many_value(sql=close_room_select_sql)
    setattr(teamData, 'open_room_num', len(open_room_select_result))
    setattr(teamData, 'pending_room_num', len(pending_room_select_result))
    setattr(teamData, 'close_room_num', len(close_room_select_result))
    other_room_select_sql = f'select * from `im_room` where room_id not in {team_room_list} and site_id = {teamData.shopId} and is_ignore = 0 and room_type != 2 and status = 1 limit 10'
else:
    setattr(teamData, 'open_room_num', 0)
    setattr(teamData, 'pending_room_num', 0)
    setattr(teamData, 'close_room_num', 0)
    other_room_select_sql = f'select * from `im_room` where `cate_id` = {teamData.brandId} and room_type != 2 and site_id = {teamData.shopId} and is_ignore = 0 and status = 1 limit 10'

other_room_select_result = teamData.team_db.select_one_value(sql=other_room_select_sql)
setattr(teamData, 'other_room_id', other_room_select_result['room_id'])
setattr(teamData, 'open_room_num_add' , teamData.open_room_num + 1)

team_member_select_sql = f'select * from `resource_item_cs_rel` where item_id = {team_id}'
team_member_select_result = teamData.team_db.select_many_value(sql=team_member_select_sql)
team_member_num = len(team_member_select_result)
team_member_id = team_member_select_result[0]['cs_id']
setattr(teamData, 'team_member_num', team_member_num)
setattr(teamData, 'team_member_id', team_member_id)
team_member_info_select_sql = f'select * from `customer_service` where id = {team_member_id}'
team_member_info_select_result = teamData.team_db.select_one_value(sql=team_member_info_select_sql)
setattr(teamData, 'team_member_name', team_member_info_select_result['name'])
setattr(teamData, 'team_member_image', team_member_info_select_result['profile'])
setattr(teamData, 'team_member_email', team_member_info_select_result['email'])


team_member_open_room_select_sql = f'select * from `im_room` where room_id in {team_room_list} and room_type != 2 and status = 1 and is_ignore = 0 and allocation_uid = {team_member_id}'
team_member_open_room_select_result = teamData.team_db.select_many_value(sql=team_member_open_room_select_sql)

setattr(teamData, 'team_member_open_room_num', len(team_member_open_room_select_result))

other_room_select_sql2 = f'select * from `im_room` where `cate_id` = {teamData.brandId} and room_type != 2 and site_id = {teamData.shopId} and is_ignore = 0 and status = 1 limit 10'
other_room_select_result2 = teamData.team_db.select_one_value(sql=other_room_select_sql)
setattr(teamData, 'other_room_id2', str(other_room_select_result['room_id']))
other_room_rel_select_sql = f'select * from `team_room_rel` where room_id = {other_room_select_result["room_id"]}'
other_room_rel_select_result = teamData.team_db.select_one_value(sql=other_room_rel_select_sql)
if other_room_rel_select_result:
    setattr(teamData, 'other_room_id2_origin_shop_id', other_room_rel_select_result['team_id'])
else:
    setattr(teamData, 'other_room_id2_origin_shop_id', team_id2)

teamData.team_db.close_db()