From ee7f8c709bdf9fe95d61cee82155a845796243dd Mon Sep 17 00:00:00 2001 From: Benas Krencius Date: Wed, 13 Dec 2017 10:53:25 +0200 Subject: [PATCH] Fixes issue when role name (group title) contains whitespaces --- drf_roles/mixins.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/drf_roles/mixins.py b/drf_roles/mixins.py index 2c87a6c..2ddf250 100644 --- a/drf_roles/mixins.py +++ b/drf_roles/mixins.py @@ -25,7 +25,7 @@ def _call_role_fn(self, fn, *args, **kwargs): """Attempts to call a role-scoped method""" try: role_name = self._get_role(self.request.user) - role_fn = "{}_for_{}".format(fn, role_name) + role_fn = "{}_for_{}".format(fn, role_name.replace(' ', '_')) return getattr(self, role_fn)(*args, **kwargs) except (AttributeError, RoleError): return getattr(super(RoleViewSetMixin, self), fn)(*args, **kwargs)