from optparse import make_option

from django.core.management.base import CommandError

from django.core.management.commands.inspectdb import Command as InspectDBCommand

class Command(InspectDBCommand):

    db_module = 'django.contrib.gis.db'

    gis_tables = {}

    def get_field_type(self, connection, table_name, row):

        field_type, field_params, field_notes = super(Command, self).get_field_type(connection, table_name, row)

        if field_type == 'GeometryField':

            geo_col = row[0]

            # Getting a more specific field type and any additional parameters

            # from the `get_geometry_type` routine for the spatial backend.

            field_type, geo_params = connection.introspection.get_geometry_type(table_name, geo_col)

            field_params.update(geo_params)

            # Adding the table name and column to the `gis_tables` dictionary, this

            # allows us to track which tables need a GeoManager.

            if table_name in self.gis_tables:

                self.gis_tables[table_name].append(geo_col)

            else:

                self.gis_tables[table_name] = [geo_col]

        return field_type, field_params, field_notes

    def get_meta(self, table_name):

        meta_lines = super(Command, self).get_meta(table_name)

        if table_name in self.gis_tables:

            # If the table is a geographic one, then we need make

            # GeoManager the default manager for the model.

            meta_lines.insert(0, '    objects = models.GeoManager()')

        return meta_lines