Replace Router.get_default_base_name by Router.get_default_basename

Deprecated in DRF 3.9.0
This commit is contained in:
Santiago L 2021-06-08 13:34:36 +02:00
parent 81c67778e5
commit 70f7551e7d
1 changed files with 9 additions and 9 deletions

View File

@ -18,33 +18,33 @@ class LogApiMixin(object):
message = _('Added.') message = _('Added.')
self.log(request, message, ADDITION, instance=self.serializer.instance) self.log(request, message, ADDITION, instance=self.serializer.instance)
return response return response
def perform_create(self, serializer): def perform_create(self, serializer):
""" stores serializer for accessing instance on create() """ """ stores serializer for accessing instance on create() """
super(LogApiMixin, self).perform_create(serializer) super(LogApiMixin, self).perform_create(serializer)
self.serializer = serializer self.serializer = serializer
def update(self, request, *args, **kwargs): def update(self, request, *args, **kwargs):
from django.contrib.admin.models import CHANGE from django.contrib.admin.models import CHANGE
response = super(LogApiMixin, self).update(request, *args, **kwargs) response = super(LogApiMixin, self).update(request, *args, **kwargs)
message = _('Changed data') message = _('Changed data')
self.log(request, message, CHANGE) self.log(request, message, CHANGE)
return response return response
def partial_update(self, request, *args, **kwargs): def partial_update(self, request, *args, **kwargs):
from django.contrib.admin.models import CHANGE from django.contrib.admin.models import CHANGE
response = super(LogApiMixin, self).partial_update(request, *args, **kwargs) response = super(LogApiMixin, self).partial_update(request, *args, **kwargs)
message = _('Changed %s') % response.data message = _('Changed %s') % response.data
self.log(request, message, CHANGE) self.log(request, message, CHANGE)
return response return response
def destroy(self, request, *args, **kwargs): def destroy(self, request, *args, **kwargs):
from django.contrib.admin.models import DELETION from django.contrib.admin.models import DELETION
message = _('Deleted') message = _('Deleted')
self.log(request, message, DELETION) self.log(request, message, DELETION)
response = super(LogApiMixin, self).destroy(request, *args, **kwargs) response = super(LogApiMixin, self).destroy(request, *args, **kwargs)
return response return response
def log(self, request, message, action, instance=None): def log(self, request, message, action, instance=None):
from django.contrib.admin.models import LogEntry from django.contrib.admin.models import LogEntry
instance = instance or self.get_object() instance = instance or self.get_object()
@ -64,21 +64,21 @@ class LinkHeaderRouter(DefaultRouter):
APIRoot = import_class(settings.ORCHESTRA_API_ROOT_VIEW) APIRoot = import_class(settings.ORCHESTRA_API_ROOT_VIEW)
APIRoot.router = self APIRoot.router = self
return APIRoot.as_view() return APIRoot.as_view()
def register(self, prefix, viewset, base_name=None): def register(self, prefix, viewset, base_name=None):
""" inserts link headers on every viewset """ """ inserts link headers on every viewset """
if base_name is None: if base_name is None:
base_name = self.get_default_base_name(viewset) base_name = self.get_default_basename(viewset)
insert_links(viewset, base_name) insert_links(viewset, base_name)
self.registry.append((prefix, viewset, base_name)) self.registry.append((prefix, viewset, base_name))
def get_viewset(self, prefix_or_model): def get_viewset(self, prefix_or_model):
for _prefix, viewset, __ in self.registry: for _prefix, viewset, __ in self.registry:
if _prefix == prefix_or_model or viewset.queryset.model == prefix_or_model: if _prefix == prefix_or_model or viewset.queryset.model == prefix_or_model:
return viewset return viewset
msg = "%s does not have a regiestered viewset" % prefix_or_model msg = "%s does not have a regiestered viewset" % prefix_or_model
raise KeyError(msg) raise KeyError(msg)
def insert(self, prefix_or_model, name, field, **kwargs): def insert(self, prefix_or_model, name, field, **kwargs):
""" Dynamically add new fields to an existing serializer """ """ Dynamically add new fields to an existing serializer """
viewset = self.get_viewset(prefix_or_model) viewset = self.get_viewset(prefix_or_model)