figpack 0.2.17__py3-none-any.whl → 0.2.18__py3-none-any.whl

figpack/extensions.py

@@ -0,0 +1,356 @@
+"""
+Extension management functionality for figpack
+"""
+
+import json
+import re
+import subprocess
+import sys
+from typing import Dict, List, Optional, Tuple
+from urllib.parse import urljoin
+
+import requests
+
+
+# Default wheel repository URL
+DEFAULT_WHEEL_REPO_URL = "https://flatironinstitute.github.io/figpack/wheels/"
+
+
+def parse_wheel_filename(filename: str) -> Optional[Dict[str, str]]:
+    """
+    Parse a wheel filename to extract package information.
+
+    Wheel filename format: {distribution}-{version}(-{build tag})?-{python tag}-{abi tag}-{platform tag}.whl
+    """
+    if not filename.endswith(".whl"):
+        return None
+
+    # Remove .whl extension
+    basename = filename[:-4]
+
+    # Split on hyphens, but be careful about package names with hyphens
+    parts = basename.split("-")
+
+    if len(parts) < 5:
+        return None
+
+    # The last 3 parts are always python_tag, abi_tag, platform_tag
+    python_tag = parts[-3]
+    abi_tag = parts[-2]
+    platform_tag = parts[-1]
+
+    # Everything before the last 3 parts, split between name and version
+    name_version_parts = parts[:-3]
+
+    # Find where version starts (first part that looks like a version number)
+    version_start_idx = 1  # Default to second part
+    for i, part in enumerate(name_version_parts[1:], 1):
+        if re.match(r"^\d+", part):  # Starts with a digit
+            version_start_idx = i
+            break
+
+    name = "-".join(name_version_parts[:version_start_idx])
+    version = "-".join(name_version_parts[version_start_idx:])
+
+    return {
+        "name": name,
+        "version": version,
+        "python_tag": python_tag,
+        "abi_tag": abi_tag,
+        "platform_tag": platform_tag,
+    }
+
+
+class ExtensionManager:
+    """Manages figpack extension packages"""
+
+    def __init__(self, wheel_repo_url: str = DEFAULT_WHEEL_REPO_URL):
+        self.wheel_repo_url = wheel_repo_url.rstrip("/") + "/"
+        self._available_cache = None
+
+    def get_installed_packages(self) -> Dict[str, str]:
+        """Get currently installed figpack extension packages and their versions"""
+        try:
+            result = subprocess.run(
+                [sys.executable, "-m", "pip", "list", "--format=json"],
+                capture_output=True,
+                text=True,
+                check=True,
+            )
+            installed_packages = json.loads(result.stdout)
+
+            # Get available extensions to filter against
+            available_extensions = self.get_available_packages()
+
+            # Filter for figpack extensions
+            figpack_extensions = {}
+            for pkg in installed_packages:
+                pkg_name = pkg["name"]
+                if pkg_name in available_extensions:
+                    figpack_extensions[pkg_name] = pkg["version"]
+
+            return figpack_extensions
+        except (subprocess.CalledProcessError, json.JSONDecodeError) as e:
+            print(f"Warning: Could not get installed packages: {e}")
+            return {}
+
+    def get_available_packages(self) -> Dict[str, Dict[str, str]]:
+        """
+        Get available packages from the wheel repository.
+
+        Returns:
+            Dict mapping package names to their info (version, description, etc.)
+        """
+        if self._available_cache is not None:
+            return self._available_cache
+
+        try:
+            # Try the simple index first as it's easier to parse
+            packages = self._get_packages_from_simple_index()
+
+            # If that fails, try the main index.html
+            if not packages:
+                packages = self._get_packages_from_main_index()
+
+            self._available_cache = packages
+            return packages
+
+        except requests.RequestException as e:
+            print(f"Warning: Could not fetch available packages: {e}")
+            return {}
+
+    def _get_packages_from_simple_index(self) -> Dict[str, Dict[str, str]]:
+        """Get packages from simple.html using regex parsing"""
+        try:
+            simple_url = urljoin(self.wheel_repo_url, "simple.html")
+            response = requests.get(simple_url, timeout=10)
+            response.raise_for_status()
+
+            packages = {}
+
+            # Parse wheel filenames from simple index using regex
+            # Look for href attributes containing .whl files
+            wheel_pattern = re.compile(r'href="([^"]*\.whl)"')
+            matches = wheel_pattern.findall(response.text)
+
+            for wheel_filename in matches:
+                wheel_info = parse_wheel_filename(wheel_filename)
+                if wheel_info and wheel_info["name"].startswith("figpack_"):
+                    pkg_name = wheel_info["name"]
+
+                    if pkg_name not in packages:
+                        packages[pkg_name] = {
+                            "version": wheel_info["version"],
+                            "description": self._get_package_description(pkg_name),
+                        }
+                    else:
+                        current_version = packages[pkg_name]["version"]
+                        new_version = wheel_info["version"]
+                        if self._is_newer_version(new_version, current_version):
+                            packages[pkg_name]["version"] = new_version
+
+            return packages
+
+        except requests.RequestException:
+            return {}
+
+    def _get_packages_from_main_index(self) -> Dict[str, Dict[str, str]]:
+        """Get packages from main index.html using regex parsing"""
+        try:
+            index_url = urljoin(self.wheel_repo_url, "index.html")
+            response = requests.get(index_url, timeout=10)
+            response.raise_for_status()
+
+            packages = {}
+
+            # Parse wheel filenames from main index using regex
+            # Look for links to .whl files
+            wheel_pattern = re.compile(r'<a[^>]+href="([^"]*\.whl)"[^>]*>')
+            matches = wheel_pattern.findall(response.text)
+
+            for wheel_filename in matches:
+                wheel_info = parse_wheel_filename(wheel_filename)
+                if wheel_info and wheel_info["name"].startswith("figpack_"):
+                    pkg_name = wheel_info["name"]
+
+                    if pkg_name not in packages:
+                        packages[pkg_name] = {
+                            "version": wheel_info["version"],
+                            "description": self._get_package_description(pkg_name),
+                        }
+                    else:
+                        current_version = packages[pkg_name]["version"]
+                        new_version = wheel_info["version"]
+                        if self._is_newer_version(new_version, current_version):
+                            packages[pkg_name]["version"] = new_version
+
+            return packages
+
+        except requests.RequestException:
+            return {}
+
+    def _get_package_description(self, package_name: str) -> str:
+        """Get a description for a package based on its name"""
+        descriptions = {
+            "figpack_3d": "3D visualization extension using Three.js",
+            "figpack_force_graph": "Force-directed graph visualization extension",
+            "figpack_franklab": "Frank Lab specific neuroscience visualization tools",
+            "figpack_spike_sorting": "Spike sorting specific visualization tools",
+        }
+        return descriptions.get(package_name, "Figpack extension package")
+
+    def _is_newer_version(self, version1: str, version2: str) -> bool:
+        """Simple version comparison - returns True if version1 > version2"""
+        try:
+            v1_parts = [int(x) for x in version1.split(".")]
+            v2_parts = [int(x) for x in version2.split(".")]
+
+            # Pad shorter version with zeros
+            max_len = max(len(v1_parts), len(v2_parts))
+            v1_parts.extend([0] * (max_len - len(v1_parts)))
+            v2_parts.extend([0] * (max_len - len(v2_parts)))
+
+            return v1_parts > v2_parts
+        except ValueError:
+            # If we can't parse versions, just do string comparison
+            return version1 > version2
+
+    def list_extensions(self) -> None:
+        """List all available extensions with their status"""
+        print("Available figpack extensions:")
+        print()
+
+        installed = self.get_installed_packages()
+        available = self.get_available_packages()
+
+        if not available:
+            print("No extensions found in the wheel repository.")
+            return
+
+        for ext_name, ext_info in available.items():
+            if ext_name in installed:
+                status = f"✓ {ext_name} (installed: {installed[ext_name]}, latest: {ext_info['version']})"
+            else:
+                status = f"✗ {ext_name} (not installed, latest: {ext_info['version']})"
+
+            print(f"{status} - {ext_info['description']}")
+
+    def install_extensions(
+        self, extensions: List[str], upgrade: bool = False, install_all: bool = False
+    ) -> bool:
+        """Install or upgrade extensions"""
+        available = self.get_available_packages()
+
+        if install_all:
+            extensions = list(available.keys())
+
+        if not extensions:
+            print("No extensions specified")
+            return False
+
+        # Validate extension names
+        invalid_extensions = [ext for ext in extensions if ext not in available]
+        if invalid_extensions:
+            print(f"Error: Unknown extensions: {', '.join(invalid_extensions)}")
+            print(f"Available extensions: {', '.join(available.keys())}")
+            return False
+
+        success = True
+        for extension in extensions:
+            if not self._install_single_extension(extension, upgrade):
+                success = False
+
+        return success
+
+    def _install_single_extension(self, extension: str, upgrade: bool = False) -> bool:
+        """Install a single extension"""
+        cmd = [
+            sys.executable,
+            "-m",
+            "pip",
+            "install",
+            "--find-links",
+            self.wheel_repo_url,
+            extension,
+        ]
+
+        if upgrade:
+            cmd.append("--upgrade")
+
+        try:
+            print(f"Installing {extension}...")
+            result = subprocess.run(cmd, capture_output=True, text=True, check=True)
+
+            # Check if it was actually installed or already satisfied
+            if "successfully installed" in result.stdout.lower():
+                print(f"✓ Successfully installed {extension}")
+            elif "already satisfied" in result.stdout.lower():
+                print(f"✓ {extension} is already installed")
+            else:
+                print(f"✓ {extension} installation completed")
+
+            return True
+
+        except subprocess.CalledProcessError as e:
+            print(f"✗ Failed to install {extension}")
+            if e.stderr:
+                print(f"Error: {e.stderr.strip()}")
+            return False
+
+    def uninstall_extensions(self, extensions: List[str]) -> bool:
+        """Uninstall extensions"""
+        if not extensions:
+            print("No extensions specified")
+            return False
+
+        available = self.get_available_packages()
+
+        # Validate extension names
+        invalid_extensions = [ext for ext in extensions if ext not in available]
+        if invalid_extensions:
+            print(f"Error: Unknown extensions: {', '.join(invalid_extensions)}")
+            return False
+
+        # Check which extensions are actually installed
+        installed = self.get_installed_packages()
+        not_installed = [ext for ext in extensions if ext not in installed]
+
+        if not_installed:
+            print(
+                f"Warning: These extensions are not installed: {', '.join(not_installed)}"
+            )
+
+        to_uninstall = [ext for ext in extensions if ext in installed]
+        if not to_uninstall:
+            print("No installed extensions to uninstall")
+            return True
+
+        # Confirm uninstallation
+        print(f"This will uninstall: {', '.join(to_uninstall)}")
+        response = input("Continue? [y/N]: ").strip().lower()
+        if response not in ["y", "yes"]:
+            print("Uninstallation cancelled")
+            return False
+
+        success = True
+        for extension in to_uninstall:
+            if not self._uninstall_single_extension(extension):
+                success = False
+
+        return success
+
+    def _uninstall_single_extension(self, extension: str) -> bool:
+        """Uninstall a single extension"""
+        cmd = [sys.executable, "-m", "pip", "uninstall", "-y", extension]
+
+        try:
+            print(f"Uninstalling {extension}...")
+            subprocess.run(cmd, capture_output=True, text=True, check=True)
+            print(f"✓ Successfully uninstalled {extension}")
+            return True
+
+        except subprocess.CalledProcessError as e:
+            print(f"✗ Failed to uninstall {extension}")
+            if e.stderr:
+                print(f"Error: {e.stderr.strip()}")
+            return False

figpack/figpack-figure-dist/assets/{index-DHWczh-Q.css → index-BJUFDPIM.css}

@@ -1 +1 @@
-.unitEntryBase{padding-right:13px;padding-top:10px}.unselectedUnitEntry{border:1px transparent solid;font-weight:400;color:#000;background-color:#fff}.selectedUnitEntry{border:1px solid blue;font-weight:700;color:#000;background-color:#b5d1ff}.unitIdsColumnLabelStyle{min-width:200px;font-weight:700;padding:7px 5px}.unitLabelsStyle{padding-right:3px;color:#333}.plotUnitLabel{text-align:left;margin-left:20px;margin-top:5px;font-weight:700}.plotWrapperButtonParent{display:inline-block}.plotWrapperStyleButton{padding-top:50px}.plotUnselectedStyle{border:3px solid rgb(202,224,231)}.plotSelectedStyle{border:3px solid #9999ff;background-color:#fff}.plotUnselectableStyle{border:3px solid transparent}.plotCurrentStyle{border:3px solid blue;background-color:#fff}.plotLabelStyle{font-weight:700;text-align:center;overflow:hidden;white-space:nowrap}:root{font-family:system-ui,Avenir,Helvetica,Arial,sans-serif;line-height:1.5;font-weight:400;font-synthesis:none;text-rendering:optimizeLegibility;-webkit-font-smoothing:antialiased;-moz-osx-font-smoothing:grayscale}a{font-weight:500;color:#646cff;text-decoration:inherit}a:hover{color:#535bf2}body{margin:0;overflow:hidden}h1{font-size:3.2em;line-height:1.1}button{border-radius:8px;border:1px solid transparent;padding:.6em 1.2em;font-size:1em;font-weight:500;font-family:inherit;background-color:#1a1a1a;cursor:pointer;transition:border-color .25s}button:hover{border-color:#646cff}button:focus,button:focus-visible{outline:4px auto -webkit-focus-ring-color}.status-bar{position:fixed;bottom:0;left:0;right:0;height:30px;background-color:#2a2a2a;color:#fff;display:flex;align-items:center;padding:0 12px;font-size:12px;border-top:1px solid #444;z-index:1000}.status-bar.warning{background-color:#ff9800;color:#000}.status-bar.error{background-color:#f44336;color:#fff}.status-bar.expired{background-color:#d32f2f;color:#fff}.manage-button{background-color:#444;color:#fff;border:none;padding:4px 8px;border-radius:4px;cursor:pointer;font-size:11px;height:22px;margin-left:12px;transition:background-color .2s}.manage-button:hover{background-color:#666}.about-button{background-color:#444;color:#fff;border:none;padding:4px 8px;border-radius:4px;cursor:pointer;font-size:11px;height:22px;margin-left:12px;transition:background-color .2s}.about-button:hover{background-color:#666}.about-dialog-overlay{position:fixed;inset:0;background-color:#00000080;display:flex;justify-content:center;align-items:center;z-index:2000}.about-dialog{background-color:#2a2a2a;color:#fff;border-radius:8px;max-width:600px;max-height:80vh;width:90%;box-shadow:0 4px 20px #0000004d;overflow:hidden}.about-dialog-header{display:flex;justify-content:space-between;align-items:center;padding:16px 20px;border-bottom:1px solid #444}.about-dialog-header h2{margin:0;font-size:18px;font-weight:600}.about-dialog-close{background:none;border:none;color:#fff;font-size:24px;cursor:pointer;padding:0;width:30px;height:30px;display:flex;align-items:center;justify-content:center;border-radius:4px;transition:background-color .2s}.about-dialog-close:hover{background-color:#444}.about-dialog-content{padding:20px;overflow-y:auto;max-height:calc(80vh - 80px)}.about-dialog-description{line-height:1.6}.about-dialog-description h1,.about-dialog-description h2,.about-dialog-description h3{margin-top:0;margin-bottom:12px}.about-dialog-description p{margin-bottom:12px}.about-dialog-description code{background-color:#444;padding:2px 4px;border-radius:3px;font-family:Courier New,monospace;font-size:.9em}.about-dialog-description pre{background-color:#444;padding:12px;border-radius:4px;overflow-x:auto;margin-bottom:12px}.about-dialog-description ul,.about-dialog-description ol{margin-bottom:12px;padding-left:20px}.about-dialog-description blockquote{border-left:4px solid #666;padding-left:16px;margin:12px 0;font-style:italic}@media (prefers-color-scheme: light){:root{color:#213547;background-color:#fff}a:hover{color:#747bff}button{background-color:#f9f9f9}.status-bar{background-color:#f5f5f5;color:#333;border-top:1px solid #ddd}.status-bar.warning{background-color:#fff3cd;color:#856404;border-top:1px solid #ffeaa7}.status-bar.error,.status-bar.expired{background-color:#f8d7da;color:#721c24;border-top:1px solid #f5c6cb}.manage-button{background-color:#e0e0e0;color:#333;border:1px solid #ccc}.manage-button:hover{background-color:#d0d0d0}.about-button{background-color:#e0e0e0;color:#333;border:1px solid #ccc}.about-button:hover{background-color:#d0d0d0}.about-dialog{background-color:#fff;color:#333;box-shadow:0 4px 20px #00000026}.about-dialog-header{border-bottom:1px solid #ddd}.about-dialog-close{color:#333}.about-dialog-close:hover{background-color:#f0f0f0}.about-dialog-description code,.about-dialog-description pre{background-color:#f5f5f5;color:#333}.about-dialog-description blockquote{border-left:4px solid #ccc}}
+.unitEntryBase{padding-right:13px;padding-top:10px}.unselectedUnitEntry{border:1px transparent solid;font-weight:400;color:#000;background-color:#fff}.selectedUnitEntry{border:1px solid blue;font-weight:700;color:#000;background-color:#b5d1ff}.unitIdsColumnLabelStyle{min-width:200px;font-weight:700;padding:7px 5px}.unitLabelsStyle{padding-right:3px;color:#333}.plotUnitLabel{text-align:left;margin-left:20px;margin-top:5px;font-weight:700}.plotWrapperButtonParent{display:inline-block}.plotWrapperStyleButton{padding-top:50px}.plotUnselectedStyle{border:3px solid rgb(202,224,231)}.plotSelectedStyle{border:3px solid #9999ff;background-color:#fff}.plotUnselectableStyle{border:3px solid transparent}.plotCurrentStyle{border:3px solid blue;background-color:#fff}.plotLabelStyle{font-weight:700;text-align:center;overflow:hidden;white-space:nowrap}:root{font-family:system-ui,Avenir,Helvetica,Arial,sans-serif;line-height:1.5;font-weight:400;font-synthesis:none;text-rendering:optimizeLegibility;-webkit-font-smoothing:antialiased;-moz-osx-font-smoothing:grayscale}a{font-weight:500;color:#646cff;text-decoration:inherit}a:hover{color:#535bf2}body{margin:0;overflow:hidden}h1{font-size:3.2em;line-height:1.1}button{border-radius:8px;border:1px solid transparent;padding:.6em 1.2em;font-size:1em;font-weight:500;font-family:inherit;background-color:#1a1a1a;cursor:pointer;transition:border-color .25s}button:hover{border-color:#646cff}button:focus,button:focus-visible{outline:4px auto -webkit-focus-ring-color}.status-bar{position:fixed;bottom:0;left:0;right:0;height:30px;background-color:#2a2a2a;color:#fff;display:flex;align-items:center;padding:0 12px;font-size:12px;border-top:1px solid #444;z-index:1000}.status-bar.warning{background-color:#ff9800;color:#000}.status-bar.error{background-color:#f44336;color:#fff}.status-bar.expired{background-color:#d32f2f;color:#fff}.manage-button{background-color:#444;color:#fff;border:none;padding:4px 8px;border-radius:4px;cursor:pointer;font-size:11px;height:22px;margin-left:12px;transition:background-color .2s}.manage-button:hover{background-color:#666}.about-button{background-color:#444;color:#fff;border:none;padding:4px 8px;border-radius:4px;cursor:pointer;font-size:11px;height:22px;margin-left:12px;transition:background-color .2s}.about-button:hover{background-color:#666}.about-dialog-overlay{position:fixed;inset:0;background-color:#00000080;display:flex;justify-content:center;align-items:center;z-index:2000}.about-dialog{background-color:#2a2a2a;color:#fff;border-radius:8px;max-width:600px;max-height:80vh;width:90%;box-shadow:0 4px 20px #0000004d;overflow:hidden}.about-dialog-header{display:flex;justify-content:space-between;align-items:center;padding:16px 20px;border-bottom:1px solid #444}.about-dialog-header h2{margin:0;font-size:18px;font-weight:600}.about-dialog-close{background:none;border:none;color:#fff;font-size:24px;cursor:pointer;padding:0;width:30px;height:30px;display:flex;align-items:center;justify-content:center;border-radius:4px;transition:background-color .2s}.about-dialog-close:hover{background-color:#444}.about-dialog-content{padding:20px;overflow-y:auto;max-height:calc(80vh - 80px)}.about-dialog-description{line-height:1.6}.about-dialog-description h1,.about-dialog-description h2,.about-dialog-description h3{margin-top:0;margin-bottom:12px}.about-dialog-description p{margin-bottom:12px}.about-dialog-description code{background-color:#444;padding:2px 4px;border-radius:3px;font-family:Courier New,monospace;font-size:.9em}.about-dialog-description pre{background-color:#444;padding:12px;border-radius:4px;overflow-x:auto;margin-bottom:12px}.about-dialog-description ul,.about-dialog-description ol{margin-bottom:12px;padding-left:20px}.about-dialog-description blockquote{border-left:4px solid #666;padding-left:16px;margin:12px 0;font-style:italic}.manage-edits-view{display:flex;align-items:center;margin-left:12px}.edits-notification{color:#ff9800;font-weight:500;margin-right:8px}.save-changes-button{background-color:#ff9800;color:#000;border:none;padding:4px 8px;border-radius:4px;cursor:pointer;font-size:11px;height:22px;transition:background-color .2s}.save-changes-button:hover{background-color:#f57c00}.dialog-overlay{position:fixed;inset:0;background-color:#00000080;display:flex;justify-content:center;align-items:center;z-index:2000}.dialog-content{background-color:#2a2a2a;color:#fff;border-radius:8px;max-width:500px;width:90%;box-shadow:0 4px 20px #0000004d;overflow:hidden}.dialog-header{display:flex;justify-content:space-between;align-items:center;padding:16px 20px;border-bottom:1px solid #444}.dialog-header h3{margin:0;font-size:18px;font-weight:600}.dialog-close{background:none;border:none;color:#fff;font-size:24px;cursor:pointer;padding:0;width:30px;height:30px;display:flex;align-items:center;justify-content:center;border-radius:4px;transition:background-color .2s}.dialog-close:hover{background-color:#444}.dialog-body{padding:20px}.dialog-body p{margin:0 0 12px;line-height:1.5}.dialog-footer{padding:16px 20px;border-top:1px solid #444;display:flex;justify-content:flex-end}.dialog-button{background-color:#444;color:#fff;border:none;padding:8px 16px;border-radius:4px;cursor:pointer;font-size:14px;transition:background-color .2s}.dialog-button:hover{background-color:#666}.dialog-fullscreen{position:fixed;inset:0;background-color:#2a2a2a;color:#fff;z-index:2000;display:flex;flex-direction:column}.dialog-content-fullscreen{display:flex;flex-direction:column;height:100%;overflow:hidden}.dialog-content-fullscreen .dialog-header{flex-shrink:0}.dialog-content-fullscreen .dialog-body{flex:1;overflow-y:auto;padding:20px}.dialog-content-fullscreen .dialog-footer{flex-shrink:0}@media (prefers-color-scheme: light){:root{color:#213547;background-color:#fff}a:hover{color:#747bff}button{background-color:#f9f9f9}.status-bar{background-color:#f5f5f5;color:#333;border-top:1px solid #ddd}.status-bar.warning{background-color:#fff3cd;color:#856404;border-top:1px solid #ffeaa7}.status-bar.error,.status-bar.expired{background-color:#f8d7da;color:#721c24;border-top:1px solid #f5c6cb}.manage-button{background-color:#e0e0e0;color:#333;border:1px solid #ccc}.manage-button:hover{background-color:#d0d0d0}.about-button{background-color:#e0e0e0;color:#333;border:1px solid #ccc}.about-button:hover{background-color:#d0d0d0}.about-dialog{background-color:#fff;color:#333;box-shadow:0 4px 20px #00000026}.about-dialog-header{border-bottom:1px solid #ddd}.about-dialog-close{color:#333}.about-dialog-close:hover{background-color:#f0f0f0}.about-dialog-description code,.about-dialog-description pre{background-color:#f5f5f5;color:#333}.about-dialog-description blockquote{border-left:4px solid #ccc}.edits-notification{color:#e65100}.save-changes-button{background-color:#ff9800;color:#fff;border:1px solid #f57c00}.save-changes-button:hover{background-color:#f57c00}.dialog-content{background-color:#fff;color:#333;box-shadow:0 4px 20px #00000026}.dialog-header{border-bottom:1px solid #ddd}.dialog-close{color:#333}.dialog-close:hover{background-color:#f0f0f0}.dialog-footer{border-top:1px solid #ddd}.dialog-button{background-color:#e0e0e0;color:#333;border:1px solid #ccc}.dialog-button:hover{background-color:#d0d0d0}.dialog-fullscreen{background-color:#fff;color:#333}}