Fixed more flak8 formatting

main.py

@@ -138,7 +138,7 @@ def decorated(*args, **kwargs):
 
         except jwt.ExpiredSignatureError:
             raise AuthError(
-                {"code": "token_expired", "description": "Token is expired"}, 401
+                {"code": "token_expired", "description": "Token is expired"}, 401,
             )
         except jwt.JWTClaimsError:
             raise AuthError(

models.py

@@ -1,7 +1,6 @@
 # models.py
 
 import time
-import uuid
 from dataclasses import dataclass, field
 
 

order_book.py

@@ -2,7 +2,6 @@
 
 from typing import List
 import uuid
-from sqlalchemy.orm import Session
 
 from models import Order, Trade
 from db_models import DbOrder, DbTrade
@@ -51,7 +50,8 @@ def add_order(self, order: Order) -> List[Trade]:
 
     def _match_buy(self, buy_order: Order) -> List[Trade]:
         """
-        Match a BUY order against the best available SELL orders in ascending price order.
+        Match a BUY order against the best available SELL orders
+        in ascending price order.
 
         :param buy_order: The BUY Order object to be matched.
         :return: A list of executed Trade objects from this matching session.
@@ -106,7 +106,8 @@ def _match_buy(self, buy_order: Order) -> List[Trade]:
 
     def _match_sell(self, sell_order: Order) -> List[Trade]:
         """
-        Match a SELL order against the best available BUY orders in descending price order.
+        Match a SELL order against the best available
+        BUY orders in descending price order.
 
         :param sell_order: The SELL Order object to be matched.
         :return: A list of executed Trade objects from this matching session.
@@ -200,7 +201,7 @@ def _save_trade(self, trade: Trade) -> None:
                 db.commit()
             except Exception as e:
                 db.rollback()
-            logger.error(f"Error inserting trade {trade.trade_id} in DB: {e}")
+                logger.error(f"Error inserting trade {trade.trade_id} in DB: {e}")
 
     def get_order_book(self):
         """

tests/test_api.py

@@ -48,7 +48,12 @@ def test_create_order_with_auth(client, valid_token):
     headers = {"Authorization": f"Bearer {valid_token}"}
     response = client.post(
         "/orders",
-        json={"user_id": "Alice", "side": "BUY", "price": 100.0, "quantity": 10},
+        json={
+            "user_id": "Alice",
+            "side": "BUY",
+            "price": 100.0,
+            "quantity": 10,
+        },
         headers=headers,
     )
     assert response.status_code == 200
@@ -61,7 +66,12 @@ def test_create_order_without_auth(client):
     """Test creating an order without JWT"""
     response = client.post(
         "/orders",
-        json={"user_id": "Alice", "side": "BUY", "price": 100.0, "quantity": 10},
+        json={
+            "user_id": "Alice",
+            "side": "BUY",
+            "price": 100.0,
+            "quantity": 10,
+        },
     )
     assert response.status_code == 401
     data = response.get_json()
@@ -78,7 +88,12 @@ def test_rate_limiting(client, valid_token):
     for i in range(9):
         response = client.post(
             "/orders",
-            json={"user_id": "Bob", "side": "SELL", "price": 95.0, "quantity": 5},
+            json={
+                "user_id": "Bob",
+                "side": "SELL",
+                "price": 95.0,
+                "quantity": 5,
+            },
             headers=headers,
         )
         assert response.status_code == 200, f"Request {i + 1} failed unexpectedly"

tests/test_order_book.py

@@ -1,4 +1,3 @@
-import pytest
 from order_book import OrderBook
 from models import Order
 
@@ -11,7 +10,7 @@ def test_add_buy_order_no_match():
     """
     ob = OrderBook()
     buy_order = Order(
-        order_id="test_buy_1", user_id="UserA", side="BUY", price=100.0, quantity=10
+        order_id="test_buy_1", user_id="UserA", side="BUY", price=100.0, quantity=10,
     )
 
     trades = ob.add_order(buy_order)
@@ -31,7 +30,7 @@ def test_add_sell_order_no_match():
     """
     ob = OrderBook()
     sell_order = Order(
-        order_id="test_sell_1", user_id="UserB", side="SELL", price=105.0, quantity=5
+        order_id="test_sell_1", user_id="UserB", side="SELL", price=105.0, quantity=5,
     )
 
     trades = ob.add_order(sell_order)
@@ -52,13 +51,13 @@ def test_buy_and_sell_match_exact():
 
     # Add BUY
     buy_order = Order(
-        order_id="test_buy_2", user_id="UserA", side="BUY", price=100.0, quantity=5
+        order_id="test_buy_2", user_id="UserA", side="BUY", price=100.0, quantity=5,
     )
     ob.add_order(buy_order)
 
     # Add SELL that matches (SELL price <= BUY price)
     sell_order = Order(
-        order_id="test_sell_2", user_id="UserB", side="SELL", price=100.0, quantity=5
+        order_id="test_sell_2", user_id="UserB", side="SELL", price=100.0, quantity=5,
     )
     trades = ob.add_order(sell_order)
 
@@ -84,13 +83,13 @@ def test_partial_fill():
 
     # BUY 10 @ 100
     buy_order = Order(
-        order_id="buy_partial", user_id="UserA", side="BUY", price=100.0, quantity=10
+        order_id="buy_partial", user_id="UserA", side="BUY", price=100.0, quantity=10,
     )
     ob.add_order(buy_order)
 
     # SELL 6 @ 95
     sell_order = Order(
-        order_id="sell_partial", user_id="UserB", side="SELL", price=95.0, quantity=6
+        order_id="sell_partial", user_id="UserB", side="SELL", price=95.0, quantity=6,
     )
     trades = ob.add_order(sell_order)