from Utils.sql_handler import test_env_conn
from TestData.inbox.ticketStatus_test_data import ticketStatusData

all_ticket_select_sql = f'select * from `im_room` where `cate_id` = {ticketStatusData.brandId} and `status` = 1 and `is_ignore` = 0 and (`room_type` in (0,1,3) or (`room_type` = 3 and `ai_handle_status` = 100))'
your_ticket_select_sql = f'select * from `im_room` where `cate_id` = {ticketStatusData.brandId} and `status` = 1 and `allocation_uid` = {ticketStatusData.userId} and `is_ignore` = 0 and (`room_type` in (0,1,3) or (`room_type` = 3 and `ai_handle_status` = 100))'
unread_ticket_select_sql = f'select * from `im_room` where `cate_id` = {ticketStatusData.brandId} and `status` = 1 and `read_status` = 1 and `is_ignore` = 0 and (`room_type` in (0,1,3) or (`room_type` = 3 and `ai_handle_status` = 100))'
unassigned_ticket_select_sql = f'select * from `im_room` where `cate_id` = {ticketStatusData.brandId} and `status` = 1 and `allocation_uid` = 0 and `is_ignore` = 0 and (`room_type` in (0,1,3) or (`room_type` = 3 and `ai_handle_status` = 100))'
chatbot_ticket_select_sql = f'select * from `im_room` where `cate_id` = {ticketStatusData.brandId} and `status` = 1 and `is_ignore` = 0 and room_type in (2,3)'


all_ticket_num = len(test_env_conn.select_many_value(sql=all_ticket_select_sql))
your_ticket_num = len(test_env_conn.select_many_value(sql=your_ticket_select_sql))
unread_ticket_num = test_env_conn.select_many_value(sql=unread_ticket_select_sql)
unassigned_ticket_num = test_env_conn.select_many_value(sql=unassigned_ticket_select_sql)
chatbot_ticket_num = len(test_env_conn.select_many_value(sql=chatbot_ticket_select_sql))

setattr(ticketStatusData,'all_ticket_num',all_ticket_num)
setattr(ticketStatusData,'your_ticket_num',your_ticket_num)
setattr(ticketStatusData,'your_ticket_num_add',your_ticket_num+1)
setattr(ticketStatusData,'unread_ticket_num',len(unread_ticket_num))
setattr(ticketStatusData,'unread_ticket_num_add',len(unread_ticket_num)+1)
setattr(ticketStatusData,'unassigned_ticket_num',len(unassigned_ticket_num))
setattr(ticketStatusData,'unassigned_ticket_num_minus',len(unassigned_ticket_num)-1)
setattr(ticketStatusData,'chatbot_ticket_num',chatbot_ticket_num)
setattr(ticketStatusData,'unassigned_ticket_id',unassigned_ticket_num[0]['room_id'])

read_ticket_select_sql = f'select * from `im_room` where `cate_id` = {ticketStatusData.brandId} and `status` = 1 and `read_status` = 2 and `room_platform` = "willdesk" and `is_ignore` = 0 and `site_id` = {ticketStatusData.shopId} and room_type != 2'
read_ticket_select_result = test_env_conn.select_one_value(sql=read_ticket_select_sql)
setattr(ticketStatusData,'read_ticket_id',str(read_ticket_select_result["room_id"]))

msg_select_sql = f'select * from `im_record` where `room_id` = {read_ticket_select_result["room_id"]} order by send_time limit 1'
msg_select_result = test_env_conn.select_one_value(sql=msg_select_sql)
setattr(ticketStatusData,'unread_msg_id',msg_select_result['msg_id'])
setattr(ticketStatusData,'unread_send_time',msg_select_result['send_time'])