|
@@ -15,11 +15,14 @@ CREATE OR REPLACE FUNCTION "public"."f_rel_calc"("func" text, "project_id" varch
|
|
|
RETURNS "pg_catalog"."int4" AS $BODY$
|
|
|
# 根据函数输入类型分类
|
|
|
classify = {"eq2bd":1, "eq2fl":1, "sp2fl":1, "eq2sh":1, "pe2bd":1, "pe2fl":1, "pe2sh":1, "pe2sp":1, "sh2sh":1, "sy2bd":1,
|
|
|
- "sy2fl":1, "sy2sh":1, "sy2sp":1, "sh2bd":1, "bd2sp":2, "eq2sp":3, "fl2fl":4}
|
|
|
+ "sy2fl":1, "sy2sh":1, "sy2sp":1, "sh2bd":1, "sp2sp_transport":1, "sp2sp_light":1, "sp2sp_convection":1, "bd2sp":2, "eq2sp":3, "fl2fl":4}
|
|
|
dependency = {"eq2bd":["eq2fl"], "eq2sh":["eq2sp"], "eq2sp":["eq2fl", "sp2fl"], "pe2bd":["eq2bd"], "pe2fl":["eq2fl"], "pe2sh":["eq2sh"],
|
|
|
"pe2sp":["eq2sp"], "sy2bd":["eq2bd"], "sy2fl":["eq2fl"], "sy2sh":["eq2sh"], "sy2sp":["eq2sp"]}
|
|
|
def get_type_by_func_name(func_name):
|
|
|
- return func_name.split('_')[-1]
|
|
|
+ if "_" in func_name:
|
|
|
+ return func_name[int(func_name.index("_")) + 1:]
|
|
|
+ else:
|
|
|
+ return func_name
|
|
|
try:
|
|
|
with plpy.subtransaction():
|
|
|
type = get_type_by_func_name(func)
|