from Utils.sql_handler import test_env_conn
from TestData.automation.template_test_data import templateData
template_parent_select_sql = f'select * from rule_template where status = 1'
template_parent_select_result = test_env_conn.select_many_value(sql=template_parent_select_sql)
for i in template_parent_select_result:
    if i['tpl_category'] == 'autoanswer':
        workflow_parent_template_id = i['tpl_id']
        setattr(templateData, 'workflow_parent_template_id', workflow_parent_template_id)
    elif i['tpl_category'] == 'willdesk':
        rule_parent_template_id = i['tpl_id']
        setattr(templateData, 'rule_parent_template_id', rule_parent_template_id)
workflow_template_select_sql = f'select * from rule_tpl_usecase where tpl_id = "{workflow_parent_template_id}" and status = 1'
workflow_template_select_result = test_env_conn.select_many_value(sql=workflow_template_select_sql)
workflow_template_num = len(workflow_template_select_result)
workflow_template_id = workflow_template_select_result[0]['tpld_id']
workflow_template_title = workflow_template_select_result[0]['title']
workflow_template_describe = workflow_template_select_result[0]['describe']
workflow_template_event_id = workflow_template_select_result[0]['event_id']

setattr(templateData, 'workflow_template_num', workflow_template_num)
setattr(templateData, 'workflow_template_id', workflow_template_id)
setattr(templateData, 'workflow_template_title', workflow_template_title)
setattr(templateData, 'workflow_template_describe', workflow_template_describe)
setattr(templateData, 'workflow_template_event_id', workflow_template_event_id)

rule_template_select_sql = f'select * from rule_tpl_usecase where tpl_id = "{rule_parent_template_id}" and status = 1'
rule_template_select_result = test_env_conn.select_many_value(sql=rule_template_select_sql)
rule_template_num = len(rule_template_select_result)
rule_template_id = rule_template_select_result[0]['tpld_id']
rule_template_title = rule_template_select_result[0]['title']
rule_template_describe = rule_template_select_result[0]['describe']
rule_template_event_id = rule_template_select_result[0]['event_id']

setattr(templateData, 'rule_template_num', rule_template_num)
setattr(templateData, 'rule_template_id', rule_template_id)
setattr(templateData, 'rule_template_title', rule_template_title)
setattr(templateData, 'rule_template_describe', rule_template_describe)
setattr(templateData, 'rule_template_event_id', rule_template_event_id)