# -*- coding: utf-8 -*-
"""Patch gen_rework_bru.py rules: table_add->table_replace for 'wu' drafts; TASKID pattern."""
p = r'e:\code\crm-backend-matt\.scratch\customer-rework\gen_rework_bru.py'
s = open(p, encoding='utf-8').read()

# 1. customer detail (kehu xiangqing)
s = s.replace("""         key_sub=[('`GET /api/customer/{id}`', '`GET /api/customer/detail`')],
         query_add=[('id', ID_EX)],
         table_add=[('id', 'string', '是', '-', '客户 id（查询参数）')]),""",
"""         key_sub=[('`GET /api/customer/{id}`', '`GET /api/customer/detail`')],
         query_add=[('id', ID_EX)],
         table_replace=[('id', 'string', '是', '-', '客户 id（查询参数）')]),""")

# 2. contacts (an kehu cha lianxiren)
s = s.replace("""         key_sub=[('`GET /api/customer/{customerId}/contacts`', '`GET /api/customer/contacts`')],
         query_add=[('customerId', ID_EX)],
         table_add=[('customerId', 'string', '是', '-', '客户 id（查询参数；商机字段17级联数据源）')]),""",
"""         key_sub=[('`GET /api/customer/{customerId}/contacts`', '`GET /api/customer/contacts`')],
         query_add=[('customerId', ID_EX)],
         table_replace=[('customerId', 'string', '是', '-', '客户 id（查询参数；商机字段17级联数据源）')]),""")

# 3. detail-head
s = s.replace("""         key_sub=[('`GET /api/customer/{id}/detail-head`', '`GET /api/customer/detail-head`')],
         query_add=[('id', ID_EX)],
         table_add=[('id', 'string', '是', '-', '客户 id（查询参数）')]),""",
"""         key_sub=[('`GET /api/customer/{id}/detail-head`', '`GET /api/customer/detail-head`')],
         query_add=[('id', ID_EX)],
         table_replace=[('id', 'string', '是', '-', '客户 id（查询参数）')]),""")

# 4. member list (tuanduichengyuan yeqian)
s = s.replace("""         key_sub=[('`GET /api/customer/{id}/members`', '`GET /api/customer/member/list`')],
         query_add=[('id', ID_EX)],
         table_add=[('id', 'string', '是', '-', '客户 id（查询参数）')]),""",
"""         key_sub=[('`GET /api/customer/{id}/members`', '`GET /api/customer/member/list`')],
         query_add=[('id', ID_EX)],
         table_replace=[('id', 'string', '是', '-', '客户 id（查询参数）')]),""")

# 5/6/7. import three: TASKID pattern + table_replace
s = s.replace("""         path_re=[(r'/api/customer/import/(\\d+)/confirm', r'/api/customer/import/confirm?taskId=\\1')],
         key_sub=[('`POST /api/customer/import/{taskId}/confirm`', '`POST /api/customer/import/confirm`')],
         query_add=[('taskId', ID_EX)],
         table_add=[('taskId', 'string', '是', '-', '导入任务 id（查询参数；来自上传预校验返回）')]),""",
"""         path_re=[(r'/api/customer/import/(\\d+|TASKID)/confirm', r'/api/customer/import/confirm?taskId=\\1')],
         key_sub=[('`POST /api/customer/import/{taskId}/confirm`', '`POST /api/customer/import/confirm`')],
         query_add=[('taskId', 'TASKID')],
         table_replace=[('taskId', 'string', '是', '-', '导入任务 id（查询参数；来自上传预校验返回）')]),""")
s = s.replace("""         path_re=[(r'/api/customer/import/(\\d+)(?![/\\w])', r'/api/customer/import/result?taskId=\\1')],
         key_sub=[('`GET /api/customer/import/{taskId}`', '`GET /api/customer/import/result`')],
         query_add=[('taskId', ID_EX)],
         table_add=[('taskId', 'string', '是', '-', '导入任务 id（查询参数）')]),""",
"""         path_re=[(r'/api/customer/import/(\\d+|TASKID)(?![/\\w])', r'/api/customer/import/result?taskId=\\1')],
         key_sub=[('`GET /api/customer/import/{taskId}`', '`GET /api/customer/import/result`')],
         query_add=[('taskId', 'TASKID')],
         table_replace=[('taskId', 'string', '是', '-', '导入任务 id（查询参数）')]),""")
s = s.replace("""         path_re=[(r'/api/customer/import/(\\d+)/failures', r'/api/customer/import/failures?taskId=\\1')],
         key_sub=[('`GET /api/customer/import/{taskId}/failures`', '`GET /api/customer/import/failures`')],
         query_add=[('taskId', ID_EX)],
         table_add=[('taskId', 'string', '是', '-', '导入任务 id（查询参数）')]),""",
"""         path_re=[(r'/api/customer/import/(\\d+|TASKID)/failures', r'/api/customer/import/failures?taskId=\\1')],
         key_sub=[('`GET /api/customer/import/{taskId}/failures`', '`GET /api/customer/import/failures`')],
         query_add=[('taskId', 'TASKID')],
         table_replace=[('taskId', 'string', '是', '-', '导入任务 id（查询参数）')]),""")

# 8/9. transfer detail + assignable
s = s.replace("""         key_sub=[('`GET /api/customer/transfer/{id}`', '`GET /api/customer/transfer/detail`')],
         query_add=[('id', ID_EX)],
         table_add=[('id', 'string', '是', '-', '交接单 id（查询参数）')]),""",
"""         key_sub=[('`GET /api/customer/transfer/{id}`', '`GET /api/customer/transfer/detail`')],
         query_add=[('id', ID_EX)],
         table_replace=[('id', 'string', '是', '-', '交接单 id（查询参数）')]),""")
s = s.replace("""         key_sub=[('`GET /api/customer/transfer/{id}/assignable`', '`GET /api/customer/transfer/assignable`')],
         query_add=[('id', ID_EX)],
         table_add=[('id', 'string', '是', '-', '交接单 id（查询参数）')]),""",
"""         key_sub=[('`GET /api/customer/transfer/{id}/assignable`', '`GET /api/customer/transfer/assignable`')],
         query_add=[('id', ID_EX)],
         table_replace=[('id', 'string', '是', '-', '交接单 id（查询参数）')]),""")

open(p, 'w', encoding='utf-8', newline='').write(s)
import re as _re
print('table_replace count:', s.count('table_replace'), '; table_add count:', s.count('table_add'))