Source code for kappa_sdk.well_production_enum

from enum import Enum
import warnings
from typing import Any




class WellProductionTypeEnum(Enum):
    """ Well production type enumeration

    An enumeration of possible production types for a well
    """
    oil_producer = "OilProducer"
    gas_producer = "GasProducer"
    condensate_producer = "CondensateProducer"
    gas_injector = "GasInjector"
    water_injector = "WaterInjector"
    unknown = "Unknown"



class DeprecatedEnumProxy:
    def __init__(self, enum: WellProductionTypeEnum) -> None:
        self.enum = enum

    def __getattr__(self, name: str) -> Any:
        warnings.warn(
            f"WellProductionEnum is deprecated, please use {WellProductionTypeEnum.__name__} instead.",
            DeprecationWarning
        )
        return getattr(self.enum, name)


WellProductionEnum = DeprecatedEnumProxy(WellProductionTypeEnum)  # type:ignore[arg-type]