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'
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'
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'
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'


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)


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,'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'
read_ticket_select_result = test_env_conn.select_one_value(sql=read_ticket_select_sql)
setattr(ticketStatusData,'read_ticket_id',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'])