mirror of
https://github.com/thecyberlearn/chat-backend.git
synced 2026-08-18 22:52:54 +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>
27 lines
986 B
Python
27 lines
986 B
Python
from django.conf import settings
|
|
from django.contrib.messages.storage import default_storage
|
|
from django.utils.deprecation import MiddlewareMixin
|
|
|
|
|
|
class MessageMiddleware(MiddlewareMixin):
|
|
"""
|
|
Middleware that handles temporary messages.
|
|
"""
|
|
|
|
def process_request(self, request):
|
|
request._messages = default_storage(request)
|
|
|
|
def process_response(self, request, response):
|
|
"""
|
|
Update the storage backend (i.e., save the messages).
|
|
|
|
Raise ValueError if not all messages could be stored and DEBUG is True.
|
|
"""
|
|
# A higher middleware layer may return a request which does not contain
|
|
# messages storage, so make no assumption that it will be there.
|
|
if hasattr(request, "_messages"):
|
|
unstored_messages = request._messages.update(response)
|
|
if unstored_messages and settings.DEBUG:
|
|
raise ValueError("Not all temporary messages could be stored.")
|
|
return response
|