Ticket #15277: 15277.2.patch

django/contrib/gis/tests/__init__.py

@@ -61 +61 @@
     from django.contrib.gis.tests import test_measure
     suite.addTest(test_measure.suite())
 
+    # Adding the ogrinspect tests.
+    from django.contrib.gis.tests import test_ogrinspect
+    suite.addTest(test_ogrinspect.suite())
+
     # Adding GDAL tests, and any test suite that depends on GDAL, to the
     # suite if GDAL is available.
     from django.contrib.gis.gdal import HAS_GDAL

new file django/contrib/gis/tests/test_ogrinspect.py

@@ +1 @@
+import os
+from django.contrib.gis.utils.ogrinspect import ogrinspect
+from django.contrib.gis.geometry.test_data import TEST_DATA
+from django.utils import unittest
+
+class OGRInspectTest(unittest.TestCase):
+    def test_poly(self):
+        shp_file = os.path.join(TEST_DATA, 'test_poly', 'test_poly.shp')
+        model_def = ogrinspect(shp_file, 'MyModel')
+
+        expected = [
+            '# This is an auto-generated Django model module created by ogrinspect.',
+            'from django.contrib.gis.db import models',
+            '',
+            'class MyModel(models.Model):',
+            '    float = models.FloatField()',
+            '    int = models.FloatField()',
+            '    str = models.CharField(max_length=80)',
+            '    geom = models.PolygonField(srid=-1)',
+            '    objects = models.GeoManager()',
+        ]
+
+        self.assertEqual(model_def, '\n'.join(expected))
+
+    def test_date_field(self):
+        shp_file = os.path.join(TEST_DATA, 'cities', 'cities.shp')
+        model_def = ogrinspect(shp_file, 'City')
+
+        expected = [
+            '# This is an auto-generated Django model module created by ogrinspect.',
+            'from django.contrib.gis.db import models',
+            '',
+            'class City(models.Model):',
+            '    name = models.CharField(max_length=80)',
+            '    population = models.FloatField()',
+            '    density = models.FloatField()',
+            '    created = models.DateField()',
+            '    geom = models.PointField(srid=-1)',
+            '    objects = models.GeoManager()',
+        ]
+
+        self.assertEqual(model_def, '\n'.join(expected))
+
+def suite():
+    s = unittest.TestSuite()
+    s.addTest(unittest.makeSuite(OGRInspectTest))
+    return s
+
+def run(verbosity=2):
+    unittest.TextTestRunner(verbosity=verbosity).run(suite())
+
+if __name__=="__main__":
+    run()

django/contrib/gis/utils/ogrinspect.py

@@ -8 +8 @@
 from itertools import izip
 # Requires GDAL to use.
 from django.contrib.gis.gdal import DataSource
-from django.contrib.gis.gdal.field import OFTDate, OFTDateTime, OFTInteger, OFTReal, OFTString
+from django.contrib.gis.gdal.field import OFTDate, OFTDateTime, OFTInteger, OFTReal, OFTString, OFTTime
 
 def mapping(data_source, geom_name='geom', layer_key=0, multi_geom=False):
     """
@@ -189 +189 @@
             yield '    %s = models.DateField(%s)' % (mfield, kwargs_str[2:])
         elif field_type is OFTDateTime:
             yield '    %s = models.DateTimeField(%s)' % (mfield, kwargs_str[2:])
-        elif field_type is OFTDate:
+        elif field_type is OFTTime:
             yield '    %s = models.TimeField(%s)' % (mfield, kwargs_str[2:])
         else:
             raise TypeError('Unknown field type %s in %s' % (field_type, mfield))