3 (h# @sldZddlmZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZddd Zd d ZdddZd S)z This module is for inspecting OGR data sources and generating either models for GeoDjango and/or mapping dictionaries for use with the `LayerMapping` utility. ) DataSource)OFTDate OFTDateTime OFTInteger OFTInteger64OFTReal OFTStringOFTTime)six)zipgeomFcCst|tjrt|}nt|tr"ntdi}x:||jD],}|j}|dddkr^|d7}|||<q:W||j}|r|jt |j ||<|S)a Given a DataSource, generates a dictionary that may be used for invoking the LayerMapping utility. Keyword Arguments: `geom_name` => The name of the geometry field to use for the model. `layer_key` => The key for specifying which layer in the DataSource to use; defaults to 0 (the first layer). May be an integer index or a string identifier for the layer. `multi_geom` => Boolean (default: False) - specify as multigeometry. z>Data source parameter must be a string or a DataSource object.N_field) isinstancer string_typesr TypeErrorfieldslower geom_typeto_multistrupper) data_source geom_name layer_key multi_geom_mappingrmfieldgtyper!G/tmp/pip-install-q3hcpn_q/Django/django/contrib/gis/utils/ogrinspect.pymappings      r#cOsdjddt||DS)a Given a data source (either a string or a DataSource object) and a string model name this function will generate a GeoDjango model. Usage: >>> from django.contrib.gis.utils import ogrinspect >>> ogrinspect('/path/to/shapefile.shp','NewModel') ...will print model definition to stout or put this in a python script and use to redirect the output to a new model like: $ python generate_model.py > myapp/models.py # generate_model.py from django.contrib.gis.utils import ogrinspect shp_file = 'data/mapping_hacks/world_borders.shp' model_name = 'WorldBorders' print(ogrinspect(shp_file, model_name, multi_geom=True, srid=4326, geom_name='shapes', blank=True)) Required Arguments `datasource` => string or DataSource object to file pointer `model name` => string of name of new model class to create Optional Keyword Arguments `geom_name` => For specifying the model name for the Geometry Field. Otherwise will default to `geom` `layer_key` => The key for specifying which layer in the DataSource to use; defaults to 0 (the first layer). May be an integer index or a string identifier for the layer. `srid` => The SRID to use for the Geometry Field. If it can be determined, the SRID of the datasource is used. `multi_geom` => Boolean (default: False) - specify as multigeometry. `name_field` => String - specifies a field name to return for the `__unicode__`/`__str__` function (which will be generated if specified). `imports` => Boolean (default: True) - set to False to omit the `from django.contrib.gis.db import models` code from the autogenerated models thus avoiding duplicated imports when building more than one model by batching ogrinspect() `decimal` => Boolean or sequence (default: False). When set to True all generated model fields corresponding to the `OFTReal` type will be `DecimalField` instead of `FloatField`. A sequence of specific field names to generate as `DecimalField` may also be used. `blank` => Boolean or sequence (default: False). When set to True all generated model fields will have `blank=True`. If the user wants to give specific fields to have blank, then a list/tuple of OGR field names may be used. `null` => Boolean (default: False) - When set to True all generated model fields will have `null=True`. If the user wants to specify give specific fields to have null, then a list/tuple of OGR field names may be used. Note: This routine calls the _ogrinspect() helper to do the heavy lifting.  css|] }|VqdS)Nr!).0sr!r!r" yszogrinspect..)join _ogrinspect)argskwargsr!r!r" ogrinspect5sDr,NTc #st|tjrt|}nt|tr"ntd||} | jfdd} | | | | | |} fdd}|rdVdVdVd |Vxht| j| j| j D]N\}}}}|j }|dd d kr|d 7}||}|t kr|j | krd||||fVnd||dd fVq|t kr>d||dd fVq|t kr`d||dd fVq|tkr|d|||fVq|tkrd||dd fVq|tkrd||dd fVq|tkrd||dd fVqtd||fqW| j}|r |j|j}|d kr^| jd kr,d}n0| jj}|d krDd}n|dkrTd}nd|}nd|}d|||fV|rdVdtjrdnd|fVd S) z Helper routine for `ogrinspect` that generates GeoDjango models corresponding to the given data source. See the `ogrinspect` docstring for more details. z>Data source parameter must be a string or a DataSource object.cs6t|ttfrdd|DS|r.ddDSgSdS)NcSsg|] }|jqSr!)r)r%r&r!r!r" sz6_ogrinspect..process_kwarg..cSsg|] }|jqSr!)r)r%r&r!r!r"r-s)rlisttuple)kwarg) ogr_fieldsr!r" process_kwargs z"_ogrinspect..process_kwargcsJg}|jkr|jd|jkr0|jd|rBddj|SdSdS)Nz null=Truez blank=Truez, )rappendr() field_namekwlist) blank_fields null_fieldsr!r"get_kwargs_strs    z#_ogrinspect..get_kwargs_strzF# This is an auto-generated Django model module created by ogrinspect.z(from django.contrib.gis.db import modelsr3zclass %s(models.Model):r Nrrz@ %s = models.DecimalField(max_digits=%d, decimal_places=%d%s)z %s = models.FloatField(%s)z %s = models.IntegerField(%s)z# %s = models.BigIntegerField(%s)z* %s = models.CharField(max_length=%s%s)z %s = models.DateField(%s)z! %s = models.DateTimeField(%s)z %s = models.TimeField(%s)zUnknown field type %s in %szsrid=-1izsrid=%sz %s = models.%s(%s)z$ def __%s__(self): return self.%srunicoder)rr rrrrr Z field_widthsZfield_precisionsZ field_typesrrrrrrrr rrZdjangoZsrssridPY3)rZ model_namerrr<rZ name_fieldZimportsdecimalZblanknullZlayerr2Zdecimal_fieldsr9r5widthZ precisionZ field_typerZ kwargs_strr Z geom_fieldZsrid_strr!)r7r8r1r"r)|sx                   r))r rF) r rNFNTFFF)__doc__Zdjango.contrib.gis.gdalrZdjango.contrib.gis.gdal.fieldrrrrrrr Z django.utilsr Zdjango.utils.six.movesr r#r,r)r!r!r!r"s $   &G