Exposed the request to ObjectEditView's alter_obj()

2017-02-15 15:03:31 -05:00
parent c26e00b5bd
commit b69564f5c9
4 changed files with 13 additions and 13 deletions
--- a/netbox/dcim/views.py
+++ b/netbox/dcim/views.py
@@ -1517,9 +1517,9 @@ class ModuleEditView(PermissionRequiredMixin, ComponentEditView):
    model = Module
    form_class = forms.ModuleForm

-    def alter_obj(self, obj, args, kwargs):
-        if 'device' in kwargs:
-            obj.device = get_object_or_404(Device, pk=kwargs['device'])
+    def alter_obj(self, obj, request, url_args, url_kwargs):
+        if 'device' in url_kwargs:
+            obj.device = get_object_or_404(Device, pk=url_kwargs['device'])
        return obj