From cfbe7af9fe7a4eb5a1fe45bf3cf3eba723017f35 Mon Sep 17 00:00:00 2001 From: Arun Lobo Date: Wed, 26 Oct 2016 20:26:53 +0100 Subject: [PATCH] Fix for django version 1.9+ --- djorm_pgarray/fields.py | 9 ++++++--- testing/settings.py | 2 +- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/djorm_pgarray/fields.py b/djorm_pgarray/fields.py index 01a04a1..8296a83 100644 --- a/djorm_pgarray/fields.py +++ b/djorm_pgarray/fields.py @@ -49,7 +49,7 @@ def _unserialize(value): return _cast_to_unicode(value) -class ArrayField(six.with_metaclass(models.SubfieldBase, models.Field)): +class ArrayField(models.TextField): empty_strings_allowed = False def __init__(self, dbtype="int", type_cast=None, dimension=1, *args, **kwargs): @@ -390,9 +390,12 @@ class AnyIContainsLookup(AnyContainsLookup): class IndexTransform(Transform): def __init__(self, index, field, *args, **kwargs): - super(IndexTransform, self).__init__(*args, **kwargs) + if django.VERSION[:2] >= (1, 9): + super(IndexTransform, self).__init__(*args, output_field=field, **kwargs) + else: + super(IndexTransform, self).__init__(*args, **kwargs) + self.field = field self.index = index - self.field = field def as_sql(self, qn, connection): lhs, params = qn.compile(self.lhs) diff --git a/testing/settings.py b/testing/settings.py index fb4a17a..0daecc6 100644 --- a/testing/settings.py +++ b/testing/settings.py @@ -48,4 +48,4 @@ "pg_array_fields", ] -TEST_RUNNER = "django.test.simple.DjangoTestSuiteRunner" +TEST_RUNNER = "django.test.runner.DiscoverRunner"