From b975c75c2fb1dccc1e03754db3f78d1e685057be Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Fri, 22 Feb 2019 04:01:46 -0400 Subject: [PATCH] ContentTypeViewMixin allow chaging URL kwargs Signed-off-by: Roberto Rosario --- mayan/apps/common/mixins.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/mayan/apps/common/mixins.py b/mayan/apps/common/mixins.py index 6e278a68f4..329bf323da 100644 --- a/mayan/apps/common/mixins.py +++ b/mayan/apps/common/mixins.py @@ -29,10 +29,20 @@ __all__ = ( class ContentTypeViewMixin(object): + content_type_url_kw_args = { + 'app_label': 'app_label', + 'model': 'model' + } + def get_content_type(self): return get_object_or_404( - klass=ContentType, app_label=self.kwargs['app_label'], - model=self.kwargs['model'] + klass=ContentType, + app_label=self.kwargs[ + self.content_type_url_kw_args['app_label'] + ], + model=self.kwargs[ + self.content_type_url_kw_args['model'] + ] )