o
    [h$                     @   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
    )Path)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t jnt j| _|| _t  t|tt	fr6zt 
t|| jt jB d d d }W n$ ty5   td| w t|| jrEt|tjrE|}ntdt| |r^|| _t |}t|| _d S td| )Nz%Could not open the datasource at "%s"z"Invalid data source input type: %szInvalid data source file "%s")capiZGDAL_OF_UPDATEZGDAL_OF_READONLY_writeencodingr   Zensure_registered
isinstancestrr   Zopen_dsr   ZGDAL_OF_VECTORr   Zptr_typetypeptrZget_dataset_driverdriver)selfZds_inputZ	ds_driverwriter   r   r    r   /home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/django/contrib/gis/gdal/datasource.py__init__8   s2   

zDataSource.__init__c                 C   s   t |trzt| jt|}W n5 ty   td| w t |tr>d|  kr.| j	k r8n nt
| j|}ntd| 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   indexlayerr   r   r   __getitem__]   s    


zDataSource.__getitem__c                 C   s   | j S )z3Return the number of layers within the data source.)r   r   r   r   r   __len__p   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__t   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   x   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#   }   s   zDataSource.nameN)FFr   )__name__
__module____qualname____doc__r   Z
destroy_dsZ
destructorr   r    r"   r$   propertyr   r#   r   r   r   r   r
   3   s    
%
r
   N)r(   pathlibr   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    $