Support async view methods in extend_schema_view.

tfranzel · Sep 12, 2024 · 6b857ef · 6b857ef
1 parent 01ec586
commit 6b857ef
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/drf_spectacular/drainage.py b/drf_spectacular/drainage.py
@@ -182,8 +182,12 @@ def get_view_method_names(view, schema=None) -> List[str]:
     return [
         item for item in dir(view) if callable(getattr(view, item)) and (
             item in view.http_method_names
-            or item in schema.method_mapping.values()
-            or item == 'list'
+            or (
+                item in schema.async_method_mapping.values()
+                if view.view_is_async
+                else item in schema.method_mapping.values()
+            )
+            or item == ('alist' if view.view_is_async else 'list')
             or hasattr(getattr(view, item), 'mapping')
         )
     ]

diff --git a/drf_spectacular/openapi.py b/drf_spectacular/openapi.py
@@ -57,6 +57,13 @@ class AutoSchema(ViewInspector):
         'patch': 'partial_update',
         'delete': 'destroy',
     }
+    async_method_mapping = {
+        'get': 'aretrieve',
+        'post': 'acreate',
+        'put': 'aupdate',
+        'patch': 'partial_aupdate',
+        'delete': 'adestroy',
+    }
 
     def get_operation(
             self,