from Utils.sql_handler import test_env_conn
from TestData.billing.buyPlan_test_data import buyPlanData


select_sql = f'select * from willdesk.business_plan_current_used where `comp_id` = {buyPlanData.companyId} and `status` = 2'
result = test_env_conn.select_one_value(sql=select_sql) #检查当前套餐是否为免费
if result['plan_id'] == buyPlanData.FreePlanId:
    update_plan_sql = f'update willdesk.business_plan_current_used set `plan_id` = {buyPlanData.ProPlanId} where `comp_id` = {buyPlanData.companyId} and `status` = 2'
    test_env_conn.execute_sql(sql=update_plan_sql) #如果是免费套餐，手动update成付费
    # now_active_plan = test_env_conn.select_one_value(sql=f'select * from willdesk.business_plan_current_used where `comp_id` = {buyPlanData.companyId} and `status` = 2')

    for num in range(1,38):
        number = 2000 if num in buyPlanData.numItem else -1
        update_item_sql = f'update willdesk.business_plan_current_used_item_detail set amount = {number},plan_id = {buyPlanData.ProPlanId} where `comp_id` = {buyPlanData.companyId} and `status` = 2 and `item_id` = {num};'
        test_env_conn.execute_sql(sql=update_item_sql)