from Utils.sql_handler import test_env_conn
from TestData.inbox.channel_test_data import channelData

channel_select_sql = f'select * from `resource_item` where brand_id = {channelData.brandId} and resource_id in (7,2,3,4,6)' #2:livechat 3:facebook 4:email 6:ins 7:whatsapp
channel_select_result = test_env_conn.select_many_value(sql=channel_select_sql)
channel_id = channel_select_result[0]['id']
channel_name = channel_select_result[0]['name']
channel_rel_id = channel_select_result[0]['rel_id']
channel_resource_id = channel_select_result[0]['resource_id']
setattr(channelData, 'channel_id', channel_id)
setattr(channelData, 'channel_name', channel_name)
setattr(channelData, 'channel_resource_id', channel_resource_id)

channel_open_room_select_sql = f'select * from `im_room` where site_id = {channel_rel_id} and status = 1 and is_ignore = 0 and room_type != 2'
channel_open_room_select_result = test_env_conn.select_many_value(sql=channel_open_room_select_sql)
setattr(channelData, 'channel_open_room_num', len(channel_open_room_select_result))
setattr(channelData, 'channel_open_room_num_add', len(channel_open_room_select_result) + 1)

customer_service_select_sql = f'select customer_service_id from `rel_customerservice_brand` where brand_id = {channelData.brandId} and is_activation = 1'
customer_service_select_result = test_env_conn.select_many_value(sql=customer_service_select_sql)
customer_service_id_list = [i['customer_service_id'] for i in customer_service_select_result]
customer_service_list_select_sql = f'select * from `customer_service` where id in {tuple(customer_service_id_list)} and customer_type = 0'
customer_service_list_select_result = test_env_conn.select_many_value(sql=customer_service_list_select_sql)

customer_service_id = customer_service_list_select_result[0]['id']
setattr(channelData, 'customer_service_num', len(customer_service_list_select_result))
setattr(channelData, 'customer_service_id', customer_service_id)
setattr(channelData, 'customer_service_name', customer_service_list_select_result[0]['name'])
setattr(channelData, 'customer_service_image', customer_service_list_select_result[0]['profile'])
setattr(channelData, 'customer_service_email', customer_service_list_select_result[0]['email'])

customer_channel_open_room_select_sql = f'select * from `im_room` where site_id = {channel_rel_id} and status = 1 and room_type != 2 and is_ignore = 0 and allocation_uid = {customer_service_id}'
customer_channel_open_room_select_result = test_env_conn.select_many_value(sql=customer_channel_open_room_select_sql)
setattr(channelData, 'customer_channel_open_room_num', len(customer_channel_open_room_select_result))

customer_open_room_select_sql = f'select * from `im_room` where status = 1 and room_type != 2 and is_ignore = 0 and allocation_uid = {customer_service_id}'
customer_open_room_select_result = test_env_conn.select_many_value(sql=customer_open_room_select_sql)
setattr(channelData, 'customer_open_room_num', len(customer_open_room_select_result))
setattr(channelData, 'customer_open_room_num_add', len(customer_open_room_select_result) + 1)


other_room_select_sql = f'select * from `im_room` where allocation_uid != {customer_service_id} and room_type != 2 and cate_id = {channelData.brandId} and status = 1 limit 10'
other_room_select_result = test_env_conn.select_one_value(sql=other_room_select_sql)
setattr(channelData, 'other_room_id', other_room_select_result['room_id'])