v^" @sdZddlmZddlmZmZmZmZmZm Z m Z dddddZ dd Z ddd dd d dddd d Z d 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 OFTStringOFTTimeZgeomFcCst|trt|}nt|tr0n tdi}xK||jD]<}|j}|dddkr|d7}||| 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) isinstancestrr TypeErrorfieldslower geom_typeto_multiupper) data_source geom_name layer_key multi_geom_mappingr mfieldgtyperE/tmp/pip-build-8lau8j11/django/django/contrib/gis/utils/ogrinspect.pymapping s      rcOsdjt||S)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 __str__() method (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: Call the _ogrinspect() helper to do the heavy lifting.  )join _ogrinspect)argskwargsrrr ogrinspect3sDr%NTc #s8t|trt|}nt|tr0n td||} | jfdd} | | | | | |} fdd}|rdVdVdVdVd |Vxt| j| j| jD]\}}}}|j }|dd d kr|d 7}||}|t krx|j | kr\d||||fVqwd||dd fVq|t krd||dd fVq|t krd||dd fVq|t krd|||fVq|tkrd||dd fVq|tkr9d||dd fVq|tkrad||dd fVqtd||fqW| j}|r|j|j}|d kr| jd krd}q| jj}|d krd}q|dkrd}qd|}n d|}d|||fV|r4dVd|Vd 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.csEt|ttfr&dd|DS|r=ddDSgSdS)NcSsg|]}|jqSr)r).0srrr s z6_ogrinspect..process_kwarg..cSsg|]}|jqSr)r)r&r'rrrr(s )rlisttuple)Zkwarg) ogr_fieldsrr process_kwargs z"_ogrinspect..process_kwargcscg}|jkr%|jd|jkrD|jd|r[ddj|SdSdS)Nz null=Truez blank=Truez, )rappendr!) field_namekwlist) blank_fields null_fieldsrrget_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 modelsr-zclass %s(models.Model):r Nr r z@ %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 __str__(self): return self.%sr )rrrrrzipZ field_widthsZfield_precisionsZ field_typesrrrrrrrr rrZdjangoZsrssrid)rZ model_namerrr6rZ name_fieldZimportsdecimalZblanknullZlayerr,Zdecimal_fieldsr3r/widthZ precisionZ field_typerZ kwargs_strrZ geom_fieldZsrid_strr)r1r2r+rr"zsx        +                      r")__doc__Zdjango.contrib.gis.gdalrZdjango.contrib.gis.gdal.fieldrrrrrrr rr%r"rrrrs4& G