feat: finish login system

- adding response based model dto
- adding logic for login system

app/controllers/auth_controller.py

@@ -11,13 +11,10 @@ class AuthController:
 
     def login(self):
         data = request.get_json()
-        users = self.auth_service.login(data)
+        response = self.auth_service.login(data)
-        response = {
+        if response.success:
-            "status": True,
+            return jsonify(response.to_dict()), 200
-            "message": "success retrive data",
+        return jsonify(response.to_dict()), 400
-            "data": users,
-        }
-        return jsonify(response)
 
     def register(self):
         return jsonify({"message": "register"})

app/di_container.py

@@ -17,4 +17,4 @@ class Container(containers.DeclarativeContainer):
     user_service = providers.Factory(UserService, user_repository)
 
     # controllers
-    auth_controller = providers.Factory(AuthController, auth_service, user_service)
+    auth_controller = providers.Factory(AuthController, user_service, auth_service)

app/models/__init__.py

@@ -0,0 +1 @@
+from .dto import ApiResponse

app/models/dto/__init__.py

@@ -0,0 +1 @@
+from .response import ApiResponse

app/models/dto/response.py

@@ -0,0 +1,22 @@
+from pydantic import BaseModel
+from typing import Generic, TypeVar, Optional
+
+T = TypeVar("T")
+
+
+class ApiResponse(BaseModel, Generic[T]):
+    success: bool
+    message: str
+    data: Optional[T] = None
+
+    def to_json(self) -> str:
+        """
+        Convert the model to a properly formatted JSON string.
+        """
+        return self.model_dump_json(indent=4)
+
+    def to_dict(self) -> dict:
+        """
+        Convert the model to a dictionary with proper key-value pairs.
+        """
+        return self.model_dump()

app/repositories/user_repository.py

@@ -8,21 +8,16 @@ class UserRepository:
 
     def get_all_users(self):
         try:
-            # mongo.db.command("ping")
+
-            # # Retrieve all users, excluding "_id"
+            users = list(self.collection.find({}, {"_id": 0}))
-            # users = list(self.collection.find({}, {"_id": 0}))
+
-            # print(f"✅ All Users Data: {users}", file=sys.stderr)
+            return users if users else []
-            return None
         except Exception as e:
-            print(f"❌ Error fetching users: {e}", file=sys.stderr)
             return []
 
     def get_user_by_email(self, email):
         try:
-            # user = mongo.db.find_one({"email": email}, {"_id": 0})
+            user = self.collection.find_one({"email": email}, {"_id": 0})
-            # print(f"✅ User Data for {email}: {user}", file=sys.stderr)
+            return user if user else None
-            # return user if user else None
-            return None
         except Exception as e:
-            print(f"❌ Error fetching user: {e}", file=sys.stderr)
             return None

app/services/auth_service.py

@@ -1,4 +1,5 @@
 from repositories import UserRepository
+from models import ApiResponse
 
 
 class AuthService:
@@ -13,13 +14,17 @@ class AuthService:
             user_data = self.user_repository.get_user_by_email(email)
 
             if user_data == None:
-                return None
+                return ApiResponse(success=False, message="User not found", data=None)
 
             if user_data["password"] == password:
                 del user_data["password"]
-                return user_data
+                return ApiResponse(
+                    success=True, message="Login success", data=user_data
+                )
 
-            return None
+            return ApiResponse(success=False, message="Invalid password", data=None)
         except Exception as e:
             print(f"the issue is {e}")
-            return None
+            return ApiResponse(
+                success=False, message="Internal server error", data=None
+            )