harini

backend/mongo_service.py

@@ -3,8 +3,10 @@ from pymongo.errors import ServerSelectionTimeoutError
 from datetime import datetime, timedelta
 from typing import Dict, List, Optional, Any
 
+
 class MongoService:
     def __init__(self, uri: str):
+        self.uri = uri  # Store URI for sync operations
         try:
             # Simple connection without custom SSL context
             self.client = AsyncIOMotorClient(
@@ -18,16 +20,15 @@ class MongoService:
             print(f"MongoDB connection failed: {e}")
             # Fallback to basic connection
             self.client = AsyncIOMotorClient(uri)
-            
-        self.db = self.client.openlearnx
         
+        self.db = self.client.openlearnx
         # Collections
         self.users = self.db.users
         self.questions = self.db.questions
         self.test_sessions = self.db.test_sessions
         self.certificates = self.db.certificates
         self.peer_reviews = self.db.peer_reviews
-    
+
     async def init_db(self):
         """Initialize database with indexes and sample data"""
         try:
@@ -38,13 +39,10 @@ class MongoService:
             # Create indexes
             await self.users.create_index("wallet_address", unique=True)
             await self.users.create_index("email", unique=True, sparse=True)
-            
             await self.questions.create_index("subject")
             await self.questions.create_index("difficulty")
-            
             await self.test_sessions.create_index("user_id")
             await self.test_sessions.create_index("created_at")
-            
             await self.certificates.create_index("user_id")
             await self.certificates.create_index("token_id", unique=True)
             
@@ -60,4 +58,85 @@ class MongoService:
             print(f"Database initialization error: {e}")
             print("Continuing without database initialization...")
 
-    # ... rest of your existing methods remain the same
+    async def get_user_by_wallet(self, wallet_address: str):
+        """Get user by wallet address"""
+        return await self.users.find_one({"wallet_address": wallet_address.lower()})
+
+    async def create_user(self, wallet_address: str):
+        """Create a new user"""
+        now = datetime.utcnow()
+        user = {
+            "wallet_address": wallet_address.lower(),
+            "created_at": now,
+            "last_login": now,
+            "total_tests": 0,
+            "certificates": []
+        }
+        result = await self.users.insert_one(user)
+        user["_id"] = result.inserted_id
+        return user
+
+    async def update_user_login(self, wallet_address: str):
+        """Update user's last login time"""
+        await self.users.update_one(
+            {"wallet_address": wallet_address.lower()},
+            {"$set": {"last_login": datetime.utcnow()}}
+        )
+
+    async def insert_sample_questions(self):
+        """Insert sample questions - implement based on your needs"""
+        # You'll need to implement this method based on your question structure
+        sample_questions = [
+            {
+                "subject": "Python",
+                "difficulty": "beginner",
+                "question": "What is a variable in Python?",
+                "options": ["A storage location", "A function", "A loop", "A condition"],
+                "correct_answer": 0,
+                "created_at": datetime.utcnow()
+            },
+            # Add more sample questions as needed
+        ]
+        await self.questions.insert_many(sample_questions)
+
+    async def close_connection(self):
+        """Close the database connection"""
+        if self.client:
+            self.client.close()
+            print("MongoDB connection closed")
+
+    def create_user_sync(self, wallet_address: str):
+        """Synchronous user creation using pymongo instead of motor"""
+        import pymongo
+        
+        # Create a synchronous connection for this operation only
+        client = pymongo.MongoClient(self.uri)
+        db = client.openlearnx
+        users = db.users
+        
+        try:
+            # Check if user exists
+            user = users.find_one({"wallet_address": wallet_address.lower()})
+            
+            if not user:
+                # Create new user
+                new_user = {
+                    "wallet_address": wallet_address.lower(),
+                    "created_at": datetime.utcnow(),
+                    "last_login": datetime.utcnow(),
+                    "total_tests": 0,
+                    "certificates": []
+                }
+                result = users.insert_one(new_user)
+                new_user["_id"] = result.inserted_id
+                return new_user
+            else:
+                # Update last login
+                users.update_one(
+                    {"wallet_address": wallet_address.lower()},
+                    {"$set": {"last_login": datetime.utcnow()}}
+                )
+                return user
+        finally:
+            # Always close the connection
+            client.close()