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

whatsapp_account_select_sql = f'select * from `whatsapp_phone` where brand_id = {whatsappIntegrationData.brandId} and is_deleted = 0'
whatsapp_account_select_result = test_env_conn.select_many_value(sql=whatsapp_account_select_sql)
if whatsapp_account_select_result:
    whatsapp_account = whatsapp_account_select_result[0]
    whatsapp_display_name = whatsapp_account['verified_name']
    whatsapp_phone_number = whatsapp_account['phone_number']
    whatsapp_status = whatsapp_account['status']
    whatsapp_id = whatsapp_account['id']
    whatsapp_waba_id = whatsapp_account['waba_id']
    waba_account_select_sql = f'select * from `whatsapp_conf` where waba_id = {whatsapp_waba_id}'
    waba_account_select_result = test_env_conn.select_one_value(sql=waba_account_select_sql)
    whatsapp_waba_name = waba_account_select_result['display_name']
    whatsapp_num = len(whatsapp_account_select_result)
else:
    whatsapp_insert_sql = f"""INSERT INTO `whatsapp_phone` (waba_id,brand_id,shop_id,phone_number,verified_name,status,quality_rating,code_verification_status,new_name_status,name_status,messaging_limit_tier,is_deleted,created_at,updated_at) VALUES
		 ('{whatsappIntegrationData.insert_waba_id}',{whatsappIntegrationData.brandId},7314,'{whatsappIntegrationData.insert_whatsapp_phone}','{whatsappIntegrationData.insert_whatsapp_name}','{whatsappIntegrationData.insert_whatsapp_status}','GREEN','EXPIRED','NONE','APPROVED','TIER_250',0,'2024-03-26 16:34:29','2024-04-15 16:32:43');
"""
    waba_insert_sql = f"""INSERT INTO `whatsapp_conf` (brand_id,cust_space_id,waba_id,display_name,currency,account_review_status,is_deleted,created_at,updated_at) VALUES
	 ({whatsappIntegrationData.brandId},'{whatsappIntegrationData.insert_space_id}','{whatsappIntegrationData.insert_waba_id}','{whatsappIntegrationData.insert_waba_name}','USD','APPROVED',0,'2024-03-25 16:24:52','2024-04-15 16:35:46');
"""
    test_env_conn.execute_sql(sql=whatsapp_insert_sql)
    test_env_conn.execute_sql(sql=waba_insert_sql)
    whatsapp_display_name = whatsappIntegrationData.insert_whatsapp_name
    whatsapp_phone_number = whatsappIntegrationData.insert_whatsapp_phone
    whatsapp_status = whatsappIntegrationData.insert_whatsapp_status
    whatsapp_waba_id = whatsappIntegrationData.insert_waba_id
    whatsapp_waba_name = whatsappIntegrationData.insert_waba_name
    whatsapp_account_select_sql = f'select * from `whatsapp_phone` where brand_id = {whatsappIntegrationData.brandId} and verified_name = "{whatsapp_display_name}" and phone_number = "{whatsapp_phone_number}" and is_deleted = 0'
    whatsapp_account_select_result = test_env_conn.select_one_value(sql=whatsapp_account_select_sql)
    whatsapp_id = whatsapp_account_select_result['id']
    whatsapp_num = 1
waba_num_select_sql = f'select * from `whatsapp_conf` where brand_id = {whatsappIntegrationData.brandId}'
waba_num_select_result = test_env_conn.select_many_value(sql=waba_num_select_sql)
waba_num = len(waba_num_select_result)
setattr(whatsappIntegrationData, 'whatsapp_id', whatsapp_id)
setattr(whatsappIntegrationData, 'whatsapp_waba_id', whatsapp_waba_id)
setattr(whatsappIntegrationData, 'whatsapp_waba_name', whatsapp_waba_name)
setattr(whatsappIntegrationData, 'whatsapp_display_name', whatsapp_display_name)
setattr(whatsappIntegrationData, 'whatsapp_phone_number', whatsapp_phone_number)
setattr(whatsappIntegrationData, 'whatsapp_status', whatsapp_status)
setattr(whatsappIntegrationData, 'waba_num', waba_num)
setattr(whatsappIntegrationData, 'whatsapp_num', whatsapp_num)