halib 0.1.96__tar.gz → 0.1.98__tar.gz

{halib-0.1.96 → halib-0.1.98}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.1.96
+Version: 0.1.98
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -40,6 +40,7 @@ Requires-Dist: timebudget
 Requires-Dist: tqdm
 Requires-Dist: tube_dl
 Requires-Dist: wandb
+Requires-Dist: ipynbname
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -52,8 +53,11 @@ Dynamic: summary
 
 # Helper package for coding and automation
 
+**Version 0.1.98**
++ `filetype/ipynb`: add `gen_ipynb_name` generator to create file name based on current notebook name as prefix (with optional timestamp)
+
 **Version 0.1.96**
-+ `research/plot': add `PlotHelper` class to plot train history + plot grid of images (e.g., image samples from dataset or model outputs)
++ `research/plot`: add `PlotHelper` class to plot train history + plot grid of images (e.g., image samples from dataset or model outputs)
 
 
 **Version 0.1.91**

{halib-0.1.96 → halib-0.1.98}/README.md

@@ -1,7 +1,10 @@
 # Helper package for coding and automation
 
+**Version 0.1.98**
++ `filetype/ipynb`: add `gen_ipynb_name` generator to create file name based on current notebook name as prefix (with optional timestamp)
+
 **Version 0.1.96**
-+ `research/plot': add `PlotHelper` class to plot train history + plot grid of images (e.g., image samples from dataset or model outputs)
++ `research/plot`: add `PlotHelper` class to plot train history + plot grid of images (e.g., image samples from dataset or model outputs)
 
 
 **Version 0.1.91**

halib-0.1.98/halib/filetype/ipynb.py

@@ -0,0 +1,63 @@
+from contextlib import contextmanager
+from pathlib import Path
+
+import ipynbname
+
+from ..common import console, now_str
+
+
+@contextmanager
+def gen_ipynb_name(
+    filename,
+    add_time_stamp=False,
+    nb_prefix="nb__",
+    separator="__",
+):
+    """
+    Context manager that prefixes the filename with the notebook name.
+    Output: NotebookName_OriginalName.ext
+    """
+    try:
+        nb_name = ipynbname.name()
+    except FileNotFoundError:
+        nb_name = "script"  # Fallback
+
+    p = Path(filename)
+
+    # --- FIX START ---
+
+    # 1. Get the parts separately
+    original_stem = p.stem  # "test" (no extension)
+    extension = p.suffix  # ".csv"
+
+    now_string = now_str() if add_time_stamp else ""
+
+    # 2. Construct the base name (Notebook + Separator + OriginalName)
+    base_name = f"{nb_prefix}{nb_name}{separator}{original_stem}"
+
+    # 3. Append timestamp if needed
+    if now_string:
+        base_name = f"{base_name}{separator}{now_string}"
+
+    # 4. Add the extension at the VERY END
+    new_filename = f"{base_name}{extension}"
+
+    # --- FIX END ---
+
+    final_path = p.parent / new_filename
+
+    # Assuming you use 'rich' console based on your snippet
+    # console.rule()
+    # print(f"📝 Saving as: {final_path}")
+
+    yield str(final_path)
+
+
+if __name__ == "__main__":
+    # --- Usage Example ---
+    # Assume Notebook Name is: "MyThesisWork"
+    filename = "results.csv"
+    with gen_ipynb_name(filename) as filename_ipynb:
+        # filename_ipynb is now: "MyThesisWork_results.csv"
+        print(f"File to save: {filename_ipynb}")
+        # df.to_csv(filename_ipynb)

{halib-0.1.96 → halib-0.1.98}/halib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.1.96
+Version: 0.1.98
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -40,6 +40,7 @@ Requires-Dist: timebudget
 Requires-Dist: tqdm
 Requires-Dist: tube_dl
 Requires-Dist: wandb
+Requires-Dist: ipynbname
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -52,8 +53,11 @@ Dynamic: summary
 
 # Helper package for coding and automation
 
+**Version 0.1.98**
++ `filetype/ipynb`: add `gen_ipynb_name` generator to create file name based on current notebook name as prefix (with optional timestamp)
+
 **Version 0.1.96**
-+ `research/plot': add `PlotHelper` class to plot train history + plot grid of images (e.g., image samples from dataset or model outputs)
++ `research/plot`: add `PlotHelper` class to plot train history + plot grid of images (e.g., image samples from dataset or model outputs)
 
 
 **Version 0.1.91**

{halib-0.1.96 → halib-0.1.98}/halib.egg-info/SOURCES.txt

@@ -16,6 +16,7 @@ halib.egg-info/requires.txt
 halib.egg-info/top_level.txt
 halib/filetype/__init__.py
 halib/filetype/csvfile.py
+halib/filetype/ipynb.py
 halib/filetype/jsonfile.py
 halib/filetype/textfile.py
 halib/filetype/videofile.py

{halib-0.1.96 → halib-0.1.98}/halib.egg-info/requires.txt

@@ -28,3 +28,4 @@ timebudget
 tqdm
 tube_dl
 wandb
+ipynbname

{halib-0.1.96 → halib-0.1.98}/setup.py

@@ -8,7 +8,7 @@ with open("requirements.txt") as f:
 
 setuptools.setup(
     name="halib",
-    version="0.1.96",
+    version="0.1.98",
     author="Hoang Van Ha",
     author_email="hoangvanhauit@gmail.com",
     description="Small library for common tasks",