added formatter and checked

Author: jprakash-db

.github/workflows/code-quality-checks.yml

@@ -169,7 +169,7 @@ jobs:
 
       -   name: Run tests
           run: poetry run python -m pytest tests/unit
-  check-linting:
+  code-format-check:
     runs-on: ubuntu-latest
     strategy:
       matrix:
@@ -216,10 +216,10 @@ jobs:
       - name: Install library
         run: poetry install --no-interaction
       #----------------------------------------------
-      #              black the code
+      #              Run Ruff format check
       #----------------------------------------------
-      - name: Black
-        run: poetry run black --check src
+      - name: Ruff format check
+        run: poetry run ruff format --check
 
   check-types:
     runs-on: ubuntu-latest

examples/custom_cred_provider.py

@@ -21,7 +21,6 @@
     http_path=os.getenv("DATABRICKS_HTTP_PATH"),
     credentials_provider=creds,
 ) as connection:
-
     for x in range(1, 5):
         cursor = connection.cursor()
         cursor.execute("SELECT 1+1")

examples/experimental/tests/test_sea_async_query.py

@@ -1,6 +1,7 @@
 """
 Test for SEA asynchronous query execution functionality.
 """
+
 import os
 import sys
 import logging

examples/experimental/tests/test_sea_metadata.py

@@ -1,6 +1,7 @@
 """
 Test for SEA metadata functionality.
 """
+
 import os
 import sys
 import logging

examples/experimental/tests/test_sea_session.py

@@ -1,6 +1,7 @@
 """
 Test for SEA session management functionality.
 """
+
 import os
 import sys
 import logging

examples/experimental/tests/test_sea_sync_query.py

@@ -1,6 +1,7 @@
 """
 Test for SEA synchronous query execution functionality.
 """
+
 import os
 import sys
 import logging

examples/insert_data.py

@@ -6,7 +6,6 @@
     http_path=os.getenv("DATABRICKS_HTTP_PATH"),
     access_token=os.getenv("DATABRICKS_TOKEN"),
 ) as connection:
-
     with connection.cursor() as cursor:
         cursor.execute("CREATE TABLE IF NOT EXISTS squares (x int, x_squared int)")
 

examples/interactive_oauth.py

@@ -17,7 +17,6 @@
     server_hostname=os.getenv("DATABRICKS_SERVER_HOSTNAME"),
     http_path=os.getenv("DATABRICKS_HTTP_PATH"),
 ) as connection:
-
     for x in range(1, 100):
         cursor = connection.cursor()
         cursor.execute("SELECT 1+1")

examples/persistent_oauth.py

@@ -51,7 +51,6 @@ def read(self, hostname: str) -> Optional[OAuthToken]:
     auth_type="databricks-oauth",
     experimental_oauth_persistence=DevOnlyFilePersistence("./sample.json"),
 ) as connection:
-
     for x in range(1, 100):
         cursor = connection.cursor()
         cursor.execute("SELECT 1+1")

examples/proxy_authentication.py

@@ -20,122 +20,128 @@
 
 # Configure logging to see proxy activity
 logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+    level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
 )
 
 # Uncomment for detailed debugging (shows HTTP requests/responses)
 # logging.getLogger("urllib3").setLevel(logging.DEBUG)
 # logging.getLogger("urllib3.connectionpool").setLevel(logging.DEBUG)
 
+
 def check_proxy_environment():
     """Check if proxy environment variables are configured."""
-    proxy_vars = ['HTTP_PROXY', 'HTTPS_PROXY', 'http_proxy', 'https_proxy']
-    configured_proxies = {var: os.environ.get(var) for var in proxy_vars if os.environ.get(var)}
-    
+    proxy_vars = ["HTTP_PROXY", "HTTPS_PROXY", "http_proxy", "https_proxy"]
+    configured_proxies = {
+        var: os.environ.get(var) for var in proxy_vars if os.environ.get(var)
+    }
+
     if configured_proxies:
         print("✓ Proxy environment variables found:")
         for var, value in configured_proxies.items():
             # Hide credentials in output for security
-            safe_value = value.split('@')[-1] if '@' in value else value
+            safe_value = value.split("@")[-1] if "@" in value else value
             print(f"  {var}: {safe_value}")
         return True
     else:
         print("⚠ No proxy environment variables found")
         print("  Set HTTP_PROXY and/or HTTPS_PROXY if using a proxy")
         return False
 
+
 def test_connection(connection_params, test_name):
     """Test a database connection with given parameters."""
     print(f"\n--- Testing {test_name} ---")
-    
+
     try:
         with sql.connect(**connection_params) as connection:
             print("✓ Successfully connected!")
-            
+
             with connection.cursor() as cursor:
                 # Test basic query
-                cursor.execute("SELECT current_user() as user, current_database() as database")
+                cursor.execute(
+                    "SELECT current_user() as user, current_database() as database"
+                )
                 result = cursor.fetchone()
                 print(f"✓ Connected as user: {result.user}")
                 print(f"✓ Default database: {result.database}")
-                
+
                 # Test a simple computation
                 cursor.execute("SELECT 1 + 1 as result")
                 result = cursor.fetchone()
                 print(f"✓ Query result: 1 + 1 = {result.result}")
-                
+
         return True
-        
+
     except Exception as e:
         print(f"✗ Connection failed: {e}")
         return False
 
+
 def main():
     print("Databricks SQL Connector - Proxy Authentication Examples")
     print("=" * 60)
-    
+
     # Check proxy configuration
     has_proxy = check_proxy_environment()
-    
+
     # Get Databricks connection parameters
-    server_hostname = os.environ.get('DATABRICKS_SERVER_HOSTNAME')
-    http_path = os.environ.get('DATABRICKS_HTTP_PATH')
-    access_token = os.environ.get('DATABRICKS_TOKEN')
-    
+    server_hostname = os.environ.get("DATABRICKS_SERVER_HOSTNAME")
+    http_path = os.environ.get("DATABRICKS_HTTP_PATH")
+    access_token = os.environ.get("DATABRICKS_TOKEN")
+
     if not all([server_hostname, http_path, access_token]):
         print("\n✗ Missing required environment variables:")
         print("  DATABRICKS_SERVER_HOSTNAME")
-        print("  DATABRICKS_HTTP_PATH") 
+        print("  DATABRICKS_HTTP_PATH")
         print("  DATABRICKS_TOKEN")
         return 1
-    
+
     print(f"\nConnecting to: {server_hostname}")
-    
+
     # Base connection parameters
     base_params = {
-        'server_hostname': server_hostname,
-        'http_path': http_path,
-        'access_token': access_token
+        "server_hostname": server_hostname,
+        "http_path": http_path,
+        "access_token": access_token,
     }
-    
+
     success_count = 0
     total_tests = 0
-    
+
     # Test 1: Default proxy behavior (no _proxy_auth_method specified)
     # This uses basic auth if credentials are in proxy URL, otherwise no auth
-    print("\n" + "="*60)
+    print("\n" + "=" * 60)
     print("Test 1: Default Proxy Behavior")
     print("Uses basic authentication if credentials are in proxy URL")
     total_tests += 1
     if test_connection(base_params, "Default Proxy Behavior"):
         success_count += 1
-    
+
     # Test 2: Explicit basic authentication
-    print("\n" + "="*60)
+    print("\n" + "=" * 60)
     print("Test 2: Explicit Basic Authentication")
     print("Explicitly requests basic authentication (same as default)")
     total_tests += 1
     basic_params = base_params.copy()
-    basic_params['_proxy_auth_method'] = 'basic'
+    basic_params["_proxy_auth_method"] = "basic"
     if test_connection(basic_params, "Basic Proxy Authentication"):
         success_count += 1
-    
+
     # Test 3: Kerberos/Negotiate authentication
-    print("\n" + "="*60)
+    print("\n" + "=" * 60)
     print("Test 3: Kerberos/Negotiate Authentication")
     print("Uses Kerberos tickets for proxy authentication")
     print("Note: Requires valid Kerberos tickets (run 'kinit' first)")
     total_tests += 1
     kerberos_params = base_params.copy()
-    kerberos_params['_proxy_auth_method'] = 'negotiate'
+    kerberos_params["_proxy_auth_method"] = "negotiate"
     if test_connection(kerberos_params, "Kerberos Proxy Authentication"):
         success_count += 1
-    
+
     # Summary
-    print(f"\n{'='*60}")
+    print(f"\n{'=' * 60}")
     print(f"Summary: {success_count}/{total_tests} tests passed")
-    
+
     if success_count == total_tests:
         print("✓ All proxy authentication methods working!")
         return 0
@@ -149,5 +155,6 @@ def main():
             print("Consider checking your proxy configuration")
         return 1
 
+
 if __name__ == "__main__":
     exit(main())