mirror of
https://github.com/th30d4y/OpenLearnX.git
synced 2026-05-26 11:25:49 +00:00
5t4l1n
64 lines
2.5 KiB
Python
64 lines
2.5 KiB
Python
from motor.motor_asyncio import AsyncIOMotorClient
|
|
from pymongo.errors import ServerSelectionTimeoutError
|
|
from datetime import datetime, timedelta
|
|
from typing import Dict, List, Optional, Any
|
|
|
|
class MongoService:
|
|
def __init__(self, uri: str):
|
|
try:
|
|
# Simple connection without custom SSL context
|
|
self.client = AsyncIOMotorClient(
|
|
uri,
|
|
serverSelectionTimeoutMS=30000,
|
|
connectTimeoutMS=30000,
|
|
socketTimeoutMS=30000
|
|
)
|
|
print("MongoDB client initialized successfully")
|
|
except Exception as e:
|
|
print(f"MongoDB connection failed: {e}")
|
|
# Fallback to basic connection
|
|
self.client = AsyncIOMotorClient(uri)
|
|
|
|
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:
|
|
# Test connection first
|
|
await self.client.admin.command('ping')
|
|
print("MongoDB connection successful!")
|
|
|
|
# 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)
|
|
|
|
# Insert sample questions if none exist
|
|
if await self.questions.count_documents({}) == 0:
|
|
await self.insert_sample_questions()
|
|
print("Sample questions inserted successfully")
|
|
|
|
except ServerSelectionTimeoutError as e:
|
|
print(f"Failed to connect to MongoDB: {e}")
|
|
print("Continuing without database initialization...")
|
|
except Exception as e:
|
|
print(f"Database initialization error: {e}")
|
|
print("Continuing without database initialization...")
|
|
|
|
# ... rest of your existing methods remain the same
|