from Utils.sql_handler import test_env_conn
from TestData.setting.team_test_data import teamData
def insert_teammate():
    teammate_insert_sql = f"""INSERT INTO `customer_service` (email,password,name,profile,is_update_password,is_open_window,enter_type,customer_type,local_lang,is_delete,create_at,update_at) VALUES
    	 ("{teamData.teammate_email}",'','','',0,1,1,0,'',0,1715397311,1715409940);
    """
    test_env_conn.execute_sql(sql=teammate_insert_sql)
    teammate_id_select_sql = f'select * from customer_service where email = "{teamData.teammate_email}"'
    teammate_id_select_result = test_env_conn.select_one_value(sql=teammate_id_select_sql)
    teammate_id = teammate_id_select_result['id']
    all_conversation_channel_id_select_sql = f'select * from `resource_item` where brand_id = {teamData.brandId} and resource_id = 0'
    all_conversation_channel_id_select_result = test_env_conn.select_one_value(
        sql=all_conversation_channel_id_select_sql)
    all_conversation_channel_id = all_conversation_channel_id_select_result['id']
    teammate_rel_channel_insert_sql = f"""INSERT INTO `resource_item_cs_rel` (item_id,cs_id,sorts,is_hup,is_delete,update_at,create_at) VALUES
    	 ({all_conversation_channel_id},{teammate_id},0,1,0,'2024-05-11 15:03:16','2024-05-11 15:03:16');
    """
    test_env_conn.execute_sql(sql=teammate_rel_channel_insert_sql)
    teammate_brand_rel_insert_sql = f"""INSERT INTO `rel_customerservice_brand` (customer_service_id,brand_id,company_id,role_name,is_activation,is_delete,create_at,update_at) VALUES
	 ({teammate_id},{teamData.brandId},{teamData.companyId},"{teamData.teammate_role}",1,0,1715415656,1715415629);
"""
    test_env_conn.execute_sql(sql=teammate_brand_rel_insert_sql)
    setattr(teamData, 'teammate_id', teammate_id)


team_id_list_select_sql = f'select id from `resource_item` where brand_id = {teamData.brandId} and resource_id in (0,1)'
team_id_list_select_result = test_env_conn.select_many_value(sql=team_id_list_select_sql)
team_id_list = tuple([team_id['id'] for team_id in team_id_list_select_result])

teammate_select_sql = f'select cs_id from `resource_item_cs_rel` where item_id in {team_id_list}'
teammate_select_result = test_env_conn.select_many_value(sql=teammate_select_sql)

if teammate_select_result:
    teammate_id_list = list(set([teammate_id['cs_id'] for teammate_id in teammate_select_result]))
    teammate_num = len(teammate_id_list)
    teammate_id_select_sql = f'select * from customer_service where email = "{teamData.teammate_email}"'
    teammate_id_select_result = test_env_conn.select_one_value(sql=teammate_id_select_sql)
    if teammate_id_select_result:
        teammate_id = teammate_id_select_result['id']
        setattr(teamData, 'teammate_id', teammate_id)
    else:
        insert_teammate()
        teammate_num += 1
else:
    insert_teammate()
    teammate_num = 1

setattr(teamData, 'teammate_num', teammate_num)