Spaces:
Running
Running
File size: 2,050 Bytes
a8a2cf5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
"""
Mock Database Module - Works without real database connection
This allows your app to run and test AI features without needing
a working database connection.
"""
from typing import Optional, Dict, Any
from datetime import datetime
import asyncio
class MockAsyncSession:
"""Mock session that simulates database operations."""
def __init__(self):
# Simulate a simple in-memory "database"
self.users = {}
self.sessions = {}
self.conversations = {}
self.messages = {}
self._user_counter = 1
self._session_counter = 1
async def execute(self, query):
"""Mock execute method."""
return MockResult()
async def commit(self):
"""Mock commit method."""
pass
async def rollback(self):
"""Mock rollback method."""
pass
def add(self, obj):
"""Mock add method."""
pass
async def refresh(self, obj):
"""Mock refresh method."""
pass
class MockResult:
"""Mock query result."""
def scalar_one_or_none(self):
"""Return None to simulate no existing user."""
return None
def fetchval(self, *args):
"""Mock fetchval."""
return None
def fetchrow(self, *args):
"""Mock fetchrow."""
return None
class MockEngine:
"""Mock database engine."""
def dispose(self):
"""Mock dispose method."""
pass
# Mock functions that replace the real database functions
async def get_session():
"""Return a mock session."""
return MockAsyncSession()
async def init_db():
"""Mock database initialization."""
print("[MOCK] Database initialized (mock mode)")
async def dispose_engine():
"""Mock engine disposal."""
print("[MOCK] Database disposed (mock mode)")
def get_engine():
"""Return mock engine."""
return MockEngine()
# Export the mock functions
__all__ = ['get_session', 'init_db', 'dispose_engine', 'get_engine']
|