duckrun 0.2.12__py3-none-any.whl → 0.2.13.dev0__py3-none-any.whl

duckrun/auth.py

@@ -38,21 +38,16 @@ def get_token() -> Optional[str]:
     except Exception as e:
         print(f"⚠️ Fabric notebook authentication failed: {e}")
 
-    # Detect environment type for fallback authentication
+    # Try local/VS Code authentication (Azure CLI + browser)
+    print("🖥️ Trying local authentication (Azure CLI + browser fallback)...")
+    token = _get_local_token()
+    if token:
+        return token
+    
+    # If local auth failed, fall back to device code flow
+    print("🔐 Falling back to device code flow for remote/headless environment...")
     try:
-        # Check if we're in Google Colab first
-        try:
-            import google.colab
-            print("🚀 Google Colab detected - using device code flow")
-            return _get_device_code_token()
-        except ImportError:
-            pass
-
-        # For all other environments (including VS Code), try Azure CLI first
-        # This includes local development, VS Code notebooks, etc.
-        print("🖥️ Local/VS Code environment detected - trying Azure CLI first, then browser fallback")
-        return _get_local_token()
-
+        return _get_device_code_token()
     except Exception as e:
         print(f"❌ Authentication failed: {e}")
         print("💡 Try refreshing and running again, or check your Azure permissions")
@@ -82,6 +77,35 @@ def _get_device_code_token() -> Optional[str]:
         return None
 
 
+def _is_databricks() -> bool:
+    """Check if we're running in a Databricks environment"""
+    # Databricks sets specific environment variables
+    return (
+        os.environ.get("DATABRICKS_RUNTIME_VERSION") is not None or
+        os.environ.get("DB_HOME") is not None or
+        "databricks" in os.environ.get("SPARK_HOME", "").lower()
+    )
+
+
+def _get_databricks_token() -> Optional[str]:
+    """Get token using DefaultAzureCredential for Databricks environments"""
+    try:
+        from azure.identity import DefaultAzureCredential
+        
+        # DefaultAzureCredential will automatically use Databricks managed identity
+        credential = DefaultAzureCredential()
+        token_obj = credential.get_token("https://storage.azure.com/.default")
+        
+        os.environ["AZURE_STORAGE_TOKEN"] = token_obj.token
+        print("✅ Databricks authentication successful!")
+        return token_obj.token
+        
+    except Exception as e:
+        print(f"❌ Databricks authentication failed: {e}")
+        print("💡 Make sure your Databricks cluster has the required Azure permissions")
+        return None
+
+
 def _get_local_token() -> Optional[str]:
     """Get token using CLI first, then browser fallback for local environments"""
     # First try Azure CLI directly

{duckrun-0.2.12.dist-info → duckrun-0.2.13.dev0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: duckrun
-Version: 0.2.12
+Version: 0.2.13.dev0
 Summary: Lakehouse task runner powered by DuckDB for Microsoft Fabric
 Author: mim
 License: MIT

{duckrun-0.2.12.dist-info → duckrun-0.2.13.dev0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 duckrun/__init__.py,sha256=cTj6KQ6hKmgu1z7k9nhDcO5lct049luxjx1V0QnymCo,235
-duckrun/auth.py,sha256=qPaLQ7InlV9leA9r6E6VEeYavFFoBi0zSN8m_l1aoQs,9545
+duckrun/auth.py,sha256=WjRpbB60lB7z3MTDyBMb0FqoHEesiC3nRiW2moAsrrs,10490
 duckrun/core.py,sha256=3usRl9SetUytVFzCzbpiFXppTjHzwTqFlSEKnUSbcK8,52460
 duckrun/files.py,sha256=Fvdjg3DyHJzIVzKo8M_j-eGz4zU61lOB38Y_onbQJkI,10137
 duckrun/lakehouse.py,sha256=j--Z3zo8AOWt1GF9VzRosmmTAy6ey2D0LVubti58twU,14109
@@ -7,8 +7,8 @@ duckrun/runner.py,sha256=yrDxfy1RVkb8iK9GKGmIFZHzCvcO_0GVQlbng7Vw_iM,14171
 duckrun/semantic_model.py,sha256=obzlN2-dbEW3JmDop-vrZGGGLi9u3ThhTbgtDjou7uY,29509
 duckrun/stats.py,sha256=oKIjZ7u5cFVT63FuOl5UqoDsOG3098woSCn-uI6i_sQ,11084
 duckrun/writer.py,sha256=svUuPCYOhrz299NgnpTKhARKjfej0PxnoND2iPDSypk,8098
-duckrun-0.2.12.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
-duckrun-0.2.12.dist-info/METADATA,sha256=MPsLnsgyPshKTIXKiO_MiAXbqPVxQ7dBVwzDggm56bQ,20766
-duckrun-0.2.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-duckrun-0.2.12.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
-duckrun-0.2.12.dist-info/RECORD,,
+duckrun-0.2.13.dev0.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
+duckrun-0.2.13.dev0.dist-info/METADATA,sha256=dgJzGSr1W2XaHsPbHbE2Vc9T03DKofX4DAC3lNRAh3I,20771
+duckrun-0.2.13.dev0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+duckrun-0.2.13.dev0.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
+duckrun-0.2.13.dev0.dist-info/RECORD,,