chore: fix lint

Author: HinsonSIDAN

examples/websocket_example.py

@@ -4,9 +4,10 @@
 """
 
 import asyncio
+from datetime import datetime
 import logging
 import os
-from datetime import datetime
+
 from dotenv import load_dotenv
 
 from deltadefi import ApiClient
@@ -15,19 +16,22 @@
 load_dotenv()
 
 # Set up logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+)
 api_key = os.getenv("DELTADEFI_API_KEY")
 
 # ============================================================================
 # DATA PARSING HANDLERS FOR EACH WEBSOCKET ENDPOINT
 # ============================================================================
 
+
 async def handle_trade_message(data):
     """
     Handle recent trades WebSocket messages.
-    
+
     Data format: Array of trade objects
-    Example: [{"timestamp": "2025-08-21T03:43:00.204624Z", "symbol": "ADAUSDM", 
+    Example: [{"timestamp": "2025-08-21T03:43:00.204624Z", "symbol": "ADAUSDM",
               "side": "sell", "price": 0.7803, "amount": 4.6}, ...]
     """
     print("\n🔄 TRADE STREAM DATA:")
@@ -37,52 +41,54 @@ async def handle_trade_message(data):
         side = trade.get("side", "unknown")
         price = trade.get("price", 0)
         amount = trade.get("amount", 0)
-        
+
         # Convert timestamp to readable format
         try:
-            dt = datetime.fromisoformat(timestamp.replace('Z', '+00:00'))
+            dt = datetime.fromisoformat(timestamp.replace("Z", "+00:00"))
             readable_time = dt.strftime("%H:%M:%S")
-        except:
+        except Exception:
             readable_time = timestamp
-            
+
         side_emoji = "🟢" if side.lower() == "buy" else "🔴"
-        print(f"  {i}. {side_emoji} {symbol}: {side.upper()} {amount} @ ${price:.4f} at {readable_time}")
+        print(
+            f"  {i}. {side_emoji} {symbol}: {side.upper()} {amount} @ ${price:.4f} at {readable_time}"
+        )
 
 
 async def handle_depth_message(data):
     """
     Handle market depth WebSocket messages.
-    
-    Data format: {"timestamp": 1755747950587, "bids": [{"price": 0.195, "quantity": 30}], 
+
+    Data format: {"timestamp": 1755747950587, "bids": [{"price": 0.195, "quantity": 30}],
                   "asks": [{"price": 0.3495, "quantity": 50.65}]}
     """
     print("\n📊 MARKET DEPTH DATA:")
     timestamp = data.get("timestamp", 0)
     bids = data.get("bids", [])
     asks = data.get("asks", [])
-    
+
     # Convert timestamp
     try:
         dt = datetime.fromtimestamp(timestamp / 1000)  # Assuming milliseconds
         readable_time = dt.strftime("%H:%M:%S")
-    except:
+    except Exception:
         readable_time = str(timestamp)
-    
+
     print(f"  📅 Time: {readable_time}")
-    
+
     # Show top 5 bids and asks
     print("  🟢 BIDS (Buy Orders):")
     for i, bid in enumerate(bids[:5], 1):
         price = bid.get("price", 0)
         quantity = bid.get("quantity", 0)
-        print(f"    {i}. ${price:.4f} × {quantity}")
-    
-    print("  🔴 ASKS (Sell Orders):")  
+        print(f"    {i}. ${price:.4f} x {quantity}")
+
+    print("  🔴 ASKS (Sell Orders):")
     for i, ask in enumerate(asks[:5], 1):
         price = ask.get("price", 0)
         quantity = ask.get("quantity", 0)
-        print(f"    {i}. ${price:.4f} × {quantity}")
-    
+        print(f"    {i}. ${price:.4f} x {quantity}")
+
     if bids and asks:
         spread = asks[0]["price"] - bids[0]["price"]
         spread_pct = (spread / bids[0]["price"]) * 100 if bids[0]["price"] > 0 else 0
@@ -92,31 +98,31 @@ async def handle_depth_message(data):
 async def handle_price_message(data):
     """
     Handle market price WebSocket messages.
-    
+
     Data format: {"type": "Market", "sub_type": "market_price", "price": 0.75}
     """
     print("\n💰 PRICE UPDATE:")
     msg_type = data.get("type", "Unknown")
-    sub_type = data.get("sub_type", "unknown") 
+    sub_type = data.get("sub_type", "unknown")
     price = data.get("price", 0)
-    
+
     print(f"  📊 Type: {msg_type}/{sub_type}")
     print(f"  💵 Current Price: ${price:.4f}")
 
 
 async def handle_account_message(data):
     """
     Handle account streams WebSocket messages.
-    
+
     Data formats:
     - Balance: {"type": "Account", "sub_type": "balance", "balance": [...]}
     - Orders: {"type": "Account", "sub_type": "open_orders", "data": [...]}
     """
     msg_type = data.get("type", "Unknown")
     sub_type = data.get("sub_type", "unknown")
-    
+
     print(f"\n👤 ACCOUNT UPDATE ({msg_type}/{sub_type}):")
-    
+
     if sub_type == "balance":
         balances = data.get("balance", [])
         print("  💰 Account Balances:")
@@ -125,12 +131,12 @@ async def handle_account_message(data):
             free = balance.get("free", 0)
             locked = balance.get("locked", 0)
             total = free + locked
-            
+
             print(f"    {asset.upper()}: ")
             print(f"      Free: {free:.6f}")
-            print(f"      Locked: {locked:.6f}")  
+            print(f"      Locked: {locked:.6f}")
             print(f"      Total: {total:.6f}")
-    
+
     elif sub_type == "open_orders":
         orders_data = data.get("data", [])
         print("  📋 Open Orders:")
@@ -141,23 +147,25 @@ async def handle_account_message(data):
                 order_count += 1
                 order_id = order.get("order_id", "unknown")
                 status = order.get("status", "unknown")
-                symbol = order.get("symbol", "unknown") 
+                symbol = order.get("symbol", "unknown")
                 side = order.get("side", "unknown")
-                
+
                 side_emoji = "🟢" if side.lower() == "buy" else "🔴"
-                print(f"    {order_count}. {side_emoji} {order_id[:8]}... - {symbol} {side.upper()} ({status})")
-        
+                print(
+                    f"    {order_count}. {side_emoji} {order_id[:8]}... - {symbol} {side.upper()} ({status})"
+                )
+
         if order_count == 0:
             print("    No open orders")
-    
+
     elif sub_type == "trading_history":
         print("  📈 Trading History Update")
         print(f"    Data: {data}")
-        
+
     elif sub_type == "orders_history":
-        print("  📜 Orders History Update") 
+        print("  📜 Orders History Update")
         print(f"    Data: {data}")
-        
+
     else:
         print(f"  ❓ Unknown sub_type: {sub_type}")
         print(f"    Raw data: {data}")
@@ -167,11 +175,12 @@ async def handle_account_message(data):
 # INDIVIDUAL ENDPOINT EXAMPLES
 # ============================================================================
 
+
 async def example_trades_stream(symbol="ADAUSDM", duration=30):
     """Example: Subscribe to recent trades stream."""
     print(f"\n🚀 Starting TRADES stream for {symbol} (running for {duration}s)")
     print("=" * 60)
-    
+
     client = ApiClient(api_key=api_key)
     ws_client = client.websocket
     ws_client.register_handler("trade", handle_trade_message)
@@ -190,7 +199,7 @@ async def example_depth_stream(symbol="ADAUSDM", duration=30):
     """Example: Subscribe to market depth stream."""
     print(f"\n🚀 Starting DEPTH stream for {symbol} (running for {duration}s)")
     print("=" * 60)
-    
+
     client = ApiClient(api_key=api_key)
     ws_client = client.websocket
     ws_client.register_handler("depth", handle_depth_message)
@@ -209,7 +218,7 @@ async def example_price_stream(symbol="ADAUSDM", duration=30):
     """Example: Subscribe to market price stream."""
     print(f"\n🚀 Starting PRICE stream for {symbol} (running for {duration}s)")
     print("=" * 60)
-    
+
     client = ApiClient(api_key=api_key)
     ws_client = client.websocket
     ws_client.register_handler("price", handle_price_message)
@@ -228,7 +237,7 @@ async def example_account_stream(duration=30):
     """Example: Subscribe to account streams."""
     print(f"\n🚀 Starting ACCOUNT stream (running for {duration}s)")
     print("=" * 60)
-    
+
     client = ApiClient(api_key=api_key)
     ws_client = client.websocket
     ws_client.register_handler("account", handle_account_message)
@@ -247,26 +256,27 @@ async def example_account_stream(duration=30):
 # MAIN FUNCTION WITH ENDPOINT SELECTION
 # ============================================================================
 
+
 async def main():
     """Main function with endpoint selection menu."""
     if not api_key:
         print("❌ Error: DELTADEFI_API_KEY not found in environment variables")
         print("Please set your API key in the .env file")
         return
-    
+
     print("🔗 DeltaDeFi WebSocket Examples")
     print("=" * 50)
     print("Available endpoints:")
     print("1. Recent Trades Stream")
-    print("2. Market Depth Stream") 
+    print("2. Market Depth Stream")
     print("3. Market Price Stream")
     print("4. Account Streams")
     print("5. Run All Endpoints (Sequential)")
     print("0. Quick Trades Test (10 seconds)")
-    
+
     try:
         choice = input("\nSelect endpoint (0-5): ").strip()
-        
+
         if choice == "1":
             await example_trades_stream()
         elif choice == "2":
@@ -288,7 +298,7 @@ async def main():
             await example_trades_stream(duration=10)
         else:
             print("❌ Invalid choice")
-            
+
     except KeyboardInterrupt:
         print("\n👋 Exiting...")
     except Exception as e:

src/deltadefi/api.py

@@ -73,10 +73,10 @@ def _handle_exception(self, response):
         if 400 <= status_code < 500:
             try:
                 err = json.loads(response.text)
-            except json.JSONDecodeError:
+            except json.JSONDecodeError as e:
                 raise ClientError(
                     status_code, None, response.text, response.headers, None
-                )
+                ) from e
             error_data = None
             if "data" in err:
                 error_data = err["data"]

src/deltadefi/api_resources/auth.py

@@ -1,16 +1,15 @@
 from dataclasses import dataclass
-from typing import Optional
 
 
 @dataclass
 class AuthHeaders:
     jwt: str
-    apiKey: str
+    api_key: str
 
 
 class ApiHeaders:
     __annotations__ = {
         "Content-Type": str,
-        "Authorization": Optional[str],
-        "X-API-KEY": Optional[str],
+        "Authorization": str | None,
+        "X-API-KEY": str | None,
     }

src/deltadefi/clients/client.py

@@ -16,9 +16,9 @@ class ApiClient:
     def __init__(
         self,
         network: str = "preprod",
-        api_key: str = None,
-        base_url: str = None,
-        master_wallet: Wallet = None,
+        api_key: str | None = None,
+        base_url: str | None = None,
+        master_wallet: Wallet | None = None,
     ):
         """
         Initialize the ApiClient.
@@ -44,13 +44,15 @@ def __init__(
         self.accounts = Accounts(base_url=self.base_url, api_key=api_key)
         self.orders = Order(base_url=self.base_url, api_key=api_key)
         self.markets = Market(base_url=self.base_url, api_key=api_key)
-        
+
         # Initialize WebSocket client with correct stream URL
         if network == "mainnet":
-            ws_base_url = "wss://stream.deltadefi.io"  # TODO: Update when mainnet is available
+            ws_base_url = (
+                "wss://stream.deltadefi.io"  # TODO: Update when mainnet is available
+            )
         else:
             ws_base_url = "wss://stream-staging.deltadefi.io"
-            
+
         self.websocket = WebSocketClient(base_url=ws_base_url, api_key=api_key)
 
     def load_operation_key(self, password: str):

src/deltadefi/clients/markets.py

@@ -1,4 +1,3 @@
-
 from typing import Literal
 
 from deltadefi.api import API