|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/django/contrib/gis/utils/__pycache__/ |
Upload File : |
o
�al � @ s d Z ddlZddlmZmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZ ddlmZmZmZmZmZmZmZ ddlmZmZ dd l m Z m Z m Z m Z dd
l!m"Z" G d d � d e#�Z$G d
d� de$�Z%G dd� de$�Z&G dd� de$�Z'G dd� de$�Z(G dd� d�Z)dS )a
The LayerMapping class provides a way to map the contents of OGR
vector files (e.g. SHP files) to Geographic-enabled Django models.
For more information, please consult the GeoDjango documentation:
https://docs.djangoproject.com/en/dev/ref/contrib/gis/layermapping/
� N)�Decimal�InvalidOperation)�Path)�
GeometryField)�CoordTransform�
DataSource�
GDALException� OGRGeometry� OGRGeomType�SpatialReference)�OFTDate� OFTDateTime�
OFTInteger� OFTInteger64�OFTReal� OFTString�OFTTime)�FieldDoesNotExist�ObjectDoesNotExist)� connections�models�router� transaction)� force_strc @ � e Zd ZdS )�
LayerMapErrorN��__name__�
__module__� __qualname__� r r �G/usr/lib/python3/dist-packages/django/contrib/gis/utils/layermapping.pyr � r c @ r )�
InvalidStringNr r r r r! r# r" r# c @ r )�InvalidDecimalNr r r r r! r$ $ r" r$ c @ r )�InvalidIntegerNr r r r r! r% ( r" r% c @ r )�MissingForeignKeyNr r r r r! r& , r" r& c
@ s� e Zd ZdZded�ded�ded�ed�jed �ed
�jed �ed �jed
�iZi eje �ej
e �ej e �ej
e eef�eje eef�eje ef�eje�eje�eje�eje�eje ef�eje�eje�ej e�ej e�ej e�ej e eef�ej e eefej!e eefej"e eefej#e eefi�Z$ d0dd�Z%dd� Z&dd� Z'dd� Z(dd � Z)d d � Z*d d � Z+d!d"� Z,d#d$� Z-d%d&� Z.d'd(� Z/d)d*� Z0d+d,� Z1d-d-d-d-d-e2j3d-fd.d/�Z4dS )1� LayerMappingz1A class that maps OGR Layers to GeoDjango Models.� �
MultiPoint� �MultiLineString� � MultiPolygon�Point25D�
MultiPoint25D�
LineString25D�MultiLineString25D�
Polygon25D�MultiPolygon25Dr N�utf-8�commit_on_successTc C s( t |ttf�rt||d�| _n|| _| j| | _|
dur |
nt�|�| _t | j j
| _ || _ || _
| �� t | j jjrB| �� | _nd}|rR| �|�| _| �� | _n|| _|reddlm} | |� || _nd| _| ru| �| � d}| | _nd| _|| _ |dkr�d| _ dS |dkr�t j | _ dS t d| ��) a
A LayerMapping object is initialized using the given Model (not an instance),
a DataSource (or string path to an OGR-supported data file), and a mapping
dictionary. See the module level docstring for more details and keyword
argument usage.
)�encodingNFr )�lookup�
autocommitr5 z!Unrecognized transaction mode: %s)!�
isinstance�strr r �ds�layerr � db_for_write�usingr �ops�spatial_backend�mapping�model� check_layer�features�supports_transform�geometry_field� geo_field� check_srs�
source_srs�coord_transform� transform�codecsr7 r6 � check_unique�unique�transaction_mode�transaction_decoratorr �atomicr ) �selfrB �datarA r< rI r6 rO rK rN r> r7 r r r! �__init__V s@
zLayerMapping.__init__c C s0 |rt |ttf�r
t|� S t |t�r|S t�dS )z Check the `fid_range` keyword.N)r9 �tuple�list�slice� TypeError)rR � fid_ranger r r! �check_fid_range� s
z LayerMapping.check_fid_rangec
s� d| _ i | _| jj� | jj}� fdd�}| j�� D ]�\}}z | jj�|�}W n
t y3 t
d| ��w |j j }t
|t�r�| j rDt
d��|j}z|dkrSt|d �}nt|�}W n
tye t
d| ��w | jj} | j�|j�s�| �| |�s�t
d ||dkr�d
nd | f ��|| _ || _|}
net
|tj�r�t
|t�r�|jj} |�� D ]!\} }
||
�}z| j�| � W q� t y� t
d | | j j f ��w | }
n-td
��|j | j vr�t
d| ��||�}|| }
t |
| j |j �s�t
d|
|
j |f ��|}
|
| j|< qdS )z�
Check the Layer metadata and ensure that it's compatible with the
mapping information and model. Unlike previous revisions, there is no
need to increment through each feature in the Layer.
Fc s, z� � | �}W |S ty td| ��w )Nz4Given mapping OGR field "%s" not found in OGR Layer.)�index�
ValueErrorr )� ogr_map_fld�idx��
ogr_fieldsr r! �
check_ogr_fld� s � �z/LayerMapping.check_layer.<locals>.check_ogr_fldz3Given mapping field "%s" not in given Model fields.zDLayerMapping does not support more than one GeometryField per model.r, �25Dz'Invalid mapping for GeometryField "%s".zDInvalid mapping geometry; model has %s%s, layer geometry type is %s.z(dim=3)� z/ForeignKey mapping field "%s" not in %s fields.z.ForeignKey mapping must be of dictionary type.z0Django field type "%s" has no OGR mapping (yet).z:OGR field "%s" (of type %s) cannot be mapped to Django %s.N) �
geom_field�fieldsr< � field_typesrA �itemsrB �_meta� get_fieldr r � __class__r r9 r �dimr
r � geom_type�name�
startswith�
make_multi� coord_dimr �
ForeignKey�dict� remote_fieldrX � FIELD_TYPES�
issubclass)rR �ogr_field_typesra �
field_name�ogr_name� model_field�fld_namerp �gtype�ltype�
fields_val� rel_model�rel_name� ogr_fieldr^ r r_ r! rC � sr �
� ��
��
� �zLayerMapping.check_layerc C sX t |t�r|}n t || j�� �r|j}nt |ttf�r t|�}n| jj}|s*td��|S )z>Check the compatibility of the given spatial reference object.z#No source reference system defined.) r9 r r@ �spatial_ref_sys�srs�intr: r< r )rR rI �srr r r! rH s
zLayerMapping.check_srsc C sN t |ttf�r|D ] }|| jvrt�q dS t |t�r#|| jvr!t�dS td��)zDCheck the `unique` keyword parameter -- may be a sequence or string.zBUnique keyword argument must be set with a tuple, list, or string.N)r9 rV rU rA r\ r: rX )rR rN �attrr r r! rM s
��
�zLayerMapping.check_uniquec C s� i }| j �� D ]>\}}| j| }t|t�r*z | �|j|�}W n" ty) td��w t|t j
j �r9| � |||�}n| �
|| |�}|||< q|S )z{
Given an OGR Feature, return a dictionary of keyword arguments for
constructing the mapped model.
z)Could not retrieve geometry from feature.)rA rg re r9 r � verify_geom�geomr r r �base� ModelBase� verify_fk�verify_ogr_field)rR �feat�kwargsrw rx ry �valr r r! �feature_kwargs, s
�
zLayerMapping.feature_kwargsc s0 t | jt�r| j� | j iS � fdd�| jD �S )z�
Given the feature keyword arguments (from `feature_kwargs`), construct
and return the uniqueness keyword arguments -- a subset of the feature
kwargs.
c s i | ]}|� | �qS r r )�.0�fld�r� r r! �
<dictcomp>V s z.LayerMapping.unique_kwargs.<locals>.<dictcomp>)r9 rN r: )rR r� r r� r! �
unique_kwargsM s zLayerMapping.unique_kwargsc
C s� t |t�r>t |tjtjf�r>| jr |jdur t|j| j�}n|j}|jr<|dur<t |�|jkr<t
d|j |jt |�f ��|S t |t �r�t |tj
�r�z tt|j��}W n ty` td|j ��w |�� }|d }|d }|j|j }|dk r�t |d|� �} nt |�| } | |kr�td|j|j|f ��|}|S t |t tf�r�t |tj�r�zt|j�}W |S ty� td|j ��w |j}|S ) z�
Verify if the OGR Field contents are acceptable to the model field. If
they are, return the verified value, otherwise raise an exception.
Nz@%s model field maximum string length is %s, given %s characters.z$Could not construct decimal from: %sr( r* r zeA DecimalField with max_digits %d, decimal_places %d must round to an absolute value less than 10^%d.z$Could not construct integer from: %s)r9 r r � CharField� TextFieldr6 �valuer �
max_length�lenr# rm r � DecimalFieldr r: �DecimalInvalidOperationr$ �as_tuple�
max_digits�decimal_places� IntegerFieldr� r\ r% )
rR r� ry r� �d�dtup�digits�d_idx�max_prec�n_precr r r! r� Y sR
� �)� � �� � ��z LayerMapping.verify_ogr_fieldc C sp i }|� � D ]\}}| �|| |j�|��||< qz|j�| j�jdi |��W S ty7 td|j |f ��w )z�
Given an OGR Feature, the related model and its dictionary mapping,
retrieve the related model for the ForeignKey mapping.
z7No ForeignKey %s model found with keyword arguments: %sNr )
rg r� rh ri �objectsr> �getr r& r )rR r� r~ � rel_mapping� fk_kwargsrw rx r r r! r� � s
���zLayerMapping.verify_fkc C s` | j |j kr
| j |_ | �|j|�r"| j|jj }t|�}|�|� n|}| jr-|�| j� |jS )z�
Verify the geometry -- construct and return a GeometryCollection
if necessary (for example if the model field is MultiPolygonField while
the mapped shapefile only contains Polygons).
) rp ro rl � MULTI_TYPES�numr �addrK �wkt)rR r� ry �
multi_type�gr r r! r� � s zLayerMapping.verify_geomc
C sX | j �� }z|j�| j�j| jjd�j}t| j |�W S t
y+ } zt d�|�d}~ww )z,Return the coordinate transformation object.)�sridz?Could not translate between the data source and model geometry.N) r@ r� r� r> r� rG r� r� r rI � Exceptionr )rR �
SpatialRefSys�
target_srs�excr r r! rJ � s
����z LayerMapping.coord_transformc C s | j j}|�| j�S )zHReturn the GeometryField instance associated with the geographic column.)rB rh ri rd )rR �optsr r r! rF � s zLayerMapping.geometry_fieldc C s |j | jv o|jjd|j kS )z�
Given the OGRGeomType for a geometry and its associated GeometryField,
determine whether the geometry should be turned into a GeometryCollection.
zMulti%s)r� r� rj r �django)rR rl ry r r r! ro � s �zLayerMapping.make_multiFc s" �� |�}� r� du st� t�sd�n� �|ddf� ������fdd� } �jdur/��| �} �jj}
|r�t|t�r�||
k r�|rDtd��d\} } }
t||
|�}t|�}t |�D ]2\}}|d |krgt
| d�}nt
| |�}z | || |
�\} }
|} W qW t y� �� d
d |f � � w dS | � dS ) a�
Save the contents from the OGR DataSource Layer into the database
according to the mapping dictionary given at initialization.
Keyword Parameters:
verbose:
If set, information will be printed subsequent to each model save
executed on the database.
fid_range:
May be set with a slice or tuple of (begin, end) feature ID's to map
from the data source. In other words, this keyword enables the user
to selectively import a subset range of features in the geographic
data source.
step:
If set with an integer, transactions will occur at every step
interval. For example, if step=1000, a commit would occur after
the 1,000th feature, the 2,000th feature etc.
progress:
When this keyword is set, status information will be printed giving
the number of features processed and successfully saved. By default,
progress information will pe printed every 1000 features processed,
however, this default may be overridden by setting this keyword with an
integer for the desired interval.
stream:
Status information will be written to this file handle. Defaults to
using `sys.stdout`, but any object with a `write` method is supported.
silent:
By default, non-fatal error notifications are printed to stdout, but
this keyword may be set to disable these notifications.
strict:
Execution of the model mapping will cease upon the first error
encountered. The default behavior is to attempt to continue.
Ti� r c s | r�j | }n�j }|D ]�}|d7 }z��|�}W n! ty; } z�r%� �s1��d|j|f � W Y d }~n�d }~ww d}�jr�zE��|�}�jj� �j �j
di |��} d}t | �j �}
|
d u rit
|�j �} n|
j} t
|�j �} | D ]}
| �|
� qut| �j | j� W n ty� �jdi |��} Y n w �jdi |��} z | j�j d� |d7 }�r���d|r�dnd| f � W n1 ty� } z%�r؈s�d |j � ��d
| � � �s��d ||f � W Y d }~nd }~ww � r�|� d kr���d
||f � q
||fS )Nr( z#Ignoring Feature ID %s because: %s
FT)r> z%s: %s
�Updated�SavedzOFailed to save the feature (id: %s) into the model with the keyword arguments:
z%s
z"Failed to save %s:
%s
Continuing
r z$Processed %d features, saved %d ...
r )r< r� r �write�fidrN r� rB r� r> r� �getattrrd r �ogrr� �setattrr� r �saver� )�
feat_range�num_feat� num_saved�
layer_iterr� r� �msg� is_update�u_kwargs�m�
geom_valuer� �newr� ��progress�progress_intervalrR �silent�stream�strict�verboser r! �_save sp ��
�������z LayerMapping.save.<locals>._saveNzOThe `step` keyword may not be used in conjunction with the `fid_range` keyword.)r r r r( z %s
Failed to save slice: %s
z(=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-)
rZ r9 r� rP r< r� r �ranger� � enumeraterW r� r� )rR r� rY �stepr� r� r� r� �
default_ranger� �nfeat�begr� r� �indices�n_i�i�end�
step_slicer r� r! r� � s8
*
L
��
zLayerMapping.save)r Nr4 r5 TNN)5r r r �__doc__r
r� r� r � AutoFieldr � BigAutoFieldr �SmallAutoField� BooleanFieldr r r� �
FloatField� DateFieldr �
DateTimeFieldr
�
EmailField� TimeFieldr r� r� � SlugFieldr� �URLField� UUIDField�BigIntegerField�SmallIntegerField�PositiveBigIntegerField�PositiveIntegerField�PositiveSmallIntegerFieldrt rT rZ rC rH rM r� r� r� r� r� rJ rF ro �sys�stdoutr� r r r r! r' 0 s� �
��� � �
��� �
�
� �
���� � �
�H a! 9
�r' )*r� r� �decimalr r r� �pathlibr � django.contrib.gis.db.modelsr �django.contrib.gis.gdalr r r r r
r � django.contrib.gis.gdal.fieldr r
r r r r r �django.core.exceptionsr r � django.dbr r r r �django.utils.encodingr r� r r# r$ r% r& r' r r r r! �<module> s $