使用并发调用api方式获取信息抽取结果

photo_review/enumeration/task.py

@@ -0,0 +1,26 @@
+from enum import Enum
+
+from config.keys import SETTLEMENT_LIST_SCHEMA, DISCHARGE_RECORD_SCHEMA, COST_LIST_SCHEMA
+
+
+class TaskEnum(Enum):
+    SETTLEMENT = "SETTLEMENT"
+    DISCHARGE = "DISCHARGE"
+    COST = "COST"
+
+    def request_url(self):
+        base_url = "http://192.168.5.9:8000"
+        if self == TaskEnum.SETTLEMENT:
+            return base_url + "/nlp/ie/settlement"
+        elif self == TaskEnum.DISCHARGE:
+            return base_url + "/nlp/ie/discharge"
+        elif self == TaskEnum.COST:
+            return base_url + "/nlp/ie/cost"
+
+    def schema(self):
+        if self == TaskEnum.SETTLEMENT:
+            return SETTLEMENT_LIST_SCHEMA
+        elif self == TaskEnum.DISCHARGE:
+            return DISCHARGE_RECORD_SCHEMA
+        elif self == TaskEnum.COST:
+            return COST_LIST_SCHEMA