Merge pull request #275 from ESSS/fb-ASIM-4926-change-esp-workflow

ASIM-4926-Change esp workflow
ESSS · Nov 17, 2022 · 491c6d9 · 491c6d9
2 parents 8fd627f + c1f096f
commit 491c6d9
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/alfasim_sdk/__init__.py b/src/alfasim_sdk/__init__.py
@@ -254,6 +254,7 @@ def get_alfasim_sdk_api_path():
 from alfasim_sdk._internal.constants import MassSourceType
 from alfasim_sdk._internal.constants import MaterialType
 from alfasim_sdk._internal.constants import MultiInputType
+from alfasim_sdk._internal.constants import EspParameters
 from alfasim_sdk._internal.constants import NodeCellType
 from alfasim_sdk._internal.constants import NonlinearSolverType
 from alfasim_sdk._internal.constants import OIL_FIELD
@@ -431,6 +432,7 @@ def get_alfasim_sdk_api_path():
     "MaterialDescription",
     "MaterialType",
     "MultiInputType",
+    "EspParameters",
     "MultipleReference",
     "NodeCellType",
     "NodeDescription",

diff --git a/src/alfasim_sdk/_internal/constants.py b/src/alfasim_sdk/_internal/constants.py
@@ -364,6 +364,17 @@ class PumpType(Enum):
     ElectricSubmersiblePump = "electric_submersible_pump"
 
 
+class EspParameters(Enum):
+    """
+    Defines if a esp is imported from catalog or defined by user.
+        - ``UserDefined``: The user will define the esp table values.
+        - ``Catalog``: The esp table values will be populated by the data imported from the catalog.
+    """
+
+    UserDefined = "user_defined"
+    Catalog = "catalog"
+
+
 class ValveOpeningType(Enum):
     ConstantOpening = "constant_opening"
     TableInterpolation = "table_interpolation"