kenenet 1.1.7__py3-none-any.whl → 1.1.9__py3-none-any.whl

kenenet/__init__.py

@@ -183,6 +183,7 @@ def pp(msg='caca', subdir=None, pps=3):
 
 def save_img(img, name=' ', reset=True, file='temp_screenshots', mute=False):
     global ospid
+    # Ensure directory exists or clean if needed
     if os.path.exists(file):
         if reset and ospid is None:
             zhmiscellany.fileio.empty_directory(file)
@@ -190,30 +191,37 @@ def save_img(img, name=' ', reset=True, file='temp_screenshots', mute=False):
     else:
         quick_print(f'New folder created {file}')
         zhmiscellany.fileio.create_folder(file)
+
     ospid = True
     frame = inspect.currentframe().f_back
     lineno = frame.f_lineno
 
-    if not isinstance(img, np.ndarray):
-        quick_print(f"Your img is not a fucking numpy array you twat, couldn't save {name}", lineno)
-        return
+    if isinstance(img, np.ndarray):
+        # Detect mask arrays: floating-point, single-channel
+        is_mask = img.dtype.kind == 'f' and (img.ndim == 2 or (img.ndim == 3 and img.shape[-1] == 1))
 
-    save_name = name + f'{time.time()}'
-    arr = img
-    if arr.ndim == 3 and arr.shape[2] == 3:
-        im = Image.fromarray(arr)
-    else:
-        g = arr.squeeze()
-        if np.issubdtype(g.dtype, np.floating):
-            g = (np.clip(g, 0, 1) * 255).astype(np.uint8)
+        # Construct timestamped, sanitized filename
+        timestamp = time.time()
+        # Sanitize the provided name to remove invalid filesystem chars
+        safe_name = re.sub(r'[^A-Za-z0-9._-]', '_', name)
+        save_name = f"{safe_name}{timestamp}"
+
+        # Prepare image to save
+        if is_mask:
+            # Scale mask [0.0,1.0] to [0,255] grayscale
+            mask_uint8 = (img.squeeze() * 255).clip(0, 255).astype(np.uint8)
+            img_to_save = Image.fromarray(mask_uint8)
         else:
-            g = g.astype(np.uint8)
-        im = Image.fromarray(g, mode='L')
+            img_to_save = Image.fromarray(img)
 
-    path = os.path.join(file, f'{save_name}.png')
-    im.save(path)
-    if not mute:
-        quick_print(f'Saved image as {save_name}', lineno)
+        # Save file
+        save_path = os.path.join(file, f"{save_name}.png")
+        img_to_save.save(save_path)
+
+        if not mute:
+            quick_print(f'Saved image as {save_name}', lineno)
+    else:
+        quick_print(f"Your img is not a fucking numpy array you twat, couldn't save {name}", lineno)
 
 class _load_audio:
     def __init__(self):

{kenenet-1.1.7.dist-info → kenenet-1.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kenenet
-Version: 1.1.7
+Version: 1.1.9
 Summary: dude what the fuck even is this package
 Home-page: https://www.youtube.com/@KiddyKene
 Author: kiddykene

kenenet-1.1.9.dist-info/RECORD

@@ -0,0 +1,5 @@
+kenenet/__init__.py,sha256=1uA-m7NkdOIPdcZOpr4Zb7caDcrwbnXhqY0eR6AsM2U,21946
+kenenet-1.1.9.dist-info/METADATA,sha256=OIAT4qSfUC1S_I_VZQsz2qU00812VTSFLaw1TuUJ2yA,1693
+kenenet-1.1.9.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+kenenet-1.1.9.dist-info/top_level.txt,sha256=gUsWXLrM0jF4b4nbYJZdksdFewIx_F3xOF-zER8fMuQ,8
+kenenet-1.1.9.dist-info/RECORD,,

kenenet-1.1.7.dist-info/RECORD

@@ -1,5 +0,0 @@
-kenenet/__init__.py,sha256=6fDdc-6WpO8jipy2CyfYUYc6E3Q-RlD7M4iVtRZ9-6Y,21482
-kenenet-1.1.7.dist-info/METADATA,sha256=gT81Xu95QgcjcRP1jzRcKn1bMpu2NdQkYb3SF21wDX4,1693
-kenenet-1.1.7.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-kenenet-1.1.7.dist-info/top_level.txt,sha256=gUsWXLrM0jF4b4nbYJZdksdFewIx_F3xOF-zER8fMuQ,8
-kenenet-1.1.7.dist-info/RECORD,,