pystashx 0.1.0__py3-none-any.whl

pystash/__init__.py

@@ -0,0 +1 @@
+# pystash package

pystash/cli.py

@@ -0,0 +1,126 @@
+"""
+pystash CLI — Interactive two-level menu to extract template scripts.
+
+Usage:
+    $ pystash
+"""
+
+import os
+import re
+import shutil
+import sys
+from importlib.resources import files
+
+
+def _natural_key(text):
+    """Sort strings containing numbers in human/numerical order (e.g. q1 before q10)."""
+    return [int(c) if c.isdigit() else c.lower() for c in re.split(r"(\d+)", text)]
+
+
+def _get_templates_root():
+    """Return a Traversable pointing at the bundled templates/ directory."""
+    return files("pystash") / "templates"
+
+
+def _list_subdirs(root):
+    """Return sorted list of subdirectory names inside *root*."""
+    return sorted(
+        item.name
+        for item in root.iterdir()
+        if item.is_dir() and not item.name.startswith("__")
+    )
+
+
+def _list_py_files(folder):
+    """Return numerically sorted list of .py filenames inside a Traversable *folder*."""
+    return sorted(
+        (
+            item.name
+            for item in folder.iterdir()
+            if item.is_file()
+            and item.name.endswith(".py")
+            and item.name != "__init__.py"
+        ),
+        key=_natural_key,
+    )
+
+
+def _extract_file(source, dest_dir):
+    """Copy a single Traversable file into *dest_dir* using shutil.copy."""
+    dest_path = os.path.join(dest_dir, source.name)
+    with open(dest_path, "wb") as out, source.open("rb") as inp:
+        shutil.copyfileobj(inp, out)
+    return dest_path
+
+
+def copy_programs():
+    """Entry-point: two-level interactive menu -> extract scripts to cwd."""
+    templates = _get_templates_root()
+    categories = _list_subdirs(templates)
+
+    if not categories:
+        print("\n[!] No template categories found inside the package.")
+        print("    Add subdirectories to  src/pystash/templates/  and rebuild.")
+        sys.exit(1)
+
+    # -- Level 1 : Folder Selection ---------------------------------------
+    print("\n+======================================+")
+    print("|       PyStash   --  Templates        |")
+    print("+======================================+")
+    print("\nAvailable categories:\n")
+    for idx, name in enumerate(categories, start=1):
+        print(f"  [{idx}] {name}")
+
+    try:
+        choice = int(input("\n>> Select a category number: "))
+        if choice < 1 or choice > len(categories):
+            raise ValueError
+    except (ValueError, EOFError):
+        print("[X] Invalid selection. Exiting.")
+        sys.exit(1)
+
+    selected_category = categories[choice - 1]
+    folder = templates / selected_category
+
+    # -- Level 2 : File Selection -----------------------------------------
+    py_files = _list_py_files(folder)
+
+    if not py_files:
+        print(f"\n[!] No .py files found in '{selected_category}/'.")
+        print("    Drop your scripts into that folder and rebuild the package.")
+        sys.exit(1)
+
+    print(f"\n-- Scripts in  {selected_category}/  --\n")
+    print("  [0] => Extract ALL programs from this folder")
+    for idx, fname in enumerate(py_files, start=1):
+        print(f"  [{idx}] {fname}")
+
+    try:
+        file_choice = int(input("\n>> Select a file number (0 for all): "))
+        if file_choice < 0 or file_choice > len(py_files):
+            raise ValueError
+    except (ValueError, EOFError):
+        print("[X] Invalid selection. Exiting.")
+        sys.exit(1)
+
+    dest_dir = os.getcwd()
+
+    if file_choice == 0:
+        # Extract every .py file in the chosen category
+        print()
+        for fname in py_files:
+            src = folder / fname
+            path = _extract_file(src, dest_dir)
+            print(f"  [+] Extracted: {fname}  ->  {path}")
+        print(f"\n[OK] All {len(py_files)} file(s) from '{selected_category}/' extracted to:\n     {dest_dir}")
+    else:
+        # Extract a single file
+        fname = py_files[file_choice - 1]
+        src = folder / fname
+        path = _extract_file(src, dest_dir)
+        print(f"\n  [+] Extracted: {fname}  ->  {path}")
+        print(f"\n[OK] File extracted to:\n     {dest_dir}")
+
+
+if __name__ == "__main__":
+    copy_programs()

pystash/templates/DIP/__init__.py

@@ -0,0 +1 @@
+# DIP — Digital Image Processing templates

pystash/templates/DIP/dft.py

@@ -0,0 +1,17 @@
+import cv2
+import numpy as np
+
+img = cv2.imread("image.jpg", 0)
+
+f = np.fft.fftshift(np.fft.fft2(img))
+mag = 20 * np.log(np.abs(f) + 1)
+m_norm = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX).astype(np.uint8)
+
+inv = np.fft.ifft2(np.fft.ifftshift(f)).real
+i_norm = np.clip(inv, 0, 255).astype(np.uint8)
+
+cv2.imshow("Original", img)
+cv2.imshow("DFT", m_norm)
+cv2.imshow("Inverse DFT", i_norm)
+cv2.waitKey(0)
+cv2.destroyAllWindows()

pystash/templates/DIP/greyscale.py

@@ -0,0 +1,11 @@
+import cv2
+
+g = cv2.imread("image.jpg", 0)
+c = cv2.imread("image.jpg")
+gray = cv2.cvtColor(c, cv2.COLOR_BGR2GRAY)
+
+cv2.imshow("Gray", g)
+cv2.imshow("Color", c)
+cv2.imshow("Converted", gray)
+cv2.waitKey(0)
+cv2.destroyAllWindows()

pystash/templates/DIP/noise.py

@@ -0,0 +1,23 @@
+import cv2
+import numpy as np
+
+img = cv2.imread("image.jpg", 0).astype(float)
+sp = img.copy()
+r = np.random.rand(*img.shape)
+sp[r < 0.02], sp[r > 0.98] = 0, 255
+
+noises = [
+    img + np.random.normal(0, 25, img.shape),
+    sp,
+    img + np.random.rayleigh(20, img.shape),
+    img + np.random.gamma(2, 10, img.shape),
+    img + np.random.exponential(20, img.shape),
+]
+
+for x, name in zip(noises, ["Gaussian", "Salt-Pepper", "Rayleigh", "Gamma", "Exponential"]):
+    x = np.uint8(np.clip(x, 0, 255))
+    cv2.imshow(name, x)
+    cv2.imshow(name + " Mean", cv2.blur(x, (3, 3)))
+
+cv2.waitKey(0)
+cv2.destroyAllWindows()

pystash/templates/DIP/rghscmy.py

@@ -0,0 +1,16 @@
+import cv2
+
+img = cv2.imread("image.jpg")
+b, g, r = cv2.split(img)
+cmy = 255 - img
+hsi = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
+
+cv2.imshow("Original Image", img)
+cv2.imshow("Blue Channel", cv2.merge([b, g * 0, r * 0]))
+cv2.imshow("Green Channel", cv2.merge([b * 0, g, r * 0]))
+cv2.imshow("Red Channel", cv2.merge([b * 0, g * 0, r]))
+cv2.imshow("CMY Image", cmy)
+cv2.imshow("HSI Image", hsi)
+
+cv2.waitKey(0)
+cv2.destroyAllWindows()

pystash/templates/DIP/sampling.py

@@ -0,0 +1,11 @@
+import cv2
+
+img = cv2.imread("image.jpg", 0)
+s = cv2.resize(img, (50, 50))
+q = (img // 64) * 64
+
+cv2.imshow("Original", img)
+cv2.imshow("Sampling", s)
+cv2.imshow("Quantization", q)
+cv2.waitKey(0)
+cv2.destroyAllWindows()

pystash/templates/DIP/sobel_prewitt.py

@@ -0,0 +1,18 @@
+import cv2
+import numpy as np
+
+img = cv2.imread("image.jpg", 0)
+
+sx = cv2.Sobel(img, cv2.CV_32F, 1, 0)
+sy = cv2.Sobel(img, cv2.CV_32F, 0, 1)
+s = cv2.convertScaleAbs(cv2.magnitude(sx, sy))
+
+k = np.array([[-1, 0, 1], [-1, 0, 1], [-1, 0, 1]])
+p = cv2.filter2D(img, cv2.CV_32F, k)
+q = cv2.filter2D(img, cv2.CV_32F, k.T)
+t = cv2.convertScaleAbs(cv2.magnitude(p, q))
+
+cv2.imshow("Sobel", s)
+cv2.imshow("Prewitt", t)
+cv2.waitKey(0)
+cv2.destroyAllWindows()

pystash/templates/__init__.py

@@ -0,0 +1 @@
+# This file marks templates/ as a Python package so importlib.resources can traverse it.

pystashx-0.1.0.dist-info/METADATA

@@ -0,0 +1,11 @@
+Metadata-Version: 2.5
+Name: pystashx
+Version: 0.1.0
+Summary: Interactive CLI to extract categorized Python script templates into your working directory.
+Author: Anonymous
+License-Expression: MIT
+Requires-Python: >=3.9
+Requires-Dist: matplotlib
+Requires-Dist: numpy
+Requires-Dist: opencv-python-headless
+Requires-Dist: scipy

pystashx-0.1.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+pystash/__init__.py,sha256=9zsAzsFgjecb55uaTlWyiI3usMLfD3s3jkP-HLOrbzw,18
+pystash/cli.py,sha256=E1WnsxPOot4mIdsbtu1WWhCFg0w5u996FlG_45TSunw,4042
+pystash/templates/__init__.py,sha256=LC9s5JtsJR0pCiV05VdwECWe3cgJXxWbjbBDYlunsDs,89
+pystash/templates/DIP/Image.jpg,sha256=8GUlNcA7eWr8c-XmYRqpKxIb9AS_YbRw_sQjJM8Qgw4,48735
+pystash/templates/DIP/__init__.py,sha256=GojToSl1rlo7zESo0HsORfwgMti9JUGjcKmRSHFOfIs,45
+pystash/templates/DIP/dft.py,sha256=1iMXxmXit3kQIR81BugN89rJyNejwKMLFXourmCeFKU,433
+pystash/templates/DIP/greyscale.py,sha256=CFVrf2NaJJI8S1CcaYZIpu3Z-CJktyrkhgh-zvnl6T0,229
+pystash/templates/DIP/noise.py,sha256=i_9nDBi0P_p5Giek7HSipg-uh67tQovOnBIjSCQocts,614
+pystash/templates/DIP/rghscmy.py,sha256=rGPDlwhQ6t14Nn3mE7ULm-hgyYtwG5h2aKsamH5Q_mE,430
+pystash/templates/DIP/sampling.py,sha256=tgzmVLteI0L0NeL7QvOV6_gaFsJAPu-QGvaSvqqpowQ,220
+pystash/templates/DIP/sobel_prewitt.py,sha256=looMjt3ziuxUQQeM92QZ615Jomape-PdtecjOgh-1kI,449
+pystashx-0.1.0.dist-info/METADATA,sha256=3JHqabVVYSQw58dTB21ulwSeNtfpnEfDZzJnaFdEG0Y,324
+pystashx-0.1.0.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
+pystashx-0.1.0.dist-info/entry_points.txt,sha256=JivIJxL1UGZgR8WjhdtN3HTJ07Gf8FFaW4cu-RYxE4U,91
+pystashx-0.1.0.dist-info/RECORD,,

pystashx-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.33.0
+Root-Is-Purelib: true
+Tag: py3-none-any

pystashx-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+pystash = pystash.cli:copy_programs
+pystashx = pystash.cli:copy_programs