jupyterlab-mlflow 0.4.1__py3-none-any.whl

jupyterlab_mlflow/__init__.py

@@ -0,0 +1,63 @@
+"""
+JupyterLab MLflow Extension
+"""
+
+import subprocess
+import sys
+import os
+
+from ._version import __version__
+
+# Auto-enable server extension on import (if not already enabled)
+def _auto_enable_server_extension():
+    """Attempt to auto-enable the server extension"""
+    try:
+        # Check if already enabled by trying to import the config
+        from jupyter_server.services.config.manager import ConfigManager
+        cm = ConfigManager()
+        
+        # Try to enable the extension
+        result = subprocess.run(
+            [sys.executable, "-m", "jupyter", "server", "extension", "enable",
+             "jupyterlab_mlflow.serverextension", "--sys-prefix"],
+            capture_output=True,
+            text=True,
+            timeout=5
+        )
+        if result.returncode != 0:
+            # Try without --sys-prefix
+            subprocess.run(
+                [sys.executable, "-m", "jupyter", "server", "extension", "enable",
+                 "jupyterlab_mlflow.serverextension"],
+                capture_output=True,
+                text=True,
+                timeout=5
+            )
+    except Exception:
+        # Silently fail - don't break installation if this doesn't work
+        pass
+
+# Only auto-enable if we're being imported in a Jupyter context
+# (not during build/installation)
+if not os.environ.get('JUPYTERLAB_MLFLOW_SKIP_AUTO_ENABLE'):
+    try:
+        _auto_enable_server_extension()
+    except Exception:
+        pass
+
+def _jupyter_labextension_paths():
+    """Called by Jupyter Lab Server to detect if it is a valid labextension and
+    to install the widget
+
+    Returns
+    =======
+    src: Source directory name to copy files from. The JupyterLab builder outputs
+        generated files into this directory and Jupyter Lab copies from this
+        directory during widget installation
+    dest: Destination directory name to install to
+    """
+    return [{
+        'src': 'labextension',
+        'dest': 'jupyterlab-mlflow'
+    }]
+

jupyterlab_mlflow/_version.py

@@ -0,0 +1,6 @@
+"""
+Version information for jupyterlab-mlflow
+"""
+
+__version__ = "0.1.0"
+

jupyterlab_mlflow/post_install.py

@@ -0,0 +1,44 @@
+"""
+Post-install hook to auto-enable the server extension
+"""
+import subprocess
+import sys
+
+
+def enable_server_extension():
+    """Enable the server extension after installation"""
+    try:
+        # Try to enable the extension
+        result = subprocess.run(
+            [sys.executable, "-m", "jupyter", "server", "extension", "enable", 
+             "jupyterlab_mlflow.serverextension", "--sys-prefix"],
+            capture_output=True,
+            text=True,
+            timeout=10
+        )
+        if result.returncode == 0:
+            print("✅ Server extension enabled successfully")
+            return True
+        else:
+            # Try without --sys-prefix
+            result = subprocess.run(
+                [sys.executable, "-m", "jupyter", "server", "extension", "enable",
+                 "jupyterlab_mlflow.serverextension"],
+                capture_output=True,
+                text=True,
+                timeout=10
+            )
+            if result.returncode == 0:
+                print("✅ Server extension enabled successfully")
+                return True
+            else:
+                print(f"⚠️  Could not enable server extension: {result.stderr}")
+                return False
+    except Exception as e:
+        print(f"⚠️  Error enabling server extension: {e}")
+        return False
+
+
+if __name__ == "__main__":
+    enable_server_extension()
+

jupyterlab_mlflow/schema/plugin.json

@@ -0,0 +1,17 @@
+{
+  "jupyter.lab.setting-icon": "mlflow:icon",
+  "jupyter.lab.setting-icon-label": "MLflow",
+  "title": "MLflow",
+  "description": "MLflow extension settings",
+  "type": "object",
+  "properties": {
+    "mlflowTrackingUri": {
+      "type": "string",
+      "title": "MLflow Tracking URI",
+      "description": "URI of the MLflow tracking server (e.g., http://localhost:5000). Leave empty to use MLFLOW_TRACKING_URI environment variable.",
+      "default": ""
+    }
+  },
+  "additionalProperties": false
+}
+

jupyterlab_mlflow/serverextension/__init__.py

@@ -0,0 +1,69 @@
+"""
+JupyterLab MLflow Server Extension
+"""
+
+import logging
+from .handlers import setup_handlers
+
+logger = logging.getLogger(__name__)
+
+# Track if extension has been loaded to avoid duplicate registration
+_extension_loaded = False
+
+
+def _jupyter_server_extension_points():
+    """
+    Returns a list of dictionaries with metadata about
+    the server extension points.
+    """
+    return [{
+        "module": "jupyterlab_mlflow.serverextension"
+    }]
+
+
+def _load_jupyter_server_extension(server_app):
+    """Registers the API handler to receive HTTP requests from the frontend extension.
+
+    Parameters
+    ----------
+    server_app: jupyter_server.serverapp.ServerApp
+        Jupyter Server application instance
+    """
+    global _extension_loaded
+    
+    # Prevent duplicate registration
+    if _extension_loaded:
+        server_app.log.warning("jupyterlab-mlflow: Extension already loaded, skipping duplicate registration")
+        return
+    
+    try:
+        setup_handlers(server_app.web_app)
+        _extension_loaded = True
+        
+        # Log success
+        success_msg = "✅ Registered jupyterlab-mlflow server extension"
+        server_app.log.info(success_msg)
+        logger.info(success_msg)
+        
+        # Log startup verification details
+        base_url = server_app.web_app.settings.get("base_url", "/")
+        verification_msg = f"✅ jupyterlab-mlflow: Server extension loaded successfully with base_url: {base_url}"
+        server_app.log.info(verification_msg)
+        logger.info(verification_msg)
+        
+    except Exception as e:
+        error_msg = f"❌ Failed to register jupyterlab-mlflow server extension: {e}"
+        server_app.log.error(error_msg)
+        logger.error(error_msg, exc_info=True)
+        # Don't raise - allow JupyterLab to continue loading even if extension fails
+        # This prevents the entire server from failing due to extension issues
+
+
+# For Jupyter Server 2.x compatibility
+def _jupyter_server_extension_paths():
+    """
+    Returns a list of server extension paths for Jupyter Server 2.x.
+    """
+    return [{
+        "module": "jupyterlab_mlflow.serverextension"
+    }]