U gO@shddlZddlmZddlmZmZddlmZGdddejZ GdddejZ Gd d d eZ dS) N)gdal) BaseCommand CommandError get_func_argsc@seZdZdZdddZdS)LayerOptionActionzy Custom argparse action for the `ogrinspect` `layer_key` keyword option which may be an integer or a string. NcCs>zt||jt|Wn"tk r8t||j|YnXdSN)setattrdestint ValueErrorselfparser namespacevalue option_stringrU/tmp/pip-unpacked-wheel-ye1bytdm/django/contrib/gis/management/commands/ogrinspect.py__call__ szLayerOptionAction.__call__)N__name__ __module__ __qualname____doc__rrrrrrsrc@seZdZdZdddZdS)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. NcCs4|dkrt||jdnt||j|ddS)NtrueT,)lowerr r splitr rrrrs zListOptionAction.__call__)Nrrrrrrsrc@s$eZdZdZgZddZddZdS)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 ZipcodecCs|jddd|jddd|jdtddd |jd tdd d |jd d dd|jddtddd|jdddd|jddd|jddddd|jd tdd!d |jd"d#d|jd$dd%ddS)&N data_sourcezPath to the data source.)help model_namezName of the model to create.z--blankFzUse 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.)actiondefaultr"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`))r%r"z--layer 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.)r r$r%r"z --multi-geom store_truez?Treat the geometry in the data source as a geometry collection.)r$r"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.)r$r r"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)rrrrr add_arguments*spzCommand.add_argumentsc s<|d|d}}zt|}Wn,tjk rP}z t|W5d}~XYnXddlmm}fdd|D}dd||f|D} |d r2|d |d |d d } ||f| } dd| D| ddd|d| g| fdd||d j D| d|d | |d fdgd | S)Nr!r#r) _ogrinspectmappingcs*i|]"\}}|tkr|dk r||qSrr.0kv)r+rr ps  z"Command.handle..cSsg|]}|qSrr)r.srrr rsz"Command.handle..r, geom_namer& multi_geom)r4r&r5cSsi|]\}}||qSrrr-rrrr1sz7# Auto-generated `LayerMapping` dictionary for %s modelz%s_mapping = {c3s|]}d||fVqdS) '%s': '%s',Nr)r.Zogr_fld) rev_mappingrr s z!Command.handle..r7} ) poprZ DataSourceZ GDALExceptionrZ#django.contrib.gis.utils.ogrinspectr+r,itemsextendrfieldsjoin) rargsoptionsr!r#Zdsmsgr,Z ogr_optionsoutputkwargsZ mapping_dictr)r+r8rhandlebs.    "zCommand.handleN)rrrr"Zrequires_system_checksr*rFrrrrr !s 8r ) argparseZdjango.contrib.gisrZdjango.core.management.baserrZdjango.utils.inspectrActionrrr rrrrs