1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
from rest_framework.routers import DefaultRouter, SimpleRouter
from rest_framework_extensions.utils import compose_parent_pk_kwarg_name
class NestedRegistryItem:
def __init__(self, router, parent_prefix, parent_item=None, parent_viewset=None):
self.router = router
self.parent_prefix = parent_prefix
self.parent_item = parent_item
self.parent_viewset = parent_viewset
def register(self, prefix, viewset, basename, parents_query_lookups):
self.router._register(
prefix=self.get_prefix(
current_prefix=prefix,
parents_query_lookups=parents_query_lookups),
viewset=viewset,
basename=basename,
)
return NestedRegistryItem(
router=self.router,
parent_prefix=prefix,
parent_item=self,
parent_viewset=viewset
)
def get_prefix(self, current_prefix, parents_query_lookups):
return '{0}/{1}'.format(
self.get_parent_prefix(parents_query_lookups),
current_prefix
)
def get_parent_prefix(self, parents_query_lookups):
prefix = '/'
current_item = self
i = len(parents_query_lookups) - 1
while current_item:
parent_lookup_value_regex = getattr(
current_item.parent_viewset, 'lookup_value_regex', '[^/.]+')
prefix = '{parent_prefix}/(?P<{parent_pk_kwarg_name}>{parent_lookup_value_regex})/{prefix}'.format(
parent_prefix=current_item.parent_prefix,
parent_pk_kwarg_name=compose_parent_pk_kwarg_name(
parents_query_lookups[i]),
parent_lookup_value_regex=parent_lookup_value_regex,
prefix=prefix
)
i -= 1
current_item = current_item.parent_item
return prefix.strip('/')
class NestedRouterMixin:
def _register(self, *args, **kwargs):
return super().register(*args, **kwargs)
def register(self, *args, **kwargs):
self._register(*args, **kwargs)
return NestedRegistryItem(
router=self,
parent_prefix=self.registry[-1][0],
parent_viewset=self.registry[-1][1]
)
class ExtendedRouterMixin(NestedRouterMixin):
pass
class ExtendedSimpleRouter(ExtendedRouterMixin, SimpleRouter):
pass
class ExtendedDefaultRouter(ExtendedRouterMixin, DefaultRouter):
pass
|
