Skip to content

Commit 7ab0338

Browse files
Unregister everything in routers
1 parent 1ded96e commit 7ab0338

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

.gitlab-ci.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ build_zip:
119119
script:
120120
- make rpm RELEASE=${CI_BUILD_ID}
121121
artifacts:
122-
expire_in: 20 minutes
122+
expire_in: 30 minutes
123123
paths:
124124
- dist/
125125
retry: 2

polemarch/api/routers.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,14 +36,20 @@ def register_view(self, prefix, view, name=None):
3636
name = name or view().get_view_name().lower()
3737
self.custom_urls.append((prefix, view, name))
3838

39-
def unregister(self, prefix):
39+
def _unreg(self, prefix, objects_list):
4040
index = 0
41-
for reg_prefix, _, _ in self.registry:
41+
for reg_prefix, _, _ in objects_list:
4242
if reg_prefix == prefix:
43-
del self.registry[index]
43+
del objects_list[index]
4444
break
4545
index += 1
4646

47+
def unregister_view(self, prefix):
48+
self._unreg(prefix, self.custom_urls) # nocv
49+
50+
def unregister(self, prefix):
51+
self._unreg(prefix, self.registry)
52+
4753

4854
class APIRouter(_AbstractRouter):
4955
root_view_name = 'api-v1'
@@ -118,6 +124,9 @@ def register_router(self, prefix, router, name=None):
118124
name = name or router.root_view_name
119125
self.routers.append((prefix, router, name))
120126

127+
def unregister_router(self, prefix):
128+
self._unreg(prefix, self.routers) # nocv
129+
121130
def get_urls(self):
122131
urls = super(MainRouter, self).get_urls()
123132
for prefix, router, _ in self.routers:

0 commit comments

Comments
 (0)