from Utils.sql_handler import test_env_conn
from TestData.setting.whatsapp_integration_test_data import whatsappIntegrationData

whatsapp_template_select_sql = f'select * from `whatsapp_template` where brand_id = {whatsappIntegrationData.brandId} and waba_id = {whatsappIntegrationData.whatsapp_waba_id}'
whatsapp_template_select_result = test_env_conn.select_many_value(sql=whatsapp_template_select_sql)
whatsapp_template = whatsapp_template_select_result[0]
whatsapp_template_code = whatsapp_template['template_code']
whatsapp_template_status = whatsapp_template['audit_status']
whatsapp_template_category = whatsapp_template['category']
whatsapp_template_language = whatsapp_template['language']
whatsapp_template_name = whatsapp_template['template_name']
whatsapp_template_content = whatsapp_template['components']
whatsapp_template_example = whatsapp_template['example']
whatsapp_template_num = len(whatsapp_template_select_result)
setattr(whatsappIntegrationData, 'whatsapp_template_num', whatsapp_template_num)
setattr(whatsappIntegrationData, 'whatsapp_template_code', whatsapp_template_code)
setattr(whatsappIntegrationData, 'whatsapp_template_status', whatsapp_template_status)
setattr(whatsappIntegrationData, 'whatsapp_template_category', whatsapp_template_category)
setattr(whatsappIntegrationData, 'whatsapp_template_language', whatsapp_template_language)
setattr(whatsappIntegrationData, 'whatsapp_template_name', whatsapp_template_name)
setattr(whatsappIntegrationData, 'whatsapp_template_content', whatsapp_template_content)
setattr(whatsappIntegrationData, 'whatsapp_template_example', whatsapp_template_example)

whatsapp_active_account_select_sql = f'select * from whatsapp_phone where brand_id = {whatsappIntegrationData.brandId} and is_deleted = 0'
whatsapp_active_account_select_result = test_env_conn.select_many_value(sql=whatsapp_active_account_select_sql)
whatsapp_active_account_num = len(whatsapp_active_account_select_result)
setattr(whatsappIntegrationData, 'whatsapp_active_account_num', whatsapp_active_account_num)
delete_template_select_sql = f'select * from whatsapp_template where brand_id = {whatsappIntegrationData.brandId} and template_name like "%autotest%"'
delete_template_select_result = test_env_conn.select_one_value(sql=delete_template_select_sql)
if delete_template_select_result:
    delete_template_code = delete_template_select_result['template_code']
    setattr(whatsappIntegrationData, 'delete_template_code', delete_template_code)