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