U
    8ÄT_Š  ã                   @   sp   d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z
 ddlmZ ddlmZmZ G d	d
„ d
eƒZdS )a·  
 DataSource is a wrapper for the OGR Data Source object, which provides
 an interface for reading vector geometry data from many different file
 formats (including ESRI shapefiles).

 When instantiating a DataSource object, use the filename of a
 GDAL-supported data source.  For example, a SHP file or a
 TIGER/Line file from the government.

 The ds_driver keyword is used internally when a ctypes pointer
 is passed in directly.

 Example:
  ds = DataSource('/home/foo/bar.shp')
  for layer in ds:
      for feature in layer:
          # Getting the geometry for the feature.
          g = feature.geom

          # Getting the 'description' field for the feature.
          desc = feature['description']

          # We can also increment through all of the fields
          #  attached to this feature.
          for field in feature:
              # Get the name of the field (e.g. 'description')
              nm = field.name

              # Get the type (integer) of the field, e.g. 0 => OFTInteger
              t = field.type

              # Returns the value the field; OFTIntegers return ints,
              #  OFTReal returns floats, all else returns string.
              val = field.value
é    )Úbyref)ÚGDALBase)ÚDriver)ÚGDALException)ÚLayer)Úds)Úforce_bytesÚ	force_strc                   @   sP   e Zd ZdZejZddd„Zdd„ Zdd	„ Z	d
d„ Z
edd„ ƒZedd„ ƒZdS )Ú
DataSourcez Wraps an OGR Data Source object.Fúutf-8c                 C   sÆ   |rd| _ nd| _ || _t ¡  t|tƒrrt ¡ }zt t	|ƒ| j t
|ƒ¡}W q  tk
rn   td| ƒ‚Y q X n.t|| jƒr�t|tjƒr�|}ntdt|ƒ ƒ‚|r¶|| _t|ƒ| _ntd| ƒ‚d S )Né   r   z%Could not open the datasource at "%s"z"Invalid data source input type: %szInvalid data source file "%s")Ú_writeÚencodingr   Zensure_registeredÚ
isinstanceÚstrZptr_typeÚcapiZopen_dsr   r   r   ÚtypeÚptrÚdriver)ÚselfZds_inputZ	ds_driverÚwriter   r   © r   úF/tmp/pip-unpacked-wheel-dv63caxw/django/contrib/gis/gdal/datasource.pyÚ__init__6   s$    
zDataSource.__init__c                 C   s    t |tƒrDzt | jt|ƒ¡}W q– tk
r@   td| ƒ‚Y q–X nRt |tƒr†d|  krd| j	k rxn nt 
| j|¡}q–td| ƒ‚ntdt|ƒ ƒ‚t|| ƒS )z@Allows use of the index [] operator to get a layer at the index.z!Invalid OGR layer name given: %s.r   z=Index out of range when accessing layers in a datasource: %s.zInvalid index type: %s)r   r   r   Zget_layer_by_namer   r   r   Ú
IndexErrorÚintÚlayer_countZ	get_layerÚ_ptrÚ	TypeErrorr   r   )r   ÚindexZlayerr   r   r   Ú__getitem__W   s    

zDataSource.__getitem__c                 C   s   | j S )z3Return the number of layers within the data source.)r   ©r   r   r   r   Ú__len__g   s    zDataSource.__len__c                 C   s   d| j | jf S )z2Return OGR GetName and Driver for the Data Source.z%s (%s))Únamer   r!   r   r   r   Ú__str__k   s    zDataSource.__str__c                 C   s   t  | j¡S )z/Return the number of layers in the data source.)r   Zget_layer_countr   r!   r   r   r   r   o   s    zDataSource.layer_countc                 C   s   t  | j¡}t|| jdd�S )z#Return the name of the data source.T)Zstrings_only)r   Zget_ds_namer   r	   r   )r   r#   r   r   r   r#   t   s    zDataSource.nameN)FFr   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   Z
destroy_dsZ
destructorr   r    r"   r$   Úpropertyr   r#   r   r   r   r   r
   2   s   
!
r
   N)r(   Úctypesr   Zdjango.contrib.gis.gdal.baser   Zdjango.contrib.gis.gdal.driverr   Zdjango.contrib.gis.gdal.errorr   Zdjango.contrib.gis.gdal.layerr   Z"django.contrib.gis.gdal.prototypesr   r   Zdjango.utils.encodingr   r	   r
   r   r   r   r   Ú<module>   s   #