scitex 2.7.3__py3-none-any.whl → 2.10.0__py3-none-any.whl

scitex/io/__init__.py

@@ -1,32 +1,57 @@
 #!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-# Timestamp: "2025-07-12 04:23:59 (ywatanabe)"
-# File: /ssh:sp:/home/ywatanabe/proj/scitex_repo/src/scitex/io/__init__.py
+# Timestamp: "2025-12-16 (ywatanabe)"
+# File: /home/ywatanabe/proj/scitex-code/src/scitex/io/__init__.py
 # ----------------------------------------
 import os
 
 __FILE__ = __file__
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
-"""Scitex IO module with lazy imports to avoid circular dependencies."""
+"""Scitex IO module with lazy imports to avoid circular dependencies.
+
+Bundle I/O is handled via scitex.fts:
+    from scitex.fts import FTS
+    bundle = FTS("Figure1.stx")
+    bundle.save()
+"""
 
 # Import commonly used functions directly
-from ._save import save
+from ._cache import cache
+from ._flush import flush
+from ._glob import glob, parse_glob
 from ._load import load
 
-# Bundle I/O for .figz, .pltz, .statsz
-# Users should use module-specific save/load functions:
-#   - scitex.plt.save_pltz() / load_pltz()
-#   - scitex.fig.save_figz() / load_figz()
-#   - scitex.stats.save_statsz() / load_statsz()
-# Internal bundle functions available via scitex.io._bundle for module implementations
+# Bundle I/O - import the bundle submodule
+from . import bundle
+
 from ._load_configs import load_configs
-from ._glob import glob, parse_glob
 from ._reload import reload
-from ._flush import flush
-from ._cache import cache
-from ._load_modules._H5Explorer import H5Explorer, explore_h5, has_h5_key
-from ._load_modules._ZarrExplorer import ZarrExplorer, explore_zarr, has_zarr_key
+from ._save import save
+
+# Optional: HDF5 explorer (requires h5py)
+try:
+    from ._load_modules._H5Explorer import H5Explorer, explore_h5, has_h5_key
+except ImportError:
+    H5Explorer = None
+    explore_h5 = None
+    has_h5_key = None
+
+# Optional: Zarr explorer (requires zarr)
+try:
+    from ._load_modules._ZarrExplorer import ZarrExplorer, explore_zarr, has_zarr_key
+except ImportError:
+    ZarrExplorer = None
+    explore_zarr = None
+    has_zarr_key = None
+
+# Import load cache control functions
+from ._load_cache import (
+    clear_cache as clear_load_cache,
+)
+from ._load_cache import (
+    configure_cache,
+    get_cache_info,
+)
 
 # Import load cache control functions
 from ._load_cache import (
@@ -39,13 +64,13 @@ from ._load_cache import (
 try:
     from ._save_modules import (
         save_image,
-        save_text,
-        save_mp4,
         save_listed_dfs_as_csv,
         save_listed_scalars_as_csv,
+        save_mp4,
         save_optuna_study_as_csv_and_pngs,
+        save_text,
     )
-except ImportError as e:
+except ImportError:
     # Fallback for missing functions
     save_image = None
     save_text = None
@@ -79,7 +104,7 @@ except ImportError:
 
 # Import metadata functions
 try:
-    from ._metadata import read_metadata, embed_metadata, has_metadata
+    from ._metadata import embed_metadata, has_metadata, read_metadata
 except ImportError:
     read_metadata = None
     embed_metadata = None
@@ -89,6 +114,8 @@ __all__ = [
     # Primary I/O
     "save",
     "load",
+    # Bundle submodule
+    "bundle",
     # Config loading
     "load_configs",
     # File utilities

scitex/io/_load.py

@@ -1,9 +1,9 @@
 #!/usr/bin/env python3
-# -*- coding: utf-8 -*-
 # Timestamp: "2025-08-11 05:54:51 (ywatanabe)"
 # File: /home/ywatanabe/proj/SciTeX-Code/src/scitex/io/_load.py
 # ----------------------------------------
 from __future__ import annotations
+
 import os
 
 __FILE__ = __file__
@@ -15,37 +15,89 @@ from pathlib import Path
 from typing import Any, Union
 
 from scitex.decorators import preserve_doc
-from scitex.str._clean_path import clean_path
+
 from ._load_cache import (
     cache_data,
-    configure_cache,
-    get_cache_info,
     get_cached_data,
     load_npy_cached,
 )
-from ._load_modules._bibtex import _load_bibtex
 
-# from ._load_modules._catboost import _load_catboost
-from ._load_modules._con import _load_con
-from ._load_modules._docx import _load_docx
-from ._load_modules._eeg import _load_eeg_data
-from ._load_modules._hdf5 import _load_hdf5
-from ._load_modules._image import _load_image
-from ._load_modules._joblib import _load_joblib
+# Core loaders (no special dependencies)
+from ._load_modules._bibtex import _load_bibtex
 from ._load_modules._json import _load_json
 from ._load_modules._markdown import _load_markdown
-from ._load_modules._matlab import _load_matlab
 from ._load_modules._numpy import _load_npy
 from ._load_modules._pandas import _load_csv, _load_excel, _load_tsv
-from ._load_modules._pdf import _load_pdf
 from ._load_modules._pickle import _load_pickle
-from ._load_modules._sqlite3 import _load_db_sqlite3
-from ._load_modules._torch import _load_torch
 from ._load_modules._txt import _load_txt
 from ._load_modules._xml import _load_xml
 from ._load_modules._yaml import _load_yaml
-from ._load_modules._zarr import _load_zarr
-from ._load_modules._canvas import _load_canvas
+
+
+# Optional loaders - wrapped for missing dependencies
+def _unavailable_loader(name):
+    def _loader(*args, **kwargs):
+        raise ImportError(
+            f"Loader for {name} not available. Install required dependencies."
+        )
+
+    return _loader
+
+
+try:
+    from ._load_modules._con import _load_con
+except ImportError:
+    _load_con = _unavailable_loader("con")
+
+try:
+    from ._load_modules._docx import _load_docx
+except ImportError:
+    _load_docx = _unavailable_loader("docx")
+
+try:
+    from ._load_modules._eeg import _load_eeg_data
+except ImportError:
+    _load_eeg_data = _unavailable_loader("eeg")
+
+try:
+    from ._load_modules._hdf5 import _load_hdf5
+except ImportError:
+    _load_hdf5 = _unavailable_loader("hdf5")
+
+try:
+    from ._load_modules._image import _load_image
+except ImportError:
+    _load_image = _unavailable_loader("image")
+
+try:
+    from ._load_modules._joblib import _load_joblib
+except ImportError:
+    _load_joblib = _unavailable_loader("joblib")
+
+try:
+    from ._load_modules._matlab import _load_matlab
+except ImportError:
+    _load_matlab = _unavailable_loader("matlab")
+
+try:
+    from ._load_modules._pdf import _load_pdf
+except ImportError:
+    _load_pdf = _unavailable_loader("pdf")
+
+try:
+    from ._load_modules._sqlite3 import _load_db_sqlite3
+except ImportError:
+    _load_db_sqlite3 = _unavailable_loader("sqlite3")
+
+try:
+    from ._load_modules._torch import _load_torch
+except ImportError:
+    _load_torch = _unavailable_loader("torch")
+
+try:
+    from ._load_modules._zarr import _load_zarr
+except ImportError:
+    _load_zarr = _unavailable_loader("zarr")
 
 
 def _load_bundle(lpath, verbose=False, **kwargs):
@@ -70,18 +122,20 @@ def _load_bundle(lpath, verbose=False, **kwargs):
     For .statsz bundles:
         dict: Stats data with 'spec' and 'comparisons'.
     """
-    from ._bundle import load_bundle, BundleType
+    from .bundle import BundleType
+    from .bundle import load as load_bundle
 
     bundle = load_bundle(lpath)
-    bundle_type = bundle.get('type')
+    bundle_type = bundle.get("type")
 
     if bundle_type == BundleType.PLTZ:
         # Return (fig, ax, data) tuple for .pltz bundles
         # Note: We return the spec and data, not a reconstructed figure
         # as matplotlib figures cannot be perfectly serialized/deserialized
-        import matplotlib.pyplot as plt
         from pathlib import Path
 
+        import matplotlib.pyplot as plt
+
         p = Path(lpath)
         bundle_dir = p
 
@@ -89,13 +143,14 @@ def _load_bundle(lpath, verbose=False, **kwargs):
         if not p.is_dir():
             import tempfile
             import zipfile
+
             temp_dir = Path(tempfile.mkdtemp())
-            with zipfile.ZipFile(p, 'r') as zf:
+            with zipfile.ZipFile(p, "r") as zf:
                 zf.extractall(temp_dir)
             bundle_dir = temp_dir
 
         # Find PNG file - layered format stores in exports/
-        basename = bundle.get('basename', 'plot')
+        basename = bundle.get("basename", "plot")
         png_path = bundle_dir / "exports" / f"{basename}.png"
         if not png_path.exists():
             # Fallback to root level (legacy format)
@@ -106,28 +161,28 @@ def _load_bundle(lpath, verbose=False, **kwargs):
             img = plt.imread(str(png_path))
             fig, ax = plt.subplots()
             ax.imshow(img)
-            ax.axis('off')
+            ax.axis("off")
 
             # Attach metadata from spec
-            spec = bundle.get('spec', {})
+            spec = bundle.get("spec", {})
             if spec:
                 # Handle both layered and legacy spec formats
-                axes_list = spec.get('axes', [])
+                axes_list = spec.get("axes", [])
                 if axes_list and isinstance(axes_list, list):
                     for key, val in axes_list[0].items():
-                        setattr(ax, f'_scitex_{key}', val)
+                        setattr(ax, f"_scitex_{key}", val)
                 # Theme from style (layered) or spec (legacy)
-                style = bundle.get('style', {})
-                theme = style.get('theme', {}) if style else spec.get('theme', {})
+                style = bundle.get("style", {})
+                theme = style.get("theme", {}) if style else spec.get("theme", {})
                 if theme:
-                    fig._scitex_theme = theme.get('mode')
+                    fig._scitex_theme = theme.get("mode")
 
             # Data from bundle (merged in load_layered_pltz_bundle)
-            data = bundle.get('data')
+            data = bundle.get("data")
             return fig, ax, data
         else:
             # No PNG, return spec and data
-            return bundle.get('spec'), None, bundle.get('data')
+            return bundle.get("spec"), None, bundle.get("data")
 
     elif bundle_type == BundleType.FIGZ:
         # Return figure dict for .figz bundles
@@ -245,10 +300,6 @@ def load(
         if verbose:
             print(f"[DEBUG] After Path conversion: {lpath}")
 
-    # Handle .canvas directories (special case - directory not file)
-    if lpath.endswith(".canvas"):
-        return _load_canvas(lpath, verbose=verbose, **kwargs)
-
     # Handle bundle formats (.pltz, .figz, .statsz and their .d variants)
     bundle_extensions = (".figz", ".pltz", ".statsz")
     for bext in bundle_extensions:

scitex/io/_load_modules/__init__.py

@@ -1,5 +1,4 @@
 #!/usr/bin/env python3
-# -*- coding: utf-8 -*-
 # Time-stamp: "2024-11-14 07:41:25 (ywatanabe)"
 # File: ./scitex_repo/src/scitex/io/_load_modules/__init__.py
 
@@ -14,13 +13,17 @@ current_dir = __os.path.dirname(__file__)
 for filename in __os.listdir(current_dir):
     if filename.endswith(".py") and not filename.startswith("__"):
         module_name = filename[:-3]  # Remove .py extension
-        module = __importlib.import_module(f".{module_name}", package=__name__)
+        try:
+            module = __importlib.import_module(f".{module_name}", package=__name__)
 
-        # Import only functions and classes from the module
-        for name, obj in __inspect.getmembers(module):
-            if __inspect.isfunction(obj) or __inspect.isclass(obj):
-                if not name.startswith("_"):
-                    globals()[name] = obj
+            # Import only functions and classes from the module
+            for name, obj in __inspect.getmembers(module):
+                if __inspect.isfunction(obj) or __inspect.isclass(obj):
+                    if not name.startswith("_"):
+                        globals()[name] = obj
+        except ImportError:
+            # Skip modules with missing optional dependencies
+            pass
 
 # Clean up temporary variables
 del (

scitex/io/_load_modules/_pandas.py

@@ -1,5 +1,4 @@
 #!/usr/bin/env python3
-# -*- coding: utf-8 -*-
 # Time-stamp: "2024-11-14 07:41:31 (ywatanabe)"
 # File: ./scitex_repo/src/scitex/io/_load_modules/_pandas.py
 
@@ -35,6 +34,8 @@ def _load_csv(lpath, **kwargs):
 
 def _load_tsv(lpath, **kwargs):
     """Load TSV files."""
+    import pandas as pd
+
     if not lpath.endswith(".tsv"):
         raise ValueError("File must have .tsv extension")
     return pd.read_csv(lpath, sep="\t", **kwargs)
@@ -42,6 +43,8 @@ def _load_tsv(lpath, **kwargs):
 
 def _load_excel(lpath, **kwargs):
     """Load Excel files."""
+    import pandas as pd
+
     if not lpath.endswith((".xls", ".xlsx", ".xlsm", ".xlsb")):
         raise ValueError("File must have Excel extension")
     return pd.read_excel(lpath, **kwargs)
@@ -49,6 +52,8 @@ def _load_excel(lpath, **kwargs):
 
 def _load_parquet(lpath, **kwargs):
     """Load Parquet files."""
+    import pandas as pd
+
     if not lpath.endswith(".parquet"):
         raise ValueError("File must have .parquet extension")
     return pd.read_parquet(lpath, **kwargs)