Add ASGI support (#127)

This commit updates the middleware to handle applications using ASGI

Type: feature

Co-authored-by: Gadhi Rodriguez <[email protected]>

pghistory/middleware.py

@@ -1,11 +1,12 @@
+from django.core.handlers.asgi import ASGIRequest as DjangoASGIRequest
 from django.core.handlers.wsgi import WSGIRequest as DjangoWSGIRequest
 from django.db import connection
 
 import pghistory
 from pghistory import config
 
 
-class WSGIRequest(DjangoWSGIRequest):
+class DjangoRequest:
     """
     Although Django's auth middleware sets the user in middleware,
     apps like django-rest-framework set the user in the view layer.
@@ -28,6 +29,14 @@ def __setattr__(self, attr, value):
         return super().__setattr__(attr, value)
 
 
+class WSGIRequest(DjangoRequest, DjangoWSGIRequest):
+    pass
+
+
+class ASGIRequest(DjangoRequest, DjangoASGIRequest):
+    pass
+
+
 def HistoryMiddleware(get_response):
     """
     Annotates the user/url in the pghistory context.
@@ -43,6 +52,8 @@ def middleware(request):
             with pghistory.context(user=user, url=request.path):
                 if isinstance(request, DjangoWSGIRequest):  # pragma: no branch
                     request.__class__ = WSGIRequest
+                elif isinstance(request, DjangoASGIRequest):  # pragma: no branch
+                    request.__class__ = ASGIRequest
 
                 return get_response(request)
         else: