Selaa lähdekoodia

每份字段后移

chencheng 1 vuosi sitten
vanhempi
commit
e85380da02
2 muutettua tiedostoa jossa 22 lisäystä ja 0 poistoa
  1. 21 0
      cores/post_decorators.py
  2. 1 0
      cores/post_hander.py

+ 21 - 0
cores/post_decorators.py

@@ -82,3 +82,24 @@ def rule5_decorator(f, *args, **kwargs):
             predict_line[2] = predict_line[2]+predict_line[1].split('营养素参考值')[1]
         predict_line[1] = predict_line[1].split('营养素参考值')[0]
     return predict_line
+
+
+def rule6_decorator(f, *args, **kwargs):
+    '''
+    predict_line = ['项目 ', '', '每份(70g)营养素参考值%', '']
+    '''
+    predict_line = args[1]
+    predict_line = f(*args, **kwargs)
+    idx = predict_line.index('')
+    dictionary = ['每份']
+    try:
+        if idx == 1:
+            for dict in dictionary:
+                if '项目' in predict_line[0] and dict in predict_line[2]:
+                    predict_line[1] = dict
+                    r = re.split(dict, predict_line[2])
+                    if len(r) == 2 and r[1]:
+                        predict_line[2] = r[1]
+    except IndexError as e:
+        print('rule1_decorator', e)
+    return predict_line

+ 1 - 0
cores/post_hander.py

@@ -39,6 +39,7 @@ class PostHandler:
         else:
             return self.predict_html
 
+    @rule6_decorator
     @rule5_decorator
     @rule4_decorator
     @rule3_decorator