shancx 1.9.33.227__py3-none-any.whl → 1.9.33.229__py3-none-any.whl

shancx/Read.py

@@ -55,30 +55,9 @@ result = getGrib(input_Path)
 import pygrib
 import numpy as np
 import pandas as pd
-import gzip
-import tempfile
-def readGrib(filepath, target_param=None):
-    filepath= filepath  
-    if filepath.endswith('.gz'):
-        with gzip.open(filepath, 'rb') as f:
-            with tempfile.NamedTemporaryFile(suffix='.grib2', delete=False) as tmp:
-                tmp.write(f.read())
-                filepath = tmp.name  
+def readGrib(file_path, target_param=None):
     try:
-        with pygrib.open(filepath) as grbs:         
-            if target_param:
-                try:
-                    grb = grbs.select(shortName=target_param)[0]
-                except:
-                    try:
-                        grb = grbs.select(parameterName=target_param)[0]
-                    except:
-                        raise ValueError(f"未找到参数: {target_param}")
-            else:
-                grb = grbs[1]
-            data = grb.values
-            lats, lons = grb.latlons() 
-            #data2 
+        with pygrib.open(file_path) as grbs:
             field_info = []
             for grb in grbs:
                 field_info.append({
@@ -91,6 +70,18 @@ def readGrib(filepath, target_param=None):
                                    'units': getattr(grb, 'units', 'N/A'),
                                    'shape': grb.values.shape
                                  })             
+            if target_param:
+                try:
+                    grb = grbs.select(shortName=target_param)[0]
+                except:
+                    try:
+                        grb = grbs.select(parameterName=target_param)[0]
+                    except:
+                        raise ValueError(f"未找到参数: {target_param}")
+            else:
+                grb = grbs[1]
+            data = grb.values
+            lats, lons = grb.latlons()            
             return {
                 'data': data,
                 'lats': lats,
@@ -99,8 +90,7 @@ def readGrib(filepath, target_param=None):
                     'parameterName': grb.parameterName,
                     'level': grb.level,
                     'validDate': grb.validDate,
-                    'units': grb.units,
-                    "field_info":field_info                    
+                    'units': grb.units
                 }
             }            
     except Exception as e:

shancx/npz.py

@@ -1,14 +1,28 @@
 import numpy as np
 from pathlib import Path
+import os
+import uuid
+import traceback
 def npzSave(output, data):
-    output = output.replace('.npy', '.npz')
-    output_path = Path(output).with_suffix('.npz')
-    output_path.parent.mkdir(parents=True, exist_ok=True)
-    np.savez_compressed(output_path, data=data)
-    print(f"{output_path} done")
+    try:
+        output_path = Path(output)
+        if output_path.suffix in ['.npy', '.nc', '.h5', '.hdf5']:
+            output_path = output_path.with_suffix('.npz')    
+        output_path.parent.mkdir(parents=True, exist_ok=True)
+        temp_filename = output_path.with_suffix(f'.tmp.{uuid.uuid4()}.npz')    
+        np.savez_compressed(temp_filename, data=data)
+        os.replace(temp_filename, output_path)   
+        print(f"{output_path} done")
+    except Exception as e:
+        print(traceback.format_exc())
+        if temp_filename and temp_filename.exists():
+            try:
+                os.remove(temp_filename)
+                print(f"⚠ Temporary files have been cleared: {temp_filename}")
+            except:
+                pass            
 def npzLoad(output_path):
     return np.load(Path(output_path).with_suffix('.npz'))['data']
-
 """
 npzSave(output, data)
 npzLoad(output_path)

{shancx-1.9.33.227.dist-info → shancx-1.9.33.229.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shancx
-Version: 1.9.33.227
+Version: 1.9.33.229
 Summary: A simple timer decorator
 Home-page: https://gitee.com/shancx
 Author: shancx

{shancx-1.9.33.227.dist-info → shancx-1.9.33.229.dist-info}/RECORD

@@ -5,7 +5,7 @@ shancx/Lib.py,sha256=GUAspllSxk39mvj-F1Q8Ys0EcY_lQfZPRGPE7L3x4SE,977
 shancx/Path1.py,sha256=vX4A5RgdwVyIVVNZRocl18rOu1Z8YMLfDb0B92-0bE8,5334
 shancx/Point.py,sha256=gyIomOVbNoz6SOcAhhTS26NHBeJ0TOwB-ljNsBWN1ZE,1909
 shancx/QC.py,sha256=MFxbPMEwrnk5l0sTFMrx13wPlXJ5WI5TovzaiBoEr0E,9848
-shancx/Read.py,sha256=94FNUWiZn71X6ArZhyqhyORNcWJffC7Hz76BAtF9CJc,4577
+shancx/Read.py,sha256=OCSiKbt_yIFFUWSDy6T83KlOYYMHOP628aXpatQCkjw,4197
 shancx/Resize.py,sha256=uwiOhD-aWPDLvIwOYEX-XXN_suOXv6hPDcaGyr4JLC4,4044
 shancx/ZR.py,sha256=5APchqVoI1G2MkrN9YJSWCwfLUygcSFIsUXSo9aq1Qg,341
 shancx/__init__.py,sha256=6sfiHElECqfy5eEEGqaCAw4YiKxgKfJ5IFFh9yvtewk,11442
@@ -17,7 +17,7 @@ shancx/geosProj.py,sha256=JMmtj3kV8rtFkKbmwvTKuDWlC8AkgLG8JvCPzkn8Dfs,2685
 shancx/getResponse.py,sha256=QmJfa4czGCOiKb8RuCLXKE7AAKVADAptNiwn7v1tfbM,1055
 shancx/info.py,sha256=0rk_L8Z5uj9RliaZrzT-CArH9ZObmdmcp-1RxAItL08,1122
 shancx/netdfJU.py,sha256=96KR9NMLr2Kcs_OOMpX7QPujdWUj0GCEUIZ7_4_pBxw,7741
-shancx/npz.py,sha256=ZydULlx1ZhgwxrdIrVDm9lGa8KY4q1MJ9dZo_wQriso,462
+shancx/npz.py,sha256=AOE-_f9OItnJLTgH_6kQj5Fn4dOHtLJ6yWRRKk853gM,1045
 shancx/sendM.py,sha256=dcqGNPPy1n0NxrRgOrHZEpaGtLy15FS_lCYe6_TkvSs,3510
 shancx/wait.py,sha256=OgT9JmEgMH7d3ir3TEksx0_Kyl_S4bkOKl50jvVpdFA,8575
 shancx/3D/__init__.py,sha256=pDFvW1OOEW-ueXH50NJIYdWlySHZU6sr5htzHUSeROo,1000
@@ -87,7 +87,7 @@ shancx/Train/multiGpu.py,sha256=D_oZeiSc7VWktpnVDwrFOC1CYZSt9rxOKY5lngE5vFg,820
 shancx/Train/prepare.py,sha256=vL_8UOA66oZCBIwCICtihsGibivtNgaVJGulJxfNdn8,6793
 shancx/Train/renet50.py,sha256=wEhYk1X96WE5zuqHqVxWLJa-A5jDNkz4z6edORNufnA,6428
 shancx/tensBoard/__init__.py,sha256=ga2C5YyJITvvQA1ocpxna_KNFnNRJVwkTjLoIglLZUQ,993
-shancx-1.9.33.227.dist-info/METADATA,sha256=2IF7Rl9DDL9BPZypQ0PtDTid3CfYF7wOu9HjWhQO42Y,850
-shancx-1.9.33.227.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-shancx-1.9.33.227.dist-info/top_level.txt,sha256=akfCS1vKWz3pNmEN_yN9ZiGp-60IQY5ET38mRx_i_-4,7
-shancx-1.9.33.227.dist-info/RECORD,,
+shancx-1.9.33.229.dist-info/METADATA,sha256=bnZn5H1P1YztHINGA-DS0ObKNoJG9NCfKkmmP7iO2JA,850
+shancx-1.9.33.229.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+shancx-1.9.33.229.dist-info/top_level.txt,sha256=akfCS1vKWz3pNmEN_yN9ZiGp-60IQY5ET38mRx_i_-4,7
+shancx-1.9.33.229.dist-info/RECORD,,