a 1$e:@sddlZddlmZddlZddlZddlZddlmZddlm Z eddZ ddd Z dd d Z d dZddZddZddZddZddZdddZdS) N)contextmanager) GeoDataFrame)_compatccsddlm}m}t||rZ|sR||VWdqX1sF0Yq|VnHt||r|}|VWdq1s0Yntd|dS)a Yield a connection within a transaction context. Engine.begin() returns a Connection with an implicit Transaction while Connection.begin() returns the Transaction. This helper will always return a Connection with an implicit (possibly nested) Transaction. Parameters ---------- conn_or_engine : Connection or Engine A sqlalchemy Connection or Engine instance Returns ------- Connection r)Engine ConnectionNzUnknown Connectable: )Zsqlalchemy.engine.baserr isinstanceZin_transactionbegin ValueError)Zconn_or_enginerrconnr L/var/www/html/django/DPS/env/lib/python3.9/site-packages/geopandas/io/sql.py _get_conns  &  &r geomc Cs||vrtd||j|dkr:td|d||}|jstjj }dd}dd}t |j d t rz|}n|}| |||<}|d urtjrt|j d }ntjj|j d j}|d krd |}t|||d S) a` Transforms a pandas DataFrame into a GeoDataFrame. The column 'geom_col' must be a geometry column in WKB representation. To be used to convert df based on pd.read_sql to gdf. Parameters ---------- df : DataFrame pandas DataFrame with geometry column in WKB representation. geom_col : string, default 'geom' column name to convert to shapely geometries crs : pyproj.CRS, optional CRS to use for the returned GeoDataFrame. The value can be anything accepted by :meth:`pyproj.CRS.from_user_input() `, such as an authority string (eg "EPSG:4326") or a WKT string. If not set, tries to determine CRS from the SRID associated with the first geometry in the database, and assigns that to all geometries. Returns ------- GeoDataFrame z"Query missing geometry column '{}'zDuplicate geometry column 'zC' detected in SQL query output. Onlyone geometry column is allowed.cSstjt|S)zLoad from Python 2 binary.shapelywkbloadsstrxr r r load_geom_bufferSsz&_df_to_geodf..load_geom_buffercSstjjt|ddS)z!Load from binary encoded as text.T)hexrrr r r load_geom_textWsz$_df_to_geodf..load_geom_textrNzepsg:{})crsgeometry)r formatcolumnsZto_listcountZdropnaemptyrrrrZiatbytesapplycompatZ SHAPELY_GE_20Zget_sridZgeosZlgeosZ GEOSGetSRIDZ_geomr) dfgeom_colrgeomsZload_geom_bytesrrZ load_geomsridr r r _df_to_geodf.s,   r'Tc s^|dur.tj|||||||d} t| dStj|||||||d} fdd| DSdS)a Returns a GeoDataFrame corresponding to the result of the query string, which must contain a geometry column in WKB representation. It is also possible to use :meth:`~GeoDataFrame.read_file` to read from a database. Especially for file geodatabases like GeoPackage or SpatiaLite this can be easier. Parameters ---------- sql : string SQL query to execute in selecting entries from database, or name of the table to read from the database. con : sqlalchemy.engine.Connection or sqlalchemy.engine.Engine Active connection to the database to query. geom_col : string, default 'geom' column name to convert to shapely geometries crs : dict or str, optional CRS to use for the returned GeoDataFrame; if not set, tries to determine CRS from the SRID associated with the first geometry in the database, and assigns that to all geometries. chunksize : int, default None If specified, return an iterator where chunksize is the number of rows to include in each chunk. See the documentation for pandas.read_sql for further explanation of the following parameters: index_col, coerce_float, parse_dates, params, chunksize Returns ------- GeoDataFrame Examples -------- PostGIS >>> from sqlalchemy import create_engine # doctest: +SKIP >>> db_connection_url = "postgresql://myusername:mypassword@myhost:5432/mydatabase" >>> con = create_engine(db_connection_url) # doctest: +SKIP >>> sql = "SELECT geom, highway FROM roads" >>> df = geopandas.read_postgis(sql, con) # doctest: +SKIP SpatiaLite >>> sql = "SELECT ST_AsBinary(geom) AS geom, highway FROM roads" >>> df = geopandas.read_postgis(sql, con) # doctest: +SKIP N) index_col coerce_float parse_datesparams chunksizer$rc3s|]}t|dVqdS)r-N)r').0r#rr$r r z _read_postgis..)pdZread_sqlr') sqlconr$rr(r)r*r+r,r#Z df_generatorr r/r _read_postgisms*;  r5cOs&ddl}|jdtddt|i|S)Nrzgeopandas.io.sql.read_postgis() is intended for internal use only, and will be deprecated. Use geopandas.read_postgis() instead. stacklevel)warningswarn FutureWarningr5)argskwargsr9r r r read_postgissr>cCst|jj}d}|D]}|dur(qqd|vrd}qt|dkrp|rLd}qt|ddurbtdqt|d}nd }t|jjr|d 7}||fS) a Get basic geometry type of a GeoDataFrame. See more info from: https://geoalchemy-2.readthedocs.io/en/latest/types.html#geoalchemy2.types._GISType Following rules apply: - if geometries all share the same geometry-type, geometries are inserted with the given GeometryType with following types: - Point, LineString, Polygon, MultiPoint, MultiLineString, MultiPolygon, GeometryCollection. - LinearRing geometries will be converted into LineString -objects. - in all other cases, geometries will be inserted with type GEOMETRY: - a mix of Polygons and MultiPolygons in GeoSeries - a mix of Points and LineStrings in GeoSeries - geometry is of type GeometryCollection, such as GeometryCollection([Point, LineStrings]) - if any of the geometries has Z-coordinate, all records will be written with 3D. FN LinearRingTrZ LINESTRINGrz No valid geometries in the data.ZGEOMETRYZ) listr geom_typeuniquelenr upperanyZhas_z)gdfZ geom_types has_curvegtZtarget_geom_typer r r _get_geometry_types"    rJcCsd}d}|jdurzPdD]F}|jj|d}|dur6q`|jjd|d}|durt|d}q`qWn"tytj|tdd Yn0|durd }tj|tdd |S) zA Get EPSG code from CRS if available. If not, return -1. NzNCould not parse CRS from the GeoDataFrame. Inserting data without defined CRS.)dF)min_confidenceZESRI)Z auth_namerNrr6r7)rZto_epsgZ to_authorityint Exceptionr9r: UserWarning)rGr&Z warning_msg confidenceZ auth_sridr r r _get_srid_from_crss*    rTcs@ddlm|jdk}|j||ffdd|j||f<|S)Nr LineStringr?cs|S)Nr )rrUr r (r1z3_convert_linearring_to_linestring..)Zshapely.geometryrVrBlocr!)rG geom_namemaskr rUr !_convert_linearring_to_linestring s    r[cstjr*tjtj||jjdddd}nZtjr`ddlm}m}||||jjdddd}n$ddl m fdd||D}t j |d d }|||<|S) zConvert geometries to ewkb.)r&T)rZ include_sridr)set_sridto_wkb)dumpscsg|]}|ddqS)T)r&rr )r.rr^r&r r Br1z$_convert_to_ewkb..F)copy) r"ZUSE_SHAPELY_20rr]r\values_dataZ USE_PYGEOSZpygeos shapely.wkbr^r2Z DataFrame)rGrYr&r%r\r]r#r r_r _convert_to_ewkb-s$ rec Csddl}ddl}|}||}|||dddd|D}|j} | 4} d |j j |j j |} | j | |dWdn1s0YdS)Nrz, css|]}d|VqdS)z"{}"N)r)r.kr r r r0Ur1z$_psql_insert_copy..z'COPY "{}"."{}" ({}) FROM STDIN WITH CSV)r3file)iocsvStringIOwriter writerowsseekjoin connectioncursorrtableschemanameZ copy_expert) tblr keysZ data_iterrhriZs_bufrkrZ dbapi_conncurr3r r r _psql_insert_copyLs    rwfailFc  Cszddlm} ddlm} Wnty6tdYn0|}|jj} t|} t |\} }|durv| | | d|| <n| | | | di}|rt || }t || | }|dur|}nd}|dkr:t |d}|j |||r|| d j||| d d}|| krd j|| d }t|Wdn1s00Yt |,}|j||||||||td Wdn1sv0YdS)a Upload GeoDataFrame into PostGIS database. This method requires SQLAlchemy and GeoAlchemy2, and a PostgreSQL Python driver (e.g. psycopg2) to be installed. Parameters ---------- name : str Name of the target table. con : sqlalchemy.engine.Connection or sqlalchemy.engine.Engine Active connection to the PostGIS database. if_exists : {'fail', 'replace', 'append'}, default 'fail' How to behave if the table already exists: - fail: Raise a ValueError. - replace: Drop the table before inserting new values. - append: Insert new values to the existing table. schema : string, optional Specify the schema. If None, use default schema: 'public'. index : bool, default True Write DataFrame index as a column. Uses *index_label* as the column name in the table. index_label : string or sequence, default None Column label for index column(s). If None is given (default) and index is True, then the index names are used. chunksize : int, optional Rows will be written in batches of this size at a time. By default, all rows will be written at once. dtype : dict of column name to SQL type, default None Specifying the datatype for columns. The keys should be the column names and the values should be the SQLAlchemy types. Examples -------- >>> from sqlalchemy import create_engine # doctest: +SKIP >>> engine = create_engine("postgresql://myusername:mypassword@myhost:5432/mydatabase";) # doctest: +SKIP >>> gdf.to_postgis("my_table", engine) # doctest: +SKIP r)Geometry)textz,'to_postgis()' requires geoalchemy2 package.N) geometry_typer&publicappendz6SELECT Find_SRID('{schema}', '{table}', '{geom_col}');)rrrqr$zkThe CRS of the target table (EPSG:{epsg_t}) differs from the CRS of current GeoDataFrame (EPSG:{epsg_src}).)Zepsg_tZepsg_src)rr if_existsindex index_labelr,dtypemethod)Z geoalchemy2ryZ sqlalchemyrz ImportErrorrarrsrTrJr[rer dialectZ has_tableexecuterZfetchoner Zto_sqlrw)rGrsr4rrr~rrr,rryrzrYr&r{rHZ schema_nameroZ target_sridmsgr r r _write_postgis_s`6        ( r)rN)rNNTNNN)NrxFNNN)r9 contextlibrZpandasr2rrdZ geopandasrrr"r r'r5r>rJrTr[rerwrr r r r s<     B V ."