o
    [h#                     @   sf   d Z ddlmZ ddlmZmZmZmZmZm	Z	m
Z
 dddZdd	 Z			
		
				dd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OFTInteger64OFTReal	OFTStringOFTTimegeomFc                 C   s   t | tr
t| } n
t | trntdi }| | jD ]}| }|dd dkr-|d7 }|||< q| | j}|r=|  t| ||< |S )a  
    Given a DataSource, generate 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.
    >Data source parameter must be a string or a DataSource object.N_field)	
isinstancestrr   	TypeErrorfieldslower	geom_typeto_multiupper)data_source	geom_name	layer_key
multi_geom_mappingr   mfieldgtype r   /home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/django/contrib/gis/utils/ogrinspect.pymapping   s$   




r    c                  O   s   d t| i |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)argskwargsr   r   r   
ogrinspect;   s   Dr&   NTc                 #   s~   t | trt| } n
t | trntd| | }|jfdd}||
||	 ||} fdd}|rCdV  dV  dV  dV  d	| V  t|j|j|jD ]\}}}}|	 }|d
d dkrh|d7 }||}|t
u r|	 |v rd||||f V  qRd||dd f V  qR|tu rd||dd f V  qR|tu rd||dd f V  qR|tu rd|||f V  qR|tu rd||dd f V  qR|tu rd||dd f V  qR|tu rd||dd f V  qRtd||f |j}|r|  |j}|du r$|jdu rd}n|jj}|du rd}n|dkrd}n	d| }nd| }d|||f V  |r=dV  d| V  dS dS )z
    Helper routine for `ogrinspect` that generates GeoDjango models corresponding
    to the given data source.  See the `ogrinspect` docstring for more details.
    r   c                    s2   t | ttfrdd | D S | rdd  D S g S )Nc                 S      g | ]}|  qS r   r   .0sr   r   r   
<listcomp>       z6_ogrinspect.<locals>.process_kwarg.<locals>.<listcomp>c                 S   r'   r   r(   r)   r   r   r   r,      r-   )r   listtuple)kwarg)
ogr_fieldsr   r   process_kwarg   s
   z"_ogrinspect.<locals>.process_kwargc                    sF   g }|   v r|d |    v r|d |r!dd| S dS )Nz	null=Truez
blank=Truez,  )r   appendr"   )
field_namekwlist)blank_fieldsnull_fieldsr   r   get_kwargs_str   s   

z#_ogrinspect.<locals>.get_kwargs_strzF# This is an auto-generated Django model module created by ogrinspect.z(from django.contrib.gis.db import modelsr3   zclass %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)zUnknown field type %s in %szsrid=-1i  zsrid=%sz    %s = models.%s(%s)z%    def __str__(self): return self.%s)r   r   r   r   r   zipZfield_widthsZfield_precisionsZfield_typesr   r   r   r   r   r   r   r	   r   r   djangoZsrssrid)r   Z
model_namer   r   r=   r   Z
name_fieldZimportsdecimalblanknulllayerr2   Zdecimal_fieldsr9   r5   width	precisionZ
field_typer   Z
kwargs_strr   Z
geom_fieldZsrid_strr   )r7   r8   r1   r   r#      s   



	




r#   )r
   r   F)	r
   r   NFNTFFF)__doc__Zdjango.contrib.gis.gdalr   Zdjango.contrib.gis.gdal.fieldr   r   r   r   r   r   r	   r    r&   r#   r   r   r   r   <module>   s    $
(J