encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-27 00:19:53 +03:00
Code Issues Packages Projects Releases Wiki Activity

Refactored FBV code

Browse Source
This commit is contained in:
enrico 2022-09-05 18:50:15 +08:00
parent 7b18380321
commit f65e859e2f
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
rest_framework/decorators.py
View File

@ -47,20 +47,17 @@ def api_view(http_method_names=None):
allowed_methods = set(http_method_names) | {'options'} allowed_methods = set(http_method_names) | {'options'}
WrappedAPIView.http_method_names = [method.lower() for method in allowed_methods] WrappedAPIView.http_method_names = [method.lower() for method in allowed_methods]
def sync_handler(self, *args, **kwargs):
return func(*args, **kwargs)
async def async_handler(self, *args, **kwargs):
return await func(*args, **kwargs)
view_is_async = asyncio.iscoroutinefunction(func) view_is_async = asyncio.iscoroutinefunction(func)
if view_is_async: if view_is_async:
for method in http_method_names: async def handler(self, *args, **kwargs):
setattr(WrappedAPIView, method.lower(), async_handler) return await func(*args, **kwargs)
else: else:
for method in http_method_names: def handler(self, *args, **kwargs):
setattr(WrappedAPIView, method.lower(), sync_handler) return func(*args, **kwargs)
for method in http_method_names:
setattr(WrappedAPIView, method.lower(), handler)
WrappedAPIView.__name__ = func.__name__ WrappedAPIView.__name__ = func.__name__
WrappedAPIView.__module__ = func.__module__ WrappedAPIView.__module__ = func.__module__