gimu 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

gimu/gmf/data.py

@@ -11,6 +11,29 @@ import json
 import os
 from collections import OrderedDict
 
+
+class WellData(dict):
+    def __init__(self, *args, **kwargs):
+            super().__init__(*args, **kwargs)
+
+    def downhole_temperature(self, date_str):
+        """ get direct access to data without fluff """
+        try:
+            raw_data = self['Downhole_Temperature'][date_str]['Data']
+        except KeyError as e:
+            print(f"Data '{date_str}' does not exist in {self['Well_Name']}")
+            print(f"-> available data set are: {list(self['Downhole_Temperature'].keys())}")
+            raise e
+        elevs, temps = [], []
+        for elev_str, temp in raw_data.items():
+            elev = float(elev_str)
+            elevs.append(elev)
+            temps.append(temp)
+
+        # TODO takes care of units in GMF
+        return elevs, temps
+
+
 class DataJSON:
     def __init__(self, max_cache_size=10):
         self._data_folders = []
@@ -44,7 +67,7 @@ class DataJSON:
             file_path = os.path.join(folder, f"{well_name}_data.json")
             if os.path.exists(file_path):
                 with open(file_path, 'r') as f:
-                    data = json.load(f)
+                    data = WellData(json.load(f))
 
                 if len(self._cache) >= self._max_cache_size:
                     self._cache.popitem(last=False)
@@ -52,7 +75,10 @@ class DataJSON:
                 self._cache[well_name] = data
                 return data
 
-        raise KeyError(f"Well '{well_name}' not found in any of the data folders.")
+        msg = '\n',join([
+            f"Well '{well_name}' not found in any of the data folders:",
+            ] + [f"    {f}" for f in self._data_folders])
+        raise KeyError(msg)
 
     def __contains__(self, well_name):
         if well_name in self._cache:

{gimu-0.2.0.dist-info → gimu-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gimu
-Version: 0.2.0
+Version: 0.3.0
 Summary: A toolkit and python library for modelling at Geothermal Institute, University of Auckland.
 Project-URL: Documentation, https://github.com/cyeh015/gimu/blob/main/README.md
 Project-URL: Issues, https://github.com/cyeh015/gimu/issues

{gimu-0.2.0.dist-info → gimu-0.3.0.dist-info}/RECORD

@@ -8,9 +8,9 @@ gimu/t2listingh5.py,sha256=sepCah3lV6etWr-aJ_Tc9d3Rubio7DitEn0SQP2oJ2E,16628
 gimu/waiwera_copy.py,sha256=G84UQWbVq--cfkETCs_jpcquTL-x5SftAFNNNs3GfJs,1286
 gimu/waiwera_listing.py,sha256=c7moQY8DQEH_MqLUuhrinnkEHnDMKWx4srChffXoK6I,27787
 gimu/gmf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gimu/gmf/data.py,sha256=aSpm_GZchVV_TCoBw9ymi5btT--53K2ibXCpmV-591I,2711
-gimu-0.2.0.dist-info/METADATA,sha256=A_BuKk58QciB8cBVdmX1bdYn1Oss2rFYDvztTX6rCOc,2320
-gimu-0.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-gimu-0.2.0.dist-info/entry_points.txt,sha256=FbtLXv_XXn6Wh2DXRsqJpTp85lxEvJiE36TTUTa0e1o,129
-gimu-0.2.0.dist-info/licenses/LICENSE.txt,sha256=E2nlaUhQtke9PAEhVEbZfNCPa-qJYGSeuR6Q3f6XgOM,1121
-gimu-0.2.0.dist-info/RECORD,,
+gimu/gmf/data.py,sha256=qzvouw-l88nDtNGMmTGKMGrtizcck1lHTbgcU5uv7UQ,3587
+gimu-0.3.0.dist-info/METADATA,sha256=Fx3AVaIO98Pt56YAJv_tQuHL2uidMPVsVjviiSl6XP0,2320
+gimu-0.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+gimu-0.3.0.dist-info/entry_points.txt,sha256=FbtLXv_XXn6Wh2DXRsqJpTp85lxEvJiE36TTUTa0e1o,129
+gimu-0.3.0.dist-info/licenses/LICENSE.txt,sha256=E2nlaUhQtke9PAEhVEbZfNCPa-qJYGSeuR6Q3f6XgOM,1121
+gimu-0.3.0.dist-info/RECORD,,