整理项目结构,优化配置项

This commit is contained in:
2024-07-12 13:51:47 +08:00
parent 0f672c7961
commit ab15cb1fc3
33 changed files with 275 additions and 506 deletions

View File

@@ -1,26 +0,0 @@
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