U ô‰VbÜ6ã@s°dZdZdZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZmZmZddlmZe jdkr‚dd lZdd lmZdd lmZmZd Zd ZdZGdd„dƒZd S)zCopyright 2021, 3Lizz GPL version 3z info@3liz.orgé)Ú lru_cache)ÚTupleÚUnion) ÚQgisÚQgsCoordinateReferenceSystemÚQgsCoordinateTransformÚQgsDataSourceUriÚQgsFeatureRequestÚ QgsGeometryÚ QgsProjectÚQgsProviderConnectionExceptionÚQgsSpatialIndexÚQgsVectorLayer)ÚQVariantéyN)ÚQgsProviderRegistry)ÚLoggerÚ profilingédz1 = 0Úc@s.eZdZd$eeeedœdd„Zedœdd„Zddœd d „Zedœd d „Z e e e d œdd„Z e ee eefdœdd„ƒZe eedeedœdd„ƒƒZe eedeedœdd„ƒƒZe eedeedœdd„ƒƒZe eedeedœdd„ƒƒZeeeee fedœdd „ƒZeeeeeeed!œd"d#„ƒZdS)%ÚFilterByPolygonF)ÚconfigÚlayerÚeditingÚuse_st_relationshipcCsRd|_||_||_||_t ¡|_||_d|_d|_ d|_ d|_ d|_ |  ¡dS)zªConstructor for the filter by polygon. :param config: The filter by polygon configuration as dictionary :param layer: The vector layer to filter N)Ú connectionrrrr ÚinstanceÚprojectrÚ primary_keyÚ filter_modeÚspatial_relationshipÚpolygonÚ group_fieldÚ_parse)Úselfrrrr©r%ú;/var/qgis-server/plugins/lizmap/server/filter_by_polygon.pyÚ__init__(s zFilterByPolygon.__init__)ÚreturncCs |jdk S)z2If the configuration is filtering the given layer.N)r©r$r%r%r&Ú is_filteredHszFilterByPolygon.is_filteredNcCs²|j ¡sdS|jdkrdS|j d¡}|s0dS|D]@}| d¡|j ¡kr4| d¡|_| d¡|_| d¡|_qvq4|jdkr„dS|j d¡}|j  |d¡|_ | d ¡|_ dS) z)Read the configuration and fill variablesNÚlayersrrrr rZpolygon_layer_idr") rÚ isSpatialrÚgetÚidrrr rÚmapLayerr!r")r$r+rrr%r%r&r#Ls$        zFilterByPolygon._parsecCsÂ|jst d¡dS|js2|j ¡r2t d¡dS|j ¡ |j¡dkrft d |j|j ¡¡¡dS|j  ¡sŽt d |j |j  ¡¡¡dS|j  ¡ |j ¡dkr¾t d |j |j  ¡¡¡dS)z& If the configuration is valid or not.z-The polygon layer for filtering is not valid.Frz4The field {} used for filtering does not exist in {}T) r!rÚcriticalÚisValidÚfieldsÚindexOfr"ÚformatÚnamerrr)r%r%r&Úis_validgs<  ÿÿ ÿÿÿÿzFilterByPolygon.is_valid)Úurir(c Csx|jdkrht ¡ d¡}| | ¡i¡|_z|j d¡Wn,tk rf}zt  |¡W5d}~XYnX|j |¡}|S)z> For a given URI, execute an SQL query and return the result. NÚpostgresz>SET application_name='QGIS Lizmap Server : Filter By Polygon';) rrrZproviderMetadataÚcreateConnectionr7Ú executeSqlr rÚ log_exception)r$r7ÚsqlÚmetadataÚeÚresultsr%r%r&Ú sql_query„s  zFilterByPolygon.sql_query)Úgroupsr(c Cs>|jdkr,|js"t d¡tdfSt d¡|j ¡dkrPtjdkrP|  |¡}n |  |¡}|  ¡rpd|_ t dfSdj|j ¡ ¡| |j ¡ ¡r–d nd ¡d }|j ¡dkr,|jsÂtjdkr,t|j ¡ƒ}|jd krÞd nd}| |j ¡|j ¡| ¡||¡}|jr||fS| |¡|f}d|_ |S| |¡}||fS)zµ Get the SQL subset string for the current groups of the user. :param groups: List of groups belongings to the user. :returns: The subset SQL string to use rzPLayer is editing only BUT we are not in an editing session. Return all features.rzZLayer is editing only AND we are in an editing session. Continue to find the subset stringr8rNzSRID={crs};{wkt}éé)ÚcrsÚwktÚcontainsFT)rrrÚinfoÚ ALL_FEATURESr!Ú providerTyperÚQGIS_VERSION_INTÚ"_polygon_for_groups_with_sql_queryÚ!_polygon_for_groups_with_qgis_apiÚisEmptyrÚ NO_FEATURESr4rDÚ postgisSridÚasWktÚ isGeographicrrrÚsourcer Ú_format_sql_st_relationshipÚ sourceCrsÚgeometryColumnÚ_features_ids_with_sql_queryÚ_features_ids_with_qgis_api) r$rAr!Zewktr7Úuse_st_intersectZ st_relationÚresultZsubsetr%r%r&Ú subset_sql’sJ ÿÿ   þû zFilterByPolygon.subset_sql)ÚmaxsizecCs`dj|jd |¡d}tƒ}| g¡| |¡g}|j |¡D]}| |  ¡¡q@t ƒ  |¡S)zF All features from the polygon layer corresponding to the user groups zÅ array_intersect( array_foreach( string_to_array("{polygon_field}"), trim(@element) ), array_foreach( string_to_array('{groups}'), trim(@element) ) )ú,)Ú polygon_fieldrA) r4r"Újoinr ÚsetSubsetOfAttributesÚsetFilterExpressionr!Ú getFeaturesÚappendÚgeometryr ÚcollectGeometry)r$rAÚ expressionÚrequestZ polygon_geomsÚfeaturer%r%r&rLÍs ô  z1FilterByPolygon._polygon_for_groups_with_qgis_apic Csèt|j ¡ƒ}zdj|jd |¡| ¡| ¡| ¡d}t   d |¡¡|  ||¡}|dd}t ƒ}t |tƒr~| ¡r~|WS|dd…}| t |¡¡|WStk râ}z&t  |¡t  d ¡| |¡WY¢Sd}~XYnXdS) z… All features from the polygon layer corresponding to the user groups for a Postgresql layer. Only for QGIS >= 3.10 a] WITH current_groups AS ( SELECT ARRAY_REMOVE( STRING_TO_ARRAY( regexp_replace( '{groups}', '[^a-zA-Z0-9_-]', ',', 'g' ), ',' ), '') AS user_group ) SELECT 1 AS id, ST_AsBinary(ST_Union("{geom}")) AS geom FROM "{schema}"."{table}" AS p, current_groups AS c WHERE c.user_group && ( ARRAY_REMOVE( STRING_TO_ARRAY( regexp_replace( "{polygon_field}", '[^a-zA-Z0-9_-]', ',', 'g' ), ',' ), '') ) r\)r]rAÚgeomÚschemaÚtablezHRequesting the database about polygons for the current groups with : {}rérCNzkThe filter_by_polygon._polygon_for_groups_with_sql_query failed when requesting PostGIS. Using the QGIS API)rr!rRr4r"r^rUrirjrrGr@r Ú isinstancerÚisNullÚfromWkbÚbinasciiZ unhexlifyÚ Exceptionr;r0rL)r$rAr7r<r?Zwkbrhr>r%r%r&rKës4 Ü&ÿ    ÿz2FilterByPolygon._polygon_for_groups_with_sql_query)Úpolygonsr(cCsÌtƒ}| |j ¡¡t|j ¡|j ¡|jƒ}| |¡|  |  ¡¡}g}|D]l}|j  |¡}|j dkrŠ|  ¡ |¡r¼| ||j¡qP|j dkr´|  ¡  |¡r¼| ||j¡qPtdƒ‚qP| |j|¡S)zY List all features using the QGIS API. :returns: The subset SQL string. rFÚ intersectszSpatial relationship unknown)r Ú addFeaturesrrarr!rDrÚ transformrrÚ boundingBoxÚ getFeaturer rcrFrbrrpÚ_format_sql_in)r$rqÚindexrtÚ candidatesÚ unique_idsZ candidate_idrgr%r%r&rW1s      z+FilterByPolygon._features_ids_with_qgis_api)Ú st_intersectr(cCsjt|j ¡ƒ}dj|j| ¡| ¡|d}t d |dd…¡¡|  ||¡}dd„|Dƒ}|  |j|¡S)zw List all features using a SQL query. Only for QGIS >= 3.10 :returns: The subset SQL string. z6SELECT {pk} FROM {schema}.{table} WHERE {st_intersect})Úpkrirjr{z6Requesting the database about IDs to filter with {}...réZcSsg|]}t|dƒ‘qS)r)Ústr)Ú.0Úrowr%r%r&Ú hsz@FilterByPolygon._features_ids_with_sql_query..) rrrRr4rrirjrrGr@rw)r$r{r7r<r?rzr%r%r&rVSs üÿ z,FilterByPolygon._features_ids_with_sql_query)rÚvaluesr(cCsT|stSg}|D].}t|tƒr0| d |¡¡q| t|ƒ¡qdj|d |¡dS)zFormat the SQL IN statement.z'{}'z"{pk}" IN ( {values} )z , )r|r‚)rNrlr~rbr4r^)Úclsrr‚ÚcleanedÚvaluer%r%r&rwls zFilterByPolygon._format_sql_in)Ú filtered_crsÚ filtering_crsÚ geom_fieldrqrXr(cCs:dj|r dnd|| | ¡r dnd¡| ¡| ¡d}|S)zzIf layer is of type PostgreSQL, use a simple ST_Intersects/ST_Contains. :returns: The subset SQL string. zc {function}( ST_Transform(ST_GeomFromText('{wkt}', {from_crs}), {to_crs}), "{geom_field}" )Z ST_IntersectsZ ST_ContainsrBrC)ÚfunctionrˆrEZfrom_crsZto_crs)r4rPrQrO)rƒr†r‡rˆrqrXr<r%r%r&rS{s  ÷ z+FilterByPolygon._format_sql_st_relationship)FF)Ú__name__Ú __module__Ú __qualname__ÚdictrÚboolr'r*r#r6rrr@rÚtupler~rZrÚCACHE_MAX_SIZEr rLrKrWrVÚ classmethodrÚlistrwrrSr%r%r%r&r&sHÿÿ :D ùr)Ú __copyright__Ú __license__Ú __email__Ú functoolsrÚtypingrrÚ qgis.corerrrrr r r r r rZqgis.PyQt.QtCorerrJrorÚlizmap.server.loggerrrrrNrHrr%r%r%r&Ús 0