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

Router viewset registration via decorator (#8917)

Browse Source
This commit is contained in:
Jamie Cockburn 2023-03-27 15:36:40 +01:00
parent c9e7b68a4c
commit 3d85c9bf3f
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
rest_framework/routers.py
View File

@ -49,7 +49,12 @@ class BaseRouter:
def __init__(self): def __init__(self):
self.registry = [] self.registry = []
def register(self, prefix, viewset, basename=None): def register(self, prefix, viewset=None, basename=None):
if viewset is None:
def decorator(viewset):
self.register(prefix, viewset, basename=basename)
return viewset
return decorator
if basename is None: if basename is None:
basename = self.get_default_basename(viewset) basename = self.get_default_basename(viewset)