from Utils.sql_handler import test_env_conn
from TestData.setting.email_integration_test_data import emailIntegrationData

email_select_sql = f'select * from `email_config` where brand_id = {emailIntegrationData.brandId}'
email_select_result = test_env_conn.select_many_value(sql=email_select_sql)
email_insert_sql = f"""INSERT INTO email_config (brand_id,email_servicer_id,email,email_name,auth_code,refresh_token,raw_token,code_type,imap_addr,imap_port,imap_ssl,smtp_addr,smtp_port,smtp_ssl,enable_signature,signature_content,auth_folder,connect_status,failed_reason,status,create_at,update_at) VALUES
 ({emailIntegrationData.brandId},7,'{emailIntegrationData.append_email_address}','{emailIntegrationData.append_email_name}','mFmir6+EXv7TMXstfilhdjOJ3YeDJ/O+qjWQnp3540k=','','',0,'imap.exmail.qq.com',993,1,'smtp.exmail.qq.com',465,1,1,'<p>Best wishes&nbsp;</p><span class="agent-name unique-tags" contenteditable="false">Agent name</span> | <span>Customer Support</span>','1',{emailIntegrationData.append_email_status},'',1,'2024-03-20 18:45:25','2024-03-20 18:50:02');
"""
if email_select_result:
    email = email_select_result[0]
    email_id = email['id']
    email_name = email['email_name']
    email_address = email['email']
    email_status = email['connect_status']
    email_num = len(email_select_result)
else:
    test_env_conn.execute_sql(sql=email_insert_sql)
    email_address = emailIntegrationData.append_email_address
    email_name = emailIntegrationData.append_email_name
    email_status = emailIntegrationData.append_email_status
    email_select_sql = f'select * from `email_config` where brand_id = {emailIntegrationData.brandId} and email = "{email_address}"'
    email_select_result = test_env_conn.select_one_value(sql=email_select_sql)
    email_id = email_select_result['id']
    email_num = 1

setattr(emailIntegrationData, 'email_id', email_id)
setattr(emailIntegrationData, 'email_address', email_address)
setattr(emailIntegrationData, 'email_name', email_name)
setattr(emailIntegrationData, 'email_status', email_status)
setattr(emailIntegrationData, 'email_num', email_num)