26
photo_review/enumeration/task.py
Normal file
26
photo_review/enumeration/task.py
Normal file
@@ -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
|
||||
Reference in New Issue
Block a user