v^X@sddlZddlmZddlmZmZddlmZGdddejZ GdddejZ Gd d d eZ dS) N)gdal) BaseCommand CommandError) get_func_argsc@s%eZdZdZdddZdS)LayerOptionActionzy Custom argparse action for the `ogrinspect` `layer_key` keyword option which may be an integer or a string. Nc CsIyt||jt|Wn%tk rDt||j|YnXdS)N)setattrdestint ValueError)selfparser namespacevalue option_stringrS/tmp/pip-build-8lau8j11/django/django/contrib/gis/management/commands/ogrinspect.py__call__ s zLayerOptionAction.__call__)__name__ __module__ __qualname____doc__rrrrrrs rc@s%eZdZdZdddZdS)ListOptionActionz Custom argparse action for `ogrinspect` keywords that require a string list. If the string is 'True'/'true' then the option value will be a boolean instead. NcCsH|jdkr(t||jdnt||j|jddS)NtrueT,)lowerrrsplit)r r r rrrrrrszListOptionAction.__call__)rrrrrrrrrrs rc@s4eZdZdZdZddZddZdS)CommandzInspects the given OGR-compatible data source (e.g., a shapefile) and outputs a GeoDjango model with the given model name. For example: ./manage.py ogrinspect zipcode.shp ZipcodeFc Cs<|jddd|jddd|jddtdd dd |jd dtdd dd |jd dddd|jddddtdddd|jddddd|jddd|jddddddd|jddtdd dd|jd dd!|jd"dddd#dS)$N data_sourcehelpzPath to the data source. model_namezName of the model to create.z--blankactiondefaultFzUse a comma separated list of OGR field names to add the `blank=True` option to the field definition. Set to `true` to apply to all applicable fields.z --decimalzUse a comma separated list of OGR float fields to generate `DecimalField` instead of the default `FloatField`. Set to `true` to apply to all OGR float fields.z --geom-nameZgeomzDSpecifies the model name for the Geometry Field (defaults to `geom`)z--layerr layer_keyrzThe key for specifying which layer in the OGR data source to use. Defaults to 0 (the first layer). May be an integer or a string identifier for the layer.z --multi-geom store_truez?Treat the geometry in the data source as a geometry collection.z --name-fieldz:Specifies a field name to return for the __str__() method.z --no-imports store_falseZimportszDDo not include `from django.contrib.gis.db import models` statement.z--nullzUse a comma separated list of OGR field names to add the `null=True` option to the field definition. Set to `true` to apply to all applicable fields.z--sridzeThe SRID to use for the Geometry Field. If it can be determined, the SRID of the data source is used.z --mappingz8Generate mapping dictionary for use with `LayerMapping`.) add_argumentrr)r r rrr add_arguments*sF       zCommand.add_argumentsc s|jd|jd}}ytj|}Wn4tjk rh}zt|WYdd}~XnXddlmm}fdd|jD}dd|||D} |d rd |d d |d d |d i} ||| } d d| jD| j ddd|d|j g| j fdd||d j D| j d|d | |d fdgdj | dS)Nrrr) _ogrinspectmappingcs=i|]3\}}|tkr|dk r||qS)N)r).0kv)r'rr os z"Command.handle..cSsg|] }|qSrr)r)srrr qs z"Command.handle..r(Z geom_namer"Z multi_geomcSsi|]\}}||qSrr)r)r*r+rrrr,~s z7# Auto-generated `LayerMapping` dictionary for %s modelz%s_mapping = {c3s#|]}d||fVqdS)z '%s': '%s',Nr)r)Zogr_fld) rev_mappingrr sz!Command.handle..z '%s': '%s',} ) poprZ DataSourceZ GDALExceptionrZ#django.contrib.gis.utils.ogrinspectr'r(itemsextendrfieldsjoin) r argsoptionsrrZdsmsgr(Z ogr_optionsoutputkwargsZ mapping_dictr)r'r0rhandlebs(   )zCommand.handleN)rrrrZrequires_system_checksr&r>rrrrr!s  8r) argparseZdjango.contrib.gisrZdjango.core.management.baserrZdjango.utils.inspectrZActionrrrrrrrs