commonlib-reader 1.0.8__tar.gz → 1.0.9__tar.gz

{commonlib_reader-1.0.8 → commonlib_reader-1.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: commonlib-reader
-Version: 1.0.8
+Version: 1.0.9
 Summary: Reader for Equinor commonlib api.
 License: MIT
 Author: Åsmund Våge Fannemel

{commonlib_reader-1.0.8 → commonlib_reader-1.0.9}/commonlib_reader/facility.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Union
 
 from commonlib_reader.utils import attributes_list_to_dict, get_code
 
@@ -75,6 +75,29 @@ class Facility:
     def __str__(self):
         return f"Facility: {self.identity}, Name: {self.name}, Description: {self.description}"
 
+    @staticmethod
+    def get_inst_code(facility: Union[str, int]) -> str:
+        """
+        Retrieves the STID installation code for the given facility identifier.
+        If the facility is not a STID facility, it recursively retrieves the installation code of its parent facility.
+
+        Parameters:
+            facility (str or int): The facility identifier, which can be a STID plant code, SAP code, or governmental facility name.
+        Returns:
+            str: The installation code corresponding to the facility.
+        Raises:
+            ValueError: If no facility is found for the provided identifier.
+        """
+
+        f = Facility(facility)
+        if isinstance(f, Facility):
+            if f.isSTID():
+                return f.STID
+            else:
+                return Facility.get_inst_code(f.ParentFacility)
+
+        raise ValueError(f"No facility found for {facility}")
+
     @classmethod
     def get_all_facilities(cls) -> List["Facility"]:
         """

{commonlib_reader-1.0.8 → commonlib_reader-1.0.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "commonlib-reader"
-version = "1.0.8"
+version = "1.0.9"
 description = "Reader for Equinor commonlib api."
 authors = [
 	{ name = "Åsmund Våge Fannemel", email = "asmf@equinor.com" }