harini

backend/models/user.py

@@ -0,0 +1,30 @@
+from bson import ObjectId
+from datetime import datetime
+from pymongo.collection import Collection
+
+class UserModel:
+    def __init__(self, collection: Collection):
+        self.collection = collection
+    
+    async def get_by_wallet(self, wallet_address: str):
+        return await self.collection.find_one({"wallet_address": wallet_address.lower()})
+
+    async def create_user(self, wallet_address: str):
+        now = datetime.utcnow()
+        user = {
+            "wallet_address": wallet_address.lower(),
+            "created_at": now,
+            "last_login": now,
+            "total_tests": 0,
+            "certificates": []
+        }
+        result = await self.collection.insert_one(user)
+        user["_id"] = result.inserted_id
+        return user
+    
+    async def update_last_login(self, wallet_address: str):
+        now = datetime.utcnow()
+        await self.collection.update_one(
+            {"wallet_address": wallet_address.lower()},
+            {"$set": {"last_login": now}}
+        )