orders component introduced and fixing apis

Author: lordlabakdas

api/helpers/inventory_helper.py

@@ -5,6 +5,7 @@
 from helpers.db_helpers import DBHelper
 from models.db import db_session
 from models.entity.inventory_entity import Inventory, InventoryCategory
+from models.entity.store_entity import Store
 from models.entity.user_entity import User
 
 logger = logging.getLogger()
@@ -23,6 +24,9 @@ def to_dict(self):
 class InventoryHelper(object):
     def __init__(self, user: User) -> None:
         self.user = user
+        self.store = (
+            db_session.query(Store).filter(Store.user_id == self.user.id).first()
+        )
 
     def add_inventory(
         self, name: str, quantity: int, category: Enum, notes: str = None
@@ -43,6 +47,7 @@ def add_inventory(
                     category=InventoryCategory(category).name,
                     user_id=self.user.id,
                     notes=notes,
+                    store_id=self.store.id,
                 )
             db_session.add(inventory_obj)
             db_session.commit()

api/inventory_api.py

@@ -12,7 +12,7 @@
 from helpers.jwt import auth_required
 from models.entity.user_entity import User
 
-logger = logging.getLogger("groceror")
+logger = logging.getLogger(__name__)
 inventory_apis = APIRouter(prefix="/inventory", tags=["inventory"])
 
 

api/order_api.py

@@ -0,0 +1,30 @@
+import logging
+
+from fastapi import APIRouter, Depends
+
+from helpers.jwt import auth_required
+from api.validators.order_validation import Order
+from engine import publisher
+from models.entity.user_entity import User
+from models.service.orders_service import OrderService
+
+logger = logging.getLogger(__name__)
+order_apis = APIRouter(prefix="/order", tags=["order"])
+
+
+@order_apis.post("/create-order", response_model=Order)
+async def create_order(order: Order, current_user: User = Depends(auth_required)):
+    logger.info(f"Creating order for user {current_user.id}")
+    # import pdb; pdb.set_trace()
+    # order_service = OrderService()
+    # order_service.create_order(order, current_user)
+    import pdb
+
+    pdb.set_trace()
+    return publisher.publish_message(
+        queue_name="order_queue",
+        routing_key="order_queue",
+        event="order_created",
+        order="Order placed",
+    )
+    # return order

api/user_api.py

@@ -61,7 +61,10 @@ async def register(registration_payload: RegistrationPayload):
         )
     else:
         publisher.publish_message(
-            event="user_registered", routing_key="email_queue", new_user=new_user
+            queue_name="email_queue",
+            routing_key="email_queue",
+            event="user_registered",
+            **new_user.dict(),
         )
         return {"id": new_user.id}
 

api/validators/order_validation.py

@@ -0,0 +1,16 @@
+from datetime import datetime
+from pydantic import BaseModel, Field
+
+from typing import List
+from uuid import UUID, uuid4
+
+
+class Order(BaseModel):
+    items: List[UUID]
+    total_price: float
+    status: str
+    created_at: datetime
+    updated_at: datetime
+    order_id: UUID = Field(default_factory=uuid4)
+    order_date: datetime = Field(default_factory=datetime.utcnow)
+    user_id: UUID

engine/publisher.py

@@ -2,40 +2,33 @@
 import logging
 import pika
 
-from models.entity.user_entity import User
-
 logger = logging.getLogger(__name__)
 
-def publish_message(event: str, routing_key: str, new_user: User):
+
+def publish_message(event: str, routing_key: str, queue_name: str, **kwargs):
     # Publish registration event to email service
-        try:
-            connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
-            channel = connection.channel()
-            
-            # Ensure exchange exists
-            channel.queue_declare(queue='email_queue', durable=True)
-            
-            # Create message payload
-            message = {
-                "event": event,
-                "user_id": str(new_user.id),
-                "email": new_user.email,
-                "name": new_user.name
-            }
-            
-            # Publish message
-            channel.basic_publish(
-                exchange='',
-                routing_key=routing_key,
-                body=json.dumps(message),
-                properties=pika.BasicProperties(
-                    delivery_mode=2,  # make message persistent
-                    content_type='application/json'
-                )
-            )
-            
-            connection.close()
-            
-        except Exception as e:
-            logger.error(f"Failed to publish registration event: {str(e)}")
-            # Don't raise exception - registration succeeded even if notification fails
+    try:
+        connection = pika.BlockingConnection(pika.ConnectionParameters("localhost"))
+        channel = connection.channel()
+
+        # Ensure exchange exists
+        channel.queue_declare(queue=queue_name, durable=True)
+
+        # Create message payload
+        message = {"event": event, **kwargs}
+        # Publish message
+        channel.basic_publish(
+            exchange="",
+            routing_key=routing_key,
+            body=json.dumps(message),
+            properties=pika.BasicProperties(
+                delivery_mode=2,  # make message persistent
+                content_type="application/json",
+            ),
+        )
+
+        connection.close()
+
+    except Exception as e:
+        logger.error(f"Failed to publish registration event: {str(e)}")
+        # Don't raise exception - registration succeeded even if notification fails

main.py

@@ -12,6 +12,7 @@
 from api.cart_api import cart_apis
 from api.google_login import google_login_apis
 from api.inventory_api import inventory_apis
+from api.order_api import order_apis
 from api.store_api import store_apis
 from api.user_api import user_apis
 from models.db import create_db_and_tables
@@ -53,6 +54,7 @@ async def welcome():
 app.include_router(inventory_apis)
 app.include_router(store_apis)
 app.include_router(cart_apis)
+app.include_router(order_apis)
 
 create_db_and_tables()
 

models/entity/orders_entity.py

@@ -0,0 +1,19 @@
+from datetime import datetime
+from typing import Dict, List, Optional
+from uuid import UUID, uuid4
+from sqlmodel import Field, SQLModel
+
+
+class Order(SQLModel):
+    id: Optional[UUID] = Field(default_factory=uuid4, primary_key=True)
+    user_id: UUID = Field(foreign_key="user.id")
+    total_price: float = Field(default=0.0)
+    status: str = Field(default="pending")
+    created_at: datetime = Field(default_factory=datetime.utcnow)
+    updated_at: datetime = Field(default_factory=datetime.utcnow)
+    items: List[UUID] = Field(default_factory=list)
+    order_id: UUID = Field(default_factory=uuid4)
+    order_date: datetime = Field(default_factory=datetime.utcnow)
+
+    class Config:
+        orm_mode = True

models/service/orders_service.py

@@ -0,0 +1,31 @@
+from typing import List
+from uuid import UUID
+
+from api.validators.order_validation import Order
+from models.db import db_session
+from models.entity.orders_entity import Order as OrderEntity
+
+
+class OrderService:
+    def __init__(self):
+        pass
+
+    def create_order(self, order: Order, current_user: str):
+        order_entity = OrderEntity(
+            order_id=order.order_id,
+            order_date=order.order_date,
+            user_id=current_user.id,
+            items=order.items,
+            total_price=order.total_price,
+            status=order.status,
+        )
+        db_session.add(order_entity)
+        db_session.commit()
+
+    def get_order_by_id(self, order_id: UUID) -> Order:
+        order = (
+            db_session.query(OrderEntity)
+            .filter(OrderEntity.order_id == order_id)
+            .first()
+        )
+        return order

models/service/store_service.py

@@ -37,9 +37,12 @@ def create_store(
         phone: str,
         email: str,
         website: str = None,
+        latitude: float = None,
+        longitude: float = None,
     ) -> Store:
         try:
-            latitude, longitude = get_coordinates(address)
+            if latitude is None or longitude is None:
+                latitude, longitude = get_coordinates(address)
             store = Store(
                 name=name,
                 user_id=user_id,