mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 11:11:10 -05:00
27 lines
1.0 KiB
Python
27 lines
1.0 KiB
Python
import pghistory
|
|
from django.contrib.auth.models import AnonymousUser
|
|
from django.core.handlers.wsgi import WSGIRequest
|
|
|
|
class RequestContextProvider(pghistory.context):
|
|
"""Custom context provider for pghistory that extracts information from the request."""
|
|
def __call__(self, request: WSGIRequest) -> dict:
|
|
return {
|
|
'user': str(request.user) if request.user and not isinstance(request.user, AnonymousUser) else None,
|
|
'ip': request.META.get('REMOTE_ADDR'),
|
|
'user_agent': request.META.get('HTTP_USER_AGENT'),
|
|
'session_key': request.session.session_key if hasattr(request, 'session') else None
|
|
}
|
|
|
|
# Initialize the context provider
|
|
request_context = RequestContextProvider()
|
|
|
|
class PgHistoryContextMiddleware:
|
|
"""
|
|
Middleware that ensures request object is available to pghistory context.
|
|
"""
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
response = self.get_response(request)
|
|
return response |