Kaynağa Gözat

每份字段前移

chencheng 1 yıl önce
ebeveyn
işleme
1e49131c71
1 değiştirilmiş dosya ile 14 ekleme ve 8 silme
  1. 14 8
      cores/post_decorators.py

+ 14 - 8
cores/post_decorators.py

@@ -66,9 +66,12 @@ def rule4_decorator(f, *args, **kwargs):
     '''
     predict_line = args[1]
     predict_line = f(*args, **kwargs)
-    if '项目' in predict_line[1] and predict_line[0]=='':
-        predict_line[0] = '项目'
-        predict_line[1] = predict_line[1].replace('项目', '')
+    try:
+        if '项目' in predict_line[1] and predict_line[0]=='':
+            predict_line[0] = '项目'
+            predict_line[1] = predict_line[1].replace('项目', '')
+    except IndexError as e:
+        print('rule4_decorator', e)
     return predict_line
 
 
@@ -76,11 +79,14 @@ def rule4_decorator(f, *args, **kwargs):
 def rule5_decorator(f, *args, **kwargs):
     predict_line = args[1]
     predict_line = f(*args, **kwargs)
-    if '项目' in predict_line[0] and '营养素参考值' in predict_line[1] and len(predict_line[1])>7 and predict_line[2] == '':
-        predict_line[2] = '营养素参考值'
-        if len(predict_line[1].split('营养素参考值'))>1:
-            predict_line[2] = predict_line[2]+predict_line[1].split('营养素参考值')[1]
-        predict_line[1] = predict_line[1].split('营养素参考值')[0]
+    try:
+        if '项目' in predict_line[0] and '营养素参考值' in predict_line[1] and len(predict_line[1])>7 and predict_line[2] == '':
+            predict_line[2] = '营养素参考值'
+            if len(predict_line[1].split('营养素参考值'))>1:
+                predict_line[2] = predict_line[2]+predict_line[1].split('营养素参考值')[1]
+            predict_line[1] = predict_line[1].split('营养素参考值')[0]
+    except IndexError as e:
+        print('rule5_decorator', e)
     return predict_line