Making it work in newers versions of django

kaleidos · Mar 20, 2020 · 8018b46 · 8018b46
1 parent ec4972c
commit 8018b46
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/sampledatahelper/__init__.py b/sampledatahelper/__init__.py
@@ -1 +1 @@
-__version__ = (0, 4, 1)
+__version__ = (0, 5, 0)
diff --git a/sampledatahelper/handlers.py b/sampledatahelper/handlers.py
@@ -150,20 +150,20 @@ def _generate(self):
 
 class ForeignKeyHandler(BaseHandler):
     def _generate(self):
-        if self.instance.rel.limit_choices_to:
-            queryset = self.instance.rel.to.objects.filter(**self.instance.rel.limit_choices_to)
+        if self.instance.remote_field.limit_choices_to:
+            queryset = self.instance.remote_field.model.objects.filter(**self.instance.remote_field.limit_choices_to)
         else:
-            queryset = self.instance.rel.to.objects.all()
+            queryset = self.instance.remote_field.model.objects.all()
         return self.sd.db_object_from_queryset(queryset)
 
 
 class OneToOneHandler(ForeignKeyHandler):
     def _generate(self):
-        if self.instance.rel.limit_choices_to:
-            queryset = self.instance.rel.to.objects.filter(
-                **self.instance.rel.limit_choices_to
+        if self.instance.remote_field.limit_choices_to:
+            queryset = self.instance.remote_field.model.objects.filter(
+                **self.instance.remote_field.limit_choices_to
             ).exclude(id__in=self.instance.model.objects.all().values(self.instance.name))
         else:
-            quersyet = self.instance.rel.to.objects.all()
+            quersyet = self.instance.remote_field.model.objects.all()
             queryset = quersyet.exclude(id__in=self.instance.model.objects.all().values(self.instance.name))
         return self.sd.db_object_from_queryset(queryset)