formatted code

Author: lordlabakdas

api/firebase_api.py

@@ -4,12 +4,14 @@
 from firebase_admin import auth as firebase_auth
 from google.oauth2 import id_token
 
-from api.validators.user_validation import (ChangePasswordPayload,
-                                            ChangePasswordResponse,
-                                            FirebaseLoginPayload,
-                                            FirebaseLoginResponse,
-                                            FirebaseRegistrationPayload,
-                                            FirebaseRegistrationResponse)
+from api.validators.user_validation import (
+    ChangePasswordPayload,
+    ChangePasswordResponse,
+    FirebaseLoginPayload,
+    FirebaseLoginResponse,
+    FirebaseRegistrationPayload,
+    FirebaseRegistrationResponse,
+)
 from models.service.user_service import User
 
 logger = logging.getLogger("groceror")

api/helpers/inventory_helper.py

@@ -14,7 +14,9 @@ class InventoryHelper(object):
     def __init__(self, user: User) -> None:
         self.user = user
 
-    def add_inventory(self, name: str, quantity: int, category: Enum, notes: str = None) -> dict:
+    def add_inventory(
+        self, name: str, quantity: int, category: Enum, notes: str = None
+    ) -> dict:
         try:
             inventory_obj = (
                 db_session.query(Inventory)
@@ -42,9 +44,7 @@ def add_inventory(self, name: str, quantity: int, category: Enum, notes: str = N
         finally:
             db_session.close()
 
-    def get_store_inventory(
-        self, items: List[str] = None, quantity_limit: int = None
-    ) -> List[Dict]:
+    def get_store_inventory(self, items: List[str] = None) -> List[Dict]:
         try:
             store_inventory = (
                 db_session.query(Inventory)
@@ -57,10 +57,6 @@ def get_store_inventory(
                     store_inventory = store_inventory.filter(
                         Inventory.name.in_(items)
                     ).all()
-                if quantity_limit:
-                    store_inventory = store_inventory.filter(
-                        Inventory.quantity > quantity_limit
-                    )
             else:
                 logger.critical(f"No store inventory found for {self.user}")
         except Exception as e:
@@ -109,10 +105,14 @@ def get_inventory_by_name(self, name: str) -> List[Dict]:
 
     def update_inventory(self, inventory: dict) -> None:
         try:
-            db_session.query(Inventory).filter(Inventory.id == inventory.id).update(inventory)
+            db_session.query(Inventory).filter(Inventory.id == inventory.id).update(
+                inventory
+            )
             db_session.commit()
         except Exception as e:
-            logger.exception(f"Error while updating inventory with exception details {e}")
+            logger.exception(
+                f"Error while updating inventory with exception details {e}"
+            )
             raise e
         finally:
             db_session.close()
@@ -122,5 +122,7 @@ def delete_inventory(self, inventory: dict) -> None:
             db_session.query(Inventory).filter(Inventory.id == inventory.id).delete()
             db_session.commit()
         except Exception as e:
-            logger.exception(f"Error while deleting inventory with exception details {e}")
-            raise e
+            logger.exception(
+                f"Error while deleting inventory with exception details {e}"
+            )
+            raise e

api/inventory_api.py

@@ -4,9 +4,11 @@
 from fastapi import APIRouter, Depends, HTTPException, status
 
 from api.helpers.inventory_helper import InventoryHelper
-from api.validators.inventory_validation import (AddInventoryPayload,
-                                                 AddInventoryResponse,
-                                                 StoreInventoryResponse)
+from api.validators.inventory_validation import (
+    AddInventoryPayload,
+    AddInventoryResponse,
+    StoreInventoryResponse,
+)
 from helpers.jwt import auth_required
 from models.entity.user_entity import User
 

api/user_api.py

@@ -6,11 +6,14 @@
 from loguru import logger
 
 from api.helpers import auth_helper
-from api.validators.user_validation import (ChangePasswordPayload,
-                                            ChangePasswordResponse,
-                                            LoginPayload, LoginResponse,
-                                            RegistrationPayload,
-                                            RegistrationResponse)
+from api.validators.user_validation import (
+    ChangePasswordPayload,
+    ChangePasswordResponse,
+    LoginPayload,
+    LoginResponse,
+    RegistrationPayload,
+    RegistrationResponse,
+)
 from config import JWTConfig
 from helpers.jwt import JWT
 
@@ -43,9 +46,7 @@ async def register(registration_payload: RegistrationPayload):
     logger.info(f"Registering user with payload: {registration_payload}")
     try:
         if not auth_helper.is_user_exists(email=registration_payload.email):
-            new_user = auth_helper.register(
-                **registration_payload.dict()
-            )
+            new_user = auth_helper.register(**registration_payload.dict())
         else:
             raise HTTPException(
                 status_code=status.HTTP_409_CONFLICT,
@@ -87,5 +88,3 @@ async def change_password(
     logger.info(f"Changing password for user with payload: {change_password_payload}")
     auth_helper.change_password(current_user, change_password_payload)
     return {"status": "success"}
-
-

engine/inventory/cart.py

@@ -1,5 +1,6 @@
 from models.entity.user_entity import User
 
+
 class Cart(object):
     def __init__(self, user: User):
         self.user = user
@@ -13,26 +14,25 @@ def remove_item(self, item: dict):
 
     def get_cart(self):
         return self.items
-    
+
     def clear_cart(self):
         self.items = []
-    
+
     def get_total(self):
-        return sum(item['price'] for item in self.items)
-    
+        return sum(item["price"] for item in self.items)
+
     def checkout(self):
         self.clear_cart()
         return self.get_cart_total()
-    
+
     def get_cart_items(self):
         return self.items
-    
+
     def get_cart_total(self):
         return self.get_total()
 
     def get_cart_count(self):
         return len(self.items)
 
     def get_cart_items_total(self):
-        return sum(item['price'] for item in self.items)
-    
+        return sum(item["price"] for item in self.items)

helpers/jwt.py

@@ -55,19 +55,16 @@ def decode_token(self, token: str) -> Union[Dict[str, str], None]:
         except jwt.ExpiredSignatureError:
             logger.exception("Token has expired")
             raise HTTPException(
-                status_code=status.HTTP_401_UNAUTHORIZED,
-                detail="Token has expired"
+                status_code=status.HTTP_401_UNAUTHORIZED, detail="Token has expired"
             )
         except jwt.InvalidTokenError:
             logger.exception("Invalid token")
             raise HTTPException(
-                status_code=status.HTTP_401_UNAUTHORIZED,
-                detail="Invalid token"
+                status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid token"
             )
         except Exception as e:
             logger.exception(f"Error while decoding token: {e}")
             raise HTTPException(
-                status_code=status.HTTP_401_UNAUTHORIZED,
-                detail="Error decoding token"
+                status_code=status.HTTP_401_UNAUTHORIZED, detail="Error decoding token"
             )
         return decoded_token

models/entity/user_entity.py

@@ -22,4 +22,4 @@ class User(SQLModel, table=True):
     created_at: datetime = Field(default_factory=datetime.utcnow, nullable=False)
     updated_at: datetime = Field(default_factory=datetime.utcnow, nullable=False)
     location: Optional[str] = None
-    #inventory: List["Inventory"] = Relationship(back_populates="user")
+    # inventory: List["Inventory"] = Relationship(back_populates="user")

models/service/user_service.py

@@ -5,6 +5,7 @@
 import sqlalchemy
 
 from helpers.exceptions import GrocerorError
+
 # from helpers.jwt import JWT
 from models.db import db_session
 from models.entity.user_entity import User