27 lines
810 B
Python
27 lines
810 B
Python
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
|