@@ -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
4854class 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