Router viewset registration via decorator (#8917)

This commit is contained in:
Jamie Cockburn 2023-03-27 15:36:40 +01:00
parent c9e7b68a4c
commit 3d85c9bf3f

View File

@ -49,7 +49,12 @@ class BaseRouter:
def __init__(self):
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:
basename = self.get_default_basename(viewset)