carddef2sql.py 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713
  1. import ast
  2. import json
  3. import re
  4. import sys
  5. import traceback
  6. import pandas as pd
  7. # 聚合函数
  8. AGGREGATION_MAP = {
  9. 'SUM': 'SUM',
  10. 'AVG': 'AVG',
  11. 'CNT': 'COUNT',
  12. 'MAX': 'MAX',
  13. 'MIN': 'MIN',
  14. 'CNT_DISTINCT': 'COUNT(DISTINCT {})'
  15. }
  16. AGGREGATION_SUFFIX_MAP = {
  17. 'SUM': '求和',
  18. 'AVG': '均值',
  19. 'CNT': '计数',
  20. 'MAX': '最大值',
  21. 'MIN': '最小值',
  22. 'CNT_DISTINCT': '去重计数'
  23. }
  24. # 筛选操作符
  25. FILTER_OPERATOR_MAP = {
  26. 'BT': {"val_nums": 2, 'template': "{field} BETWEEN {value_1} AND {value_2}"},
  27. 'CLOSE_BT_OPEN': {"val_nums": 2, 'template': "{field} >= {value_1} AND {field} < {value_2}"},
  28. 'EQ': {"val_nums": 1, 'template': "{field} = {value}"},
  29. 'GE': {"val_nums": 1, 'template': "{field} >= {value}"},
  30. 'GT': {"val_nums": 1, 'template': "{field} > {value}"},
  31. 'IN': {"val_nums": 9, 'template': "{field} IN ({values})"},
  32. 'IS_NULL': {"val_nums": 0, 'template': "{field} IS NULL"},
  33. 'LE': {"val_nums": 1, 'template': "{field} <= {value}"},
  34. 'LT': {"val_nums": 1, 'template': "{field} < {value}"},
  35. 'NE': {"val_nums": 1, 'template': "{field} != {value}"},
  36. 'NI': {"val_nums": 9, 'template': "{field} NOT IN ({values})"},
  37. 'NOT_NULL': {"val_nums": 0, 'template': "{field} IS NOT NULL"},
  38. 'OPEN_BT_CLOSE': {"val_nums": 2, 'template': "{field} > {value_1} AND {field} <= {value_2}"},
  39. 'OPEN_BT_OPEN': {"val_nums": 2, 'template': "{field} > {value_1} AND {field} < {value_2}"},
  40. 'CONTAINS': {"val_nums": 1, 'template': "{field} LIKE '%{value}%'"},
  41. 'NOT_CONTAINS': {"val_nums": 1, 'template': "{field} NOT LIKE '%{value}%'"},
  42. 'STARTS_WITH': {"val_nums": 1, 'template': "{field} LIKE '{value}%'"},
  43. 'NOT_STARTS_WITH': {"val_nums": 1, 'template': "{field} NOT LIKE '{value}%'"},
  44. 'ENDS_WITH': {"val_nums": 1, 'template': "{field} LIKE '%{value}'"},
  45. 'NOT_ENDS_WITH': {"val_nums": 1, 'template': "{field} NOT LIKE '%{value}'"},
  46. 'CUSTOM': 'CUSTOM',
  47. 'SPARK_EXPR': 'SPARK_EXPR'
  48. }
  49. # 引用
  50. IDENTIFIER_QUOTE = '`'
  51. QUOTE_FLAG = True
  52. # 副词
  53. ADV_FILTER_EXP_MAP = {
  54. 'TODAY': "{field} = '{{today}}'",
  55. 'YESTERDAY': "{field} = date_sub('{{today}}', 1)",
  56. 'DAY_BEFORE_YESTERDAY': "{field} = date_sub('{{today}}', 2)",
  57. 'LAST_14_DAY': "{field} between date_sub('{{today}}', 13) and '{{today}}'",
  58. 'LAST_1_YEAR': "{field} between date_sub('{{today}}', 364) and '{{today}}'",
  59. 'LAST_30_DAY': "{field} between date_sub('{{today}}', 29) and '{{today}}'",
  60. 'LAST_7_DAY': "{field} between date_sub('{{today}}', 6) and '{{today}}'",
  61. 'LAST_90_DAY': "{field} between date_sub('{{today}}', 89) and '{{today}}'",
  62. 'LAST_MONTH': "{field} between datesub(add_months('{{today}}', -1), day('{{today}}') - 1) and date_sub('{{today}}', day('{{today}}'))",
  63. 'LAST_WEEK': "{field} between date_sub('{{today}}', case when dayofweek('{{today}}') = 1 then 13 else dayofweek('{{today}}')+5 end) and date_sub('{{today}}', case when dayofweek('{{today}}') = 1 then 7 else dayofweek('{{today}}')-1 end)",
  64. 'MONTH_BEFORE_LAST_MONTH': "{field} between date_sub(add_months('{{today}}', -2), day(add_months('{{today}}', -2))-1) and date_sub(add_months('{{today}}', -1), day(add_months('{{today}}', -1)))",
  65. 'MONTH_TO_DAY': "{field} between date_sub('{{today}}', day('{{today}}')-1) and '{{today}}'",
  66. 'MONTH_TO_YESTERDAY': "{field} between date_sub('{{today}}', day('{{today}}')-1) and date_sub('{{today}}', 1)",
  67. 'QUARTER_TO_DAY': """{field} between concat(year('{{today}}'), case when quarter('{{today}}') = 1 then '-01-01'
  68. when quarter('{{today}}') = 2 then '-04-01'
  69. when quarter('{{today}}') = 3 then '-07-01'
  70. when quarter('{{today}}') = 4 then '-10-01' end) and '{{today}}'""",
  71. 'QUARTER_TO_YESTERDAY': """{field} between concat(year('{{today}}'), case when quarter('{{today}}') = 1 then '-01-01'
  72. when quarter('{{today}}') = 2 then '-04-01'
  73. when quarter('{{today}}') = 3 then '-07-01'
  74. when quarter('{{today}}') = 4 then '-10-01' end) and date_sub('{{today}}', 1)""",
  75. 'WEEK_TO_DAY': "{field} between date_sub('{{today}}', (dayofweek('{{today}}')+5)%7) and '{{today}}'",
  76. 'WEEK_TO_YESTERDAY': "{field} between date_sub('{{today}}', (dayofweek('{{today}}')+5)%7) and date_sub('{{today}}', 1)",
  77. 'YEAR_TO_DAY': "{field} between concat(year('{{today}}'), '-01-01') and '{{today}}'",
  78. 'YEAR_TO_LAST_MONTH': "{field} between concat(year('{{today}}'), '-01-01') and date_sub('{{today}}', day('{{today}}'))",
  79. 'YEAR_TO_LAST_QUARTER': """{field} between concat(year('{{today}}'), '-01-01') and date_sub(concat(year('{{today}}'), case when quarter('{{today}}') = 1 then '-01-01'
  80. when quarter('{{today}}') = 2 then '-04-01'
  81. when quarter('{{today}}') = 3 then '-07-01'
  82. when quarter('{{today}}') = 4 then '-10-01' end), 1)""",
  83. 'YEAR_TO_YESTERDAY': "{field} between concat(year('{{today}}'), '-01-01') and date_sub('{{today}}', 1)",
  84. }
  85. # 自带日期转换
  86. PARTIAL_DATE_EXPRESSION = {
  87. 'day': "`{old}` as `{new}`",
  88. 'month': "concat(year(`{old}`), '-', lpad(month(`{old}`), 2, '0') as `{new}`",
  89. 'year': "year(`{old}`) as `{new}`",
  90. 'quarter': "concat(year(`{old}`), '-S', quarter(`{old}`)) as `{new}`",
  91. 'week': "weekofyear(`{old}`) as `{new}`",
  92. 'dayofweek': "dayofweek(`{old}`)-1 as `{new}`",
  93. 'hour': "hour(`{old}`)-8 as `{new}`",
  94. 'minute': "minute(`{old}`)-8 as `{new}`",
  95. }
  96. # 获取新增字段
  97. def get_added_fields_info(added_fields_df):
  98. if added_fields_df.empty:
  99. return {}
  100. added_fields_info = {}
  101. for _, row in added_fields_df.iterrows():
  102. try:
  103. content = json.loads(row['calc_field_logic'])
  104. except:
  105. print(f'ERROR: 新增字段解析错误: {row["calc_field_logic"]}')
  106. continue
  107. field_name = content['name']
  108. field_id = content['fdId']
  109. added_fields_info[field_id] = {"field_name": field_name, 'calculation': content}
  110. added_fields_info[field_name] = {"field_id": field_id, 'calculation': content}
  111. return added_fields_info
  112. def get_fid_name_map(field_def_df):
  113. field_id_name = {}
  114. for i, row in field_def_df.iterrows():
  115. field_id_name[row['field_id']] = row['field_name']
  116. return field_id_name
  117. # 字段映射关系
  118. def get_fields_rename_map(field_info):
  119. ret = {}
  120. try:
  121. tmp_map = json.loads(field_info)
  122. except:
  123. return ret
  124. dimensions, metrics = tmp_map.get("dimensions"), tmp_map.get("metrics")
  125. if dimensions and dimensions != 'null':
  126. for one_map in dimensions:
  127. ret[one_map["name"]] = one_map["alias"]
  128. if metrics and metrics != 'null':
  129. for one_map in metrics:
  130. ret[one_map["name"]] = one_map["alias"]
  131. return ret
  132. # 递归解析嵌套的计算字段
  133. def resolve_calculation_formula(formula, calculation_fields, visited=None):
  134. if not formula:
  135. return formula
  136. if visited is None:
  137. visited = set()
  138. def replace_calculation_field(match):
  139. field_key = match.group(1).strip()
  140. field_def = calculation_fields.get(field_key)
  141. if not field_def:
  142. return match.group(0)
  143. field_id = field_def.get("field_id") or field_key
  144. if field_id in visited:
  145. raise ValueError(f"计算字段存在循环引用: {field_key}")
  146. nested_formula = field_def["calculation"].get("formula", "")
  147. if "consolidation" in nested_formula:
  148. return match.group(0)
  149. resolved = resolve_calculation_formula(nested_formula, calculation_fields, visited | {field_id})
  150. return f"({resolved})"
  151. return re.sub(r"\[([^\[\]]+)\]", replace_calculation_field, formula)
  152. def build_with_part(new_date_fields, new_dimension_fields, dataset_fid_name_map, added_fields_info, dataset_id):
  153. override_field_names = set()
  154. for _, new_name in new_date_fields:
  155. override_field_names.add(new_name)
  156. for fid, _ in new_dimension_fields:
  157. override_field_names.add(added_fields_info[fid]["field_name"])
  158. base_columns = []
  159. seen_columns = set()
  160. for field_name in dataset_fid_name_map.values():
  161. if field_name in override_field_names or field_name in seen_columns:
  162. continue
  163. seen_columns.add(field_name)
  164. base_columns.append(quote_identifier(field_name))
  165. with_expressions = []
  166. for fid, new_name in new_date_fields:
  167. old_fid, partial_date = fid.split('_')
  168. if old_fid in dataset_fid_name_map:
  169. old_name = dataset_fid_name_map[old_fid]
  170. elif old_fid in added_fields_info:
  171. old_name = added_fields_info[old_fid]["calculation"]["formula"].replace("[", "").replace("]", '')
  172. else:
  173. raise ValueError(f"字段 {fid} {new_name} 不存在")
  174. tmp_part = PARTIAL_DATE_EXPRESSION.get(partial_date, None)
  175. if tmp_part:
  176. tmp_part = tmp_part.format(old=old_name, new=new_name)
  177. with_expressions.append(tmp_part)
  178. else:
  179. raise ValueError(f"日期转换方式 {partial_date} 不存在")
  180. for fid, new_name in new_dimension_fields:
  181. field_def = added_fields_info[fid]
  182. new_name = field_def["field_name"]
  183. formula = field_def["calculation"]["formula"]
  184. if "consolidation" in formula:
  185. consolidation_dict = json.loads(formula)["consolidation"]
  186. tmp_part = get_consolidation_field(consolidation_dict)
  187. tmp_part += f" AS `{new_name}`"
  188. else:
  189. # 递归解析计算字段是否有嵌套情况
  190. formula = resolve_calculation_formula(formula, added_fields_info, {fid})
  191. tmp_part = quote_identifier(formula, formula=True) + f" AS `{new_name}`"
  192. with_expressions.append(tmp_part)
  193. select_parts = base_columns + with_expressions
  194. sql_part = "WITH tmp as (\nSELECT " + ",\n".join(select_parts)
  195. sql_part += f"\nFROM {quote_identifier(str(dataset_id))}\n)"
  196. return sql_part
  197. # 处理计算字段
  198. def process_measure_fields(measure_fields, measure_aggs, calculation_fields, card_id, card_name):
  199. ## 数值字段数量 小于 聚合函数数量,不合法
  200. if len(measure_fields) < len(measure_aggs):
  201. print(f"警告: 卡片 {card_id} {card_name}: 数值字段数量小于聚合函数数量,不合法")
  202. print(f"警告: 卡片 {card_id} {card_name}: 不添加任何数值字段.")
  203. return [], [], [], False
  204. ## 数值字段 大于 聚合函数数量,存在聚合类型的计算字段,尝试填充
  205. elif len(measure_fields) > len(measure_aggs):
  206. ## 计算数值字段数量
  207. num_cals = 0
  208. for field in measure_fields:
  209. if field in calculation_fields: # and calculation_fields[field]["calculation"]["isAggregated"] is True:
  210. num_cals += 1
  211. ## 如果不存在任何计算字段,补全剩余的NUL聚合函数
  212. if num_cals == 0:
  213. measure_aggs.extend(['NULL'] * (len(measure_fields) - len(measure_aggs)))
  214. return [quote_identifier(field) for field in measure_fields], measure_aggs, [False] * len(measure_fields), True
  215. ## 如果存在计算字段,且相加后的 聚合函数数量 仍小于 数值字段数量,不合法
  216. if num_cals + len(measure_aggs) != len(measure_fields):
  217. print(f"警告: 卡片 {card_id} {card_name}: 数值字段数量大于聚合函数数量,不合法")
  218. print(f"警告: 卡片 {card_id} {card_name}: 不添加任何数值字段.")
  219. return [], [], [], False
  220. ## 通过验证,填充聚合函数
  221. new_measure_fields, new_measure_aggs, measure_is_aggregated, agg_flag = [], [], [], False
  222. for i, field in enumerate(measure_fields):
  223. ## 非计算字段
  224. if field not in calculation_fields:
  225. new_measure_fields.append(quote_identifier(field))
  226. new_measure_aggs.append(measure_aggs.pop(0))
  227. measure_is_aggregated.append(False)
  228. ## 计算字段
  229. else:
  230. formula = calculation_fields[field]["calculation"]["formula"]
  231. formula = resolve_calculation_formula(formula, calculation_fields, {calculation_fields[field]["field_id"]})
  232. new_measure_fields.append(quote_identifier(formula, formula=True))
  233. if calculation_fields[field]["calculation"]["isAggregated"] is True:
  234. new_measure_aggs.append("NUL")
  235. measure_is_aggregated.append(True)
  236. agg_flag = True
  237. else:
  238. new_measure_aggs.append(measure_aggs.pop(0))
  239. measure_is_aggregated.append(False)
  240. return new_measure_fields, new_measure_aggs, measure_is_aggregated, agg_flag
  241. # sql部分去重
  242. def dedupe_sql_parts(parts):
  243. deduped = []
  244. seen = set()
  245. for part in parts:
  246. if not part or part in seen:
  247. continue
  248. seen.add(part)
  249. deduped.append(part)
  250. return deduped
  251. def quote_identifier(identifier, formula=False):
  252. if not QUOTE_FLAG:
  253. return identifier
  254. if not identifier:
  255. return ''
  256. # 简单处理,如果包含非字母数字下划线或可能是关键字,则加反引号
  257. # 更复杂的关键字检查可以添加
  258. if formula:
  259. params = re.findall(r"\[DYNAMIC_PARAMS\.\w+\]", identifier)
  260. for p in params:
  261. subs = p[1:-1]
  262. subs = "{{{"+subs+"}}}"
  263. identifier = identifier.replace(p, subs, 1)
  264. # 仅替换配置里用于包裹字段名的 [字段],保留 Hive 下标访问里的 [2] 等表达式
  265. def replace_bracket_identifier(match):
  266. content = match.group(1)
  267. if re.fullmatch(r"\d+", content.strip()):
  268. return match.group(0)
  269. return f"{IDENTIFIER_QUOTE}{content}{IDENTIFIER_QUOTE}"
  270. identifier = re.sub(r"\[([^\[\]]+)\]", replace_bracket_identifier, identifier)
  271. else:
  272. identifier = identifier.replace('\n', ' ')
  273. if not re.match(r'[a-zA-Z_][a-zA-Z0-9_]*$', identifier):
  274. return f'{IDENTIFIER_QUOTE}{identifier}{IDENTIFIER_QUOTE}'
  275. return identifier
  276. def parse_multi_value_field(field_value):
  277. # 解析包含多个值的字段
  278. if not field_value or field_value == "":
  279. return []
  280. try:
  281. res = ast.literal_eval(field_value)
  282. except Exception:
  283. print(field_value)
  284. print(traceback.format_exc())
  285. return ast.literal_eval(field_value)
  286. # 处理过滤条件的操作符
  287. def get_format_args(field, fd_type, op_dict, values):
  288. # 按照数据类型及操作符,判断是否需要加引号
  289. if fd_type in ('DECIMAL', 'DOUBLE', 'INT', 'FLOAT', 'LONG', 'SHORT'):
  290. values = [x for x in values if x]
  291. elif fd_type in ('DATE', 'STRING', 'SUB_DATE', 'TIMESTAMP'):
  292. if op_dict.get('quote', True):
  293. values = [f"'{x}'" for x in values]
  294. elif fd_type == 'BOOL':
  295. values = [value.upper() for value in values]
  296. else:
  297. pass
  298. # 按照操作符所需参数个数构造format参数
  299. format_dict = {}
  300. value_nums = op_dict['val_nums']
  301. if value_nums == 9:
  302. format_dict.update(**{"values": ", ".join(values)})
  303. elif value_nums == 2:
  304. format_dict.update(**{"value_1": values[0], "value_2": values[1]})
  305. elif value_nums == 1:
  306. format_dict.update(**{"value": values[0]})
  307. else:
  308. pass
  309. format_dict["field"] = field
  310. return format_dict
  311. # 处理过滤条件中的consolidation
  312. def get_consolidation_field(consolidation_dict):
  313. field_name = quote_identifier(consolidation_dict["sourceName"])
  314. group_type = consolidation_dict["groupType"]
  315. fd_type = consolidation_dict["sourceFdType"]
  316. group_rules = consolidation_dict.get('groups')
  317. fixed_step = consolidation_dict.get('fixedStepSetting')
  318. when_part = []
  319. else_part = None
  320. if group_type == 'ITEM':
  321. for group in group_rules:
  322. group_name = group["groupName"]
  323. if group.get('isOtherGroup', False):
  324. else_part = f"ELSE '{group_name}'"
  325. else:
  326. selected_values = group.get('selectedValues', [])
  327. when_value = str(selected_values)
  328. when_value = when_value[1:-1] # 去除中括号
  329. if when_value is None:
  330. when_part.append(f"WHEN {field_name} IS NULL THEN '{group_name}'")
  331. else:
  332. when_part.append(f"WHEN {field_name} IN ({when_value}) THEN '{group_name}'")
  333. elif group_type == "CONDITION":
  334. for group in group_rules:
  335. group_name = group["groupName"]
  336. if group.get('isOtherGroup', False):
  337. else_part = f"ELSE '{group_name}'"
  338. else:
  339. rules = group["rules"]
  340. cond_list = []
  341. combine_type = " " + rules["combineType"] + " " # 加空格以便join
  342. for cond in rules["conditions"]:
  343. filter_type = cond["filterType"]
  344. filter_value = cond["filterValue"]
  345. op_dict = FILTER_OPERATOR_MAP[filter_type]
  346. format_args = get_format_args(field_name, fd_type, op_dict, [filter_value])
  347. cond_str = op_dict["template"].format(**format_args)
  348. cond_list.append(cond_str)
  349. when_str = "WHEN " + combine_type.join(cond_list) + f" THEN '{group_name}'"
  350. when_part.append(when_str)
  351. elif group_type == "CUSTOM_STEP":
  352. for group in group_rules:
  353. group_name = group["groupName"]
  354. if group.get('isOtherGroup', False):
  355. else_part = f"ELSE '{group_name}'"
  356. else:
  357. setting = group['customStepSetting']
  358. operator = setting['operator']
  359. start = setting['startValue']
  360. end = setting['endValue']
  361. condition = ''
  362. if operator == 'BT':
  363. condition = f"{field_name} BETWEEN {start} AND {end}"
  364. elif operator == 'OPEN_BT_CLOSE':
  365. condition = f"{field_name} > {start} AND {field_name} <= {end}"
  366. elif operator == 'OPEN_BT_OPEN':
  367. condition = f"{field_name} > {start} AND {field_name} < {end}"
  368. elif operator == 'CLOSE_BT_OPEN':
  369. condition = f"{field_name} >= {start} AND {field_name} < {end}"
  370. else:
  371. raise ValueError(f"未知的操作符: {operator}")
  372. when_part.append(f"WHEN {condition} THEN '{group_name}'")
  373. elif group_type == "FIXED_STEP":
  374. start = fixed_step['startValue']
  375. end = fixed_step['endValue']
  376. step = fixed_step['stepSize']
  377. # 生成每个区间的case when部分
  378. lower = start
  379. while lower < end:
  380. upper = lower + step
  381. case_part = f"WHEN {field_name} >= {lower} AND {field_name} < {upper} THEN '{lower}-{upper}'"
  382. when_part.append(case_part)
  383. lower = upper
  384. # 处理最后一个区间
  385. case_part = f"WHEN {field_name} >= {lower} AND {field_name} <= {end} THEN '{lower}-{end}'"
  386. when_part.append(case_part)
  387. else_part = "ELSE NULL"
  388. else:
  389. raise ValueError(f"未知的groupType: {group_type}")
  390. field = "CASE "
  391. field += "\n".join(when_part)
  392. if else_part:
  393. field += f"\n{else_part}"
  394. field += "\nEND"
  395. return field
  396. def parse_filter_string(filter_relation_str):
  397. conditions = {}
  398. if not filter_relation_str or filter_relation_str == "[]":
  399. return conditions
  400. raw_conditions = json.loads(filter_relation_str)
  401. for cond_dict in raw_conditions:
  402. fdId = cond_dict.get("fdId")
  403. field = cond_dict.get("name")
  404. fd_type = cond_dict.get("fdType")
  405. op_name = cond_dict.get("filterType")
  406. op_dict = FILTER_OPERATOR_MAP.get(op_name)
  407. values = cond_dict.get("filterValue") # list
  408. is_aggregated = cond_dict.get("isAggregated", False)
  409. # 检查条件合法
  410. if any([fdId is None, field is None, fd_type is None, op_name is None, values is None]):
  411. print(f"fdId: {fdId} field: {field} fd_type: {fd_type} op_name: {op_name} values: {values}")
  412. print(f"警告: 无法解析筛选条件,缺少必须字段,跳过此条件。")
  413. continue
  414. if op_dict is None:
  415. print(f"警告: 无法解析筛选条件,未定义的筛选类型: {op_name},跳过此条件。")
  416. continue
  417. # 特殊操作符
  418. if op_dict == 'CUSTOM':
  419. if "advFilter" not in cond_dict:
  420. print(f"警告: CUSTOM筛选类型不存在advFilter, 跳过此条件。")
  421. continue
  422. if 'formula' in cond_dict:
  423. field = quote_identifier(cond_dict['formula'], formula=True)
  424. else:
  425. field = quote_identifier(cond_dict['name'])
  426. expression = ADV_FILTER_EXP_MAP.get(cond_dict["advFilter"])
  427. if not expression:
  428. print(f"警告: CUSTOM筛选类型出现未定义的advFilter: {cond_dict['advFilter']}, 跳过此条件。")
  429. continue
  430. expression = expression.format(field=field)
  431. conditions[fdId] = {"exp": expression, "agg": is_aggregated}
  432. continue
  433. elif op_dict == 'SPARK_EXPR':
  434. if 'formula' in cond_dict:
  435. formula = quote_identifier(cond_dict['formula'], formula=True)
  436. conditions[fdId] = {"exp": formula, "agg": is_aggregated}
  437. else:
  438. if isinstance(cond_dict['filterValue'], list) and len(cond_dict['filterValue']) == 1:
  439. field = quote_identifier(cond_dict['name'])
  440. value = cond_dict['filterValue'][0]
  441. conditions[fdId] = {"exp": f"{field} = {value}", "agg": is_aggregated}
  442. else:
  443. print(f"警告: 无法解析筛选条件,SPARK_EXPR中未定义。跳过此条件。")
  444. continue
  445. # 处理条件
  446. value_nums = op_dict["val_nums"]
  447. if value_nums != 0 and len(values) != value_nums:
  448. print(f"警告: 无法解析筛选条件,值数量与操作符不匹配。跳过此条件。")
  449. continue
  450. field = quote_identifier(field)
  451. # consolidation 情况,将consolidation公式替换条件左边的field
  452. if "consolidation" in cond_dict:
  453. consolidation = cond_dict["consolidation"]
  454. consolidation_field = get_consolidation_field(consolidation)
  455. if not consolidation_field:
  456. print(f"警告: 无法解析consolidation字段。跳过此条件。")
  457. continue
  458. else:
  459. field = consolidation_field
  460. else:
  461. # 公式,非 consolidation情况
  462. if "formula" in cond_dict:
  463. field = quote_identifier(cond_dict["formula"], formula=True)
  464. if op_name in ("NI", "IN") and len(values) == 0:
  465. print(f"警告: 无法解析筛选条件,IN或NI中参数个数为0。跳过此条件。")
  466. continue
  467. # 特殊情况
  468. if op_name in ('NI', 'IN') and None in values:
  469. conditions[fdId] = {"exp": f"{field} IS NOT NULL", "agg": is_aggregated}
  470. values = [x for x in values if x is not None]
  471. if len(values) == 0:
  472. continue
  473. # 填充模板所需要的参数
  474. format_args = get_format_args(field, fd_type, op_dict, values)
  475. condition_str = op_dict["template"].format(**format_args)
  476. conditions[fdId] = {"exp": condition_str, "agg": is_aggregated}
  477. return conditions
  478. def build_sql_query(card_data, added_fields_info, dataset_fid_name_map):
  479. card_id = card_data["card_id"]
  480. card_name = card_data["card_name"]
  481. dataset_id = card_data.get("ds_id")
  482. if not dataset_id:
  483. print(f"错误: {card_id} {card_name} 数据集ID为空.")
  484. return "", "", "", ""
  485. added_fields_info = get_added_fields_info(added_fields_info)
  486. dataset_fid_name_map = get_fid_name_map(dataset_fid_name_map)
  487. dimension_fids = parse_multi_value_field(card_data.get("field_id", []))
  488. dimension_fields = parse_multi_value_field(card_data.get("field_name", []))
  489. dimension_fid_name_map = dict(zip(dimension_fids, dimension_fields))
  490. dimension_name_fid_map = dict(zip(dimension_fields, dimension_fids))
  491. measure_fids = parse_multi_value_field(card_data.get("num_value_field_id", []))
  492. measure_fields = parse_multi_value_field(card_data.get("num_value_field_name", []))
  493. measure_aggs = parse_multi_value_field(card_data.get("num_value_field_merge_way", []))
  494. filter_relation_str = card_data.get("filters_field_value_name_rela")
  495. sort_fids = parse_multi_value_field(card_data.get("sort_field_id", []))
  496. sort_fields = parse_multi_value_field(card_data.get("sort_field_name", []))
  497. sort_method = parse_multi_value_field(card_data.get("sort_way", []))
  498. all_field_ids = dimension_fids + \
  499. parse_multi_value_field(card_data.get("filters_field_id", [])) + \
  500. sort_fids + \
  501. measure_fids
  502. all_field_names = dimension_fields + \
  503. parse_multi_value_field(card_data.get("filters_field_name", [])) + \
  504. sort_fields + \
  505. measure_fields
  506. all_field_id_name_map = dict(zip(all_field_ids, all_field_names))
  507. # 处理字段重命名关系
  508. fields_rename_map = get_fields_rename_map(card_data.get("field_info", ""))
  509. # 处理field_id与重命名关系,用于筛选Order by子句中的字段
  510. # 需要处理的只有日期转换类型,将转换前的原始字段名加入map
  511. # 只需要更新有重命名的字段即可
  512. selected_fid_alias_map = dict(zip(dimension_fids+measure_fids, dimension_fields+measure_fields))
  513. # 构建WITH
  514. with_part = ""
  515. new_date_fields = []
  516. # 日期转换
  517. for fid, name in all_field_id_name_map.items():
  518. fid_splits = fid.split('_')
  519. if len(fid_splits) == 2:
  520. new_date_fields.append((fid, name))
  521. old_fid = fid_splits[0]
  522. selected_fid_alias_map[old_fid] = name
  523. # 新增维度字段
  524. new_dimension_fields = []
  525. for fid, name in dimension_fid_name_map.items():
  526. if fid in added_fields_info:
  527. new_dimension_fields.append((fid, name))
  528. # 如果有新增日期字段、新增维度字段,构建WITH
  529. if new_date_fields or new_dimension_fields:
  530. with_part = build_with_part(new_date_fields, new_dimension_fields, dataset_fid_name_map, added_fields_info, dataset_id)
  531. # 构建SELECT
  532. select_parts = []
  533. has_aggregation = False
  534. non_aggregated_select_parts = []
  535. # 添加维度字段
  536. for field in dimension_fields:
  537. fid = dimension_name_fid_map[field]
  538. alias = fields_rename_map.get(field)
  539. if alias and alias != "null":
  540. select_parts.append(f"{quote_identifier(field)} AS {quote_identifier(alias)}")
  541. selected_fid_alias_map[fid] = alias
  542. else:
  543. select_parts.append(f"{quote_identifier(field)}")
  544. selected_fid_alias_map[fid] = field
  545. # 加工计算字段
  546. new_measure_fields, measure_aggs, measure_is_aggregated, agg_flag = process_measure_fields(measure_fields, measure_aggs, added_fields_info, card_id, card_name)
  547. if agg_flag:
  548. has_aggregation = True
  549. for i, field in enumerate(new_measure_fields):
  550. fid = measure_fids[i]
  551. alias = fields_rename_map.get(field.strip('`'))
  552. # measure_agg是NUL,不需要聚合(等同于维度字段)或公式本身已经有聚合函数
  553. agg_func_template = AGGREGATION_MAP.get(measure_aggs[i])
  554. if not agg_func_template:
  555. if not alias or alias == "null":
  556. alias = measure_fields[i]
  557. select_parts.append(f"{field} AS {quote_identifier(alias)}")
  558. # 属于计算字段,但没有聚合函数,等同于维度字段,需要加入groupbyby。
  559. if not measure_is_aggregated[i] and field and re.search(r"\b(sum|avg|count|max|min|stddev|variance|collect_list|collect_set|percentile|percentile_approx)|\s*\(", field, flags=re.IGNORECASE) is None:
  560. non_aggregated_select_parts.append(field)
  561. selected_fid_alias_map[fid] = alias
  562. else:
  563. has_aggregation = True
  564. # 特殊处理 count distinct
  565. if '{}' in agg_func_template:
  566. agg_expression = agg_func_template.format(field)
  567. else:
  568. agg_expression = f"{agg_func_template}({field})"
  569. # 添加别名
  570. if not alias or alias == "null":
  571. suffix = AGGREGATION_SUFFIX_MAP.get(measure_aggs[i])
  572. alias = f"{measure_fields[i]}_{suffix}"
  573. select_parts.append(f"{agg_expression} AS {quote_identifier(alias)}")
  574. selected_fid_alias_map[fid] = alias
  575. # BI 卡片配置里可能存在重复字段。这里只对完全相同的 SELECT 表达式去重,保留表达式不同但别名相同的情况。
  576. select_parts = dedupe_sql_parts(select_parts)
  577. if not select_parts:
  578. print(f"错误: {card_id} {card_name} 没有select字段。")
  579. return '', '', '', ''
  580. else:
  581. select_clause = "SELECT " + ",\n ".join(select_parts)
  582. # 构建FROM
  583. if with_part:
  584. from_clause = "FROM tmp"
  585. else:
  586. from_clause = f"FROM {quote_identifier(str(dataset_id))}"
  587. # 构建WHERE
  588. filter_conditions = {}
  589. try:
  590. filter_conditions = parse_filter_string(filter_relation_str)
  591. except Exception as e:
  592. print(f"错误: 卡片 {card_id} {card_name} 解析筛选条件出错:{e}。WHERE字句缺失。")
  593. print("详细错误信息:")
  594. print(traceback.format_exc())
  595. # 构建GROUPBY
  596. group_by_clause = ""
  597. if has_aggregation:
  598. group_by_parts = [quote_identifier(field) for field in dimension_fields]
  599. group_by_parts.extend(non_aggregated_select_parts)
  600. group_by_parts = dedupe_sql_parts(group_by_parts)
  601. if group_by_parts:
  602. group_by_clause = "GROUP BY " + ", ".join(group_by_parts)
  603. # 构建ORDERBY
  604. order_by_clause = ""
  605. if sort_fields and sort_method and len(sort_fields) == len(sort_method):
  606. order_by_parts = []
  607. for i, field in enumerate(sort_fields):
  608. fid = sort_fids[i]
  609. if fid not in selected_fid_alias_map:
  610. continue
  611. alias = selected_fid_alias_map[fid]
  612. order_by_parts.append(f"{quote_identifier(alias)} {sort_method[i]}")
  613. if order_by_parts:
  614. order_by_clause = "ORDER BY " + ", ".join(order_by_parts)
  615. # 组装SQL
  616. sql_parts = [with_part, select_clause, from_clause]
  617. # 返回 select, where, groupby, orderby
  618. return ("\n".join(sql_parts)).strip(), json.dumps(filter_conditions, ensure_ascii=False), group_by_clause, order_by_clause
  619. def generate(start=None, end=None, test_card_id=None):
  620. res_list = []
  621. df = pd.read_csv("data/card.csv").fillna("").reset_index()
  622. add_field_info = pd.read_csv("data/calc.csv").fillna('').set_index("card_id")
  623. all_field_info = pd.read_csv("data/field.csv").fillna('').set_index("ds_id")
  624. for i, row in df.iterrows():
  625. if start and i < start:
  626. continue
  627. if end and i > end:
  628. break
  629. card_id = row["card_id"]
  630. if test_card_id and card_id != test_card_id:
  631. continue
  632. if row["card_type_cd"] != '图表' or row["ds_id"] == "":
  633. continue
  634. try:
  635. added_fields_info = add_field_info.loc[[card_id]]
  636. except KeyError:
  637. added_fields_info = pd.DataFrame()
  638. try:
  639. dataset_fid_name_map = all_field_info.loc[[row["ds_id"]]]
  640. except KeyError:
  641. print(f"错误: 没有数据及字段信息: {card_id}")
  642. continue
  643. select, where, groupby, orderby = '', '', '', ''
  644. try:
  645. select, where, groupby, orderby = build_sql_query(row, added_fields_info, dataset_fid_name_map)
  646. except Exception as e:
  647. print(f"错误: 卡片 {card_id} 发生未知错误: {e}")
  648. print(i, traceback.format_exc())
  649. if not select:
  650. print(f"{card_id} 生成失败")
  651. continue
  652. res_list.append([str(card_id), str(row["card_name"]), select, where, groupby, orderby])
  653. res_df = pd.DataFrame(res_list, columns=["card_id", "card_name", "select", 'where', 'groupby', 'orderby'])
  654. return res_df
  655. if __name__ == "__main__":
  656. df = generate()
  657. df.to_parquet("output/sql.parquet")
  658. df.to_excel("output/sql.xlsx")