mirror of
https://github.com/thecyberlearn/chat-backend.git
synced 2026-08-18 23:32:53 +00:00
a0fce0b484
Features: - Django REST API backend - Multi-strategy web scraping system (Beautiful Soup, Playwright, Firecrawl) - Anti-detection features (proxy rotation, user-agent rotation) - Data export functionality (JSON, CSV, TXT) - Business and CrawledPage models - Enhanced crawling service with fallback mechanisms 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
from django.core import signals
|
|
from django.db.utils import (
|
|
DEFAULT_DB_ALIAS,
|
|
DJANGO_VERSION_PICKLE_KEY,
|
|
ConnectionHandler,
|
|
ConnectionRouter,
|
|
DatabaseError,
|
|
DataError,
|
|
Error,
|
|
IntegrityError,
|
|
InterfaceError,
|
|
InternalError,
|
|
NotSupportedError,
|
|
OperationalError,
|
|
ProgrammingError,
|
|
)
|
|
from django.utils.connection import ConnectionProxy
|
|
|
|
__all__ = [
|
|
"close_old_connections",
|
|
"connection",
|
|
"connections",
|
|
"reset_queries",
|
|
"router",
|
|
"DatabaseError",
|
|
"IntegrityError",
|
|
"InternalError",
|
|
"ProgrammingError",
|
|
"DataError",
|
|
"NotSupportedError",
|
|
"Error",
|
|
"InterfaceError",
|
|
"OperationalError",
|
|
"DEFAULT_DB_ALIAS",
|
|
"DJANGO_VERSION_PICKLE_KEY",
|
|
]
|
|
|
|
connections = ConnectionHandler()
|
|
|
|
router = ConnectionRouter()
|
|
|
|
# For backwards compatibility. Prefer connections['default'] instead.
|
|
connection = ConnectionProxy(connections, DEFAULT_DB_ALIAS)
|
|
|
|
|
|
# Register an event to reset saved queries when a Django request is started.
|
|
def reset_queries(**kwargs):
|
|
for conn in connections.all(initialized_only=True):
|
|
conn.queries_log.clear()
|
|
|
|
|
|
signals.request_started.connect(reset_queries)
|
|
|
|
|
|
# Register an event to reset transaction state and close connections past
|
|
# their lifetime.
|
|
def close_old_connections(**kwargs):
|
|
for conn in connections.all(initialized_only=True):
|
|
conn.close_if_unusable_or_obsolete()
|
|
|
|
|
|
signals.request_started.connect(close_old_connections)
|
|
signals.request_finished.connect(close_old_connections)
|