Ticket #15277: 15277.3.patch

deleted file django/contrib/gis/management/base.py

@@ -1 @@
-from django.core.management.base import BaseCommand, CommandError
-
-class ArgsCommand(BaseCommand):
-    """
-    Command class for commands that take multiple arguments.
-    """
-    args = '<arg arg ...>'
-
-    def handle(self, *args, **options):
-        if not args:
-            raise CommandError('Must provide the following arguments: %s' % self.args)
-        return self.handle_args(*args, **options)
-
-    def handle_args(self, *args, **options):
-        raise NotImplementedError()

django/contrib/gis/management/commands/ogrinspect.py

@@ -1 +1 @@
 import os
 from optparse import make_option
 from django.contrib.gis import gdal
-from django.contrib.gis.management.base import ArgsCommand, CommandError
+from django.core.management.base import LabelCommand, CommandError
 
 def layer_option(option, opt, value, parser):
     """
@@ -26 +26 @@
         dest = [s for s in value.split(',')]
     setattr(parser.values, option.dest, dest)
     
-class Command(ArgsCommand):
+class Command(LabelCommand):
     help = ('Inspects the given OGR-compatible data source (e.g., a shapefile) and outputs\n'
             'a GeoDjango model with the given model name. For example:\n'
             ' ./manage.py ogrinspect zipcode.shp Zipcode')
     args = '[data_source] [model_name]'
 
-    option_list = ArgsCommand.option_list + (
+    option_list = LabelCommand.option_list + (
         make_option('--blank', dest='blank', type='string', action='callback',  
                     callback=list_option, default=False,
                     help='Use a comma separated list of OGR field names to add '
@@ -72 +72 @@
 
     requires_model_validation = False
 
-    def handle_args(self, *args, **options):
+    def handle(self, *args, **options):
         try:
             data_source, model_name = args
         except ValueError:
@@ -81 +81 @@
         if not gdal.HAS_GDAL:
             raise CommandError('GDAL is required to inspect geospatial data sources.')
 
-        # TODO: Support non file-based OGR datasources.
-        if not os.path.isfile(data_source):
-            raise CommandError('The given data source cannot be found: "%s"' % data_source)
-        
         # Removing options with `None` values.
         options = dict([(k, v) for k, v in options.items() if not v is None])
 
@@ -97 +93 @@
         # Whether the user wants to generate the LayerMapping dictionary as well.
         show_mapping = options.pop('mapping', False)
 
-        # Popping the verbosity global option, as it's not accepted by `_ogrinspect`.
+        # Getting rid of settings that `_ogrinspect` doesn't like.
         verbosity = options.pop('verbosity', False)
+        settings = options.pop('settings', False)
 
         # Returning the output of ogrinspect with the given arguments
         # and options.
@@ -119 +116 @@
                            '%s_mapping = {' % model_name.lower()])
             output.extend(["    '%s' : '%s'," % (rev_mapping[ogr_fld], ogr_fld) for ogr_fld in ds[options['layer_key']].fields])
             output.extend(["    '%s' : '%s'," % (options['geom_name'], mapping_dict[options['geom_name']]), '}'])
-        return '\n'.join(output)
+        return '\n'.join(output) + '\n'

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))