from TestData.setting.macro_test_data import macroData
from Utils.mockData_handler import data_handler

macro_select_sql = f'select * from `macro` where brand_id = {macroData.brandId} and deleted_at is NULL'
macro_select_result = macroData.macro_db.select_many_value(sql=macro_select_sql)
if macro_select_result:
    macro_id_list = tuple([i['macro_id'] for i in macro_select_result])
    macro_num_select_sql = f'select * from `macro_permission` where macro_id in {macro_id_list}'
    macro_num_select_result = macroData.macro_db.select_many_value(sql=macro_num_select_sql)
    team_macro_num = 0
    brand_macro_num = 0
    user_macro_num = 0
    for macro in macro_num_select_result:
        if macro['perm_type'] == 1:
            brand_macro_num += 1
        elif macro['perm_type'] == 2:
            user_macro_num += 1
        elif macro['perm_type'] == 3:
            team_macro_num += 1
    setattr(macroData,'brand_macro_num',brand_macro_num)
    setattr(macroData,'user_macro_num',user_macro_num)
    setattr(macroData,'team_macro_num',team_macro_num)
    for i in macro_select_result:
        if i['title'] == macroData.append_macro_title:
            append_macro_id = i['macro_id']
            break
    macro_id = macro_select_result[0]['macro_id']
    macro_title = macro_select_result[0]['title']
    macro_content = macro_select_result[0]['content']
    macro_sorts = macro_select_result[0]['sorts']
    macro_clicks = macro_select_result[0]['clicks']
    macro_brand_id = [macro_select_result[0]['brand_id']]
    macro_create_by_name = macro_select_result[0]['create_by_name']
    macro_create_by_id = macro_select_result[0]['create_by_uid']
else:
    append_macro_id = data_handler.random_id()
    macro_insert_sql = f'''INSERT INTO `macro` (macro_id,title,content,sorts,clicks,brand_id,create_by_name,create_by_uid,deleted_at,update_at,create_at) VALUES
	 ('{append_macro_id}','{macroData.append_macro_title}','{macroData.append_macro_content}',1,0,{macroData.brandId},'{macroData.userName}',{macroData.userId},NULL,'2024-04-23 14:54:31','2024-04-23 14:54:31');
'''
    macroData.macro_db.execute_sql(sql=macro_insert_sql)
    setattr(macroData,'brand_macro_num',1)
    setattr(macroData,'team_macro_num',0)
    setattr(macroData,'user_macro_num',0)
    macro_id = append_macro_id
    macro_title  = macroData.append_macro_title
    macro_content = macroData.append_macro_content
    macro_sorts = 1
    macro_clicks = 0
    macro_brand_id = [macroData.brandId]
    macro_create_by_name = macroData.userName
    macro_create_by_id = macroData.userId
setattr(macroData,'macro_id',macro_id)
setattr(macroData,'append_macro_id',append_macro_id)
setattr(macroData,'macro_title',macro_title)
setattr(macroData,'macro_content',macro_content)
setattr(macroData,'macro_sorts',macro_sorts)
setattr(macroData,'macro_clicks',macro_clicks)
setattr(macroData,'macro_brand_id',macro_brand_id)
setattr(macroData,'macro_create_by_name',macro_create_by_name)
setattr(macroData,'macro_create_by_id',macro_create_by_id)
macroData.macro_db.close_db()