from Utils.sql_handler import test_env_conn
from TestData.customer.customer_test_data import customerData

customer_select_sql = f'select * from `customer` where brand_id = {customerData.brandId} and is_visitor = 0 and email != "" and email not like "%@willdeskvisitor.com%" order by create_at desc'
customer_select_result = test_env_conn.select_many_value(sql=customer_select_sql)

customer_num = len(customer_select_result)
customer = customer_select_result[0]
customer_id = customer['id']
customer_email = customer['email']
customer_name = customer['name']
customer_phone = customer['phone']
customer_location = customer['location']
customer_timezone = customer['time_zone']
customer_tag_select_sql = f'select * from `tag` where brand_id = {customerData.brandId}'
customer_tag_select_result = test_env_conn.select_one_value(sql=customer_tag_select_sql)
customer_tag_id = customer_tag_select_result['id']
customer_tag_name = customer_tag_select_result['name']

setattr(customerData, 'customer_num', customer_num)
setattr(customerData, 'customer_id', customer_id)
setattr(customerData, 'customer_email', customer_email)
setattr(customerData, 'customer_name', customer_name)
setattr(customerData, 'customer_phone', customer_phone)
setattr(customerData, 'customer_location', customer_location)
setattr(customerData, 'customer_timezone', customer_timezone)
setattr(customerData, 'customer_tag_id', customer_tag_id)
setattr(customerData, 'customer_tag_name', customer_tag_name)