Add date/datetime types support.

djorm_pgarray/fields.py

@@ -18,6 +18,8 @@
     'text': str,
     'double precision': float,
     'varchar': str,
+    'date': lambda x: x,
+    'datetime': lambda x: x,
 }
 
 

testing/pg_array_fields/models.py

@@ -52,6 +52,15 @@ class MultiTypeModel(models.Model):
     varchars = ArrayField(dbtype="varchar(30)")
     objects = ExpressionManager()
 
+class DateModel(models.Model):
+    dates = ArrayField(dbtype="date")
+    objects = ExpressionManager()
+
+
+class DateTimeModel(models.Model):
+    dates = ArrayField(dbtype="timestamp")
+    objects = ExpressionManager()
+
 
 class ChoicesModel(models.Model):
     choices = ArrayField(dbtype='text', choices=[('A', 'A'), ('B', 'B')])

testing/pg_array_fields/tests.py

@@ -1,5 +1,7 @@
 # -*- coding: utf-8 -*-
 
+import datetime
+
 from django.contrib.admin import AdminSite, ModelAdmin
 from django.core.serializers import serialize, deserialize
 from django.db import connection
@@ -16,6 +18,8 @@
                      MultiTypeModel,
                      ChoicesModel,
                      Item,
+                     DateModel,
+                     DateTimeModel,
                      ArrayExpression,
                      MacAddrModel)
 
@@ -79,6 +83,21 @@ def test_subquery(self):
         qs2 = Item.objects.filter(id__in=qs1.values('id'))
         # print(qs2.query.__str__())
 
+    def test_date(self):
+        d = datetime.date(2011, 11, 11)
+        instance = DateModel.objects.create(dates=[d])
+
+        instance = DateModel.objects.get(pk=instance.pk)
+        self.assertEqual(instance.dates[0], d)
+
+
+    def test_datetime(self):
+        d = datetime.datetime(2011, 11, 11, 11, 11, 11)
+        instance = DateTimeModel.objects.create(dates=[d])
+        instance = DateTimeModel.objects.get(pk=instance.pk)
+        self.assertEqual(instance.dates[0], d)
+
+
     def test_empty_create(self):
         instance = IntModel.objects.create(lista=[])
         instance = IntModel.objects.get(pk=instance.pk)