Django 2.0 compatibility

tahajahangir · Jan 10, 2019 · 09b7a56 · 09b7a56
1 parent c6e9975
commit 09b7a56
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/bitfield/models.py b/bitfield/models.py
@@ -240,7 +240,12 @@ def __init__(self, fields):
     def contribute_to_class(self, cls, name):
         self.name = name
         self.model = cls
-        cls._meta.virtual_fields.append(self)
+
+        # virtual_fields was deprecated in Django 1.10 and removed in 2.0
+        private_fields = getattr(cls._meta, "private_fields", None)
+        if private_fields is None:
+            private_fields = cls._meta.virtual_fields
+        private_fields.append(self)
 
         signals.class_prepared.connect(self.validate_fields, sender=cls)