from Utils.sql_handler import test_env_conn
from TestData.homepage.homepage_test_data import homepageTestData

willdesk_cs_id_select_sql = f'select * from `rel_customerservice_brand` where brand_id = {homepageTestData.willdesk_brand_id} and is_activation = 1'
willdesk_cs_id_select_result = test_env_conn.select_many_value(sql=willdesk_cs_id_select_sql)
cs_num = len(willdesk_cs_id_select_result)
cs = willdesk_cs_id_select_result[0]
cs_id = cs['customer_service_id']

cs_info_select_sql = f'select * from `customer_service` where id = {cs_id}'
cs_info_select_result = test_env_conn.select_one_value(sql=cs_info_select_sql)
cs_name = cs_info_select_result['name']
cs_profile = cs_info_select_result['profile']
cs_customer_type = cs_info_select_result['customer_type']
setattr(homepageTestData, 'cs_id', cs_id)
setattr(homepageTestData, 'cs_name', cs_name)
setattr(homepageTestData, 'cs_profile', cs_profile)
setattr(homepageTestData, 'cs_customer_type', cs_customer_type)
setattr(homepageTestData, 'cs_num', cs_num)