U Yh"@sdZddlZddlZddlmZddlmZddlmZm Z ddl m Z m Z m Z ddlmZdd lmZmZmZdd lmZdd lmZmZd Zd ZdZdeeefe eddddZGdddeZ Gddde Z!dS)u *************************************************************************** QGIS Server Plugin Filters: Add a new request to print a specific atlas feature --------------------- Date : December 2019 Copyright : (C) 2019 by René-Luc D'Hont - 3Liz Email : rldhont at 3liz dot com *************************************************************************** * * * This program is free software; you can redistribute it and/or modify * * it under the terms of the GNU General Public License as published by * * the Free Software Foundation; either version 2 of the License, or * * (at your option) any later version. * * * *************************************************************************** N)Path)Dict) QgsExpression QgsProject)QgsServerRequestQgsServerResponse QgsService)pluginMetadata)AtlasPrintExceptionparse_output_format print_layout)Logger)get_lizmap_groupsget_lizmap_user_loginzCopyright 2021, 3Lizz GPL version 3z info@3liz.org)dataresponsecodereturncCs*|||dd|t|dS)z! Write data as json response Content-Typezapplication/jsonN) setStatusCode setHeaderwritejsondumps)rrrr4/var/www/lizmap/lizmap/plugins/atlasprint/service.pywrite_json_response&s  rcs6eZdZeeddfdd ZeddddZZS)AtlasPrintErrorN)rmsgrcs0t|||_||_td||dS)Nz Atlas print request error {}: {})super__init__r rrcriticalformat)selfrr  __class__rrr"0s zAtlasPrintError.__init__)rrcCs&d|jd}|t|||jdS)z Format error response Zfail)statusmessageN)r clearrr)r%rbodyrrrformatResponse6s zAtlasPrintError.formatResponse) __name__ __module__ __qualname__intstrr"rr, __classcell__rrr&rr.srcseZdZdeddfdd ZedddZedd d Zej ed d d Z ee e ddddZ eeeefe e ddddZeeefe e eeddddZZS)AtlasPrintServiceFN)debugrcst|}t|_dS)N)r!r"rlogger)r%r4_r&rrr"@s zAtlasPrintService.__init__)rcCsdS)z Service name ATLASrr%rrrnameGszAtlasPrintService.namecCsdS)z Service version z1.0.0rr8rrrversionLszAtlasPrintService.version)methodrcCs|tjtjfkS)z& Check supported HTTP methods )r GetMethod PostMethod)r%r;rrr allowMethodRszAtlasPrintService.allowMethod)requestrprojectrc Csn|}z&z|dd}|dkr6||||n|dkr| }t ||}t ||} |}|dd|kr|jd ||d<t | |d<||| ||||| ntd d |Wnjtk r} z| |W5d} ~ XYn@tk r,|jd ttd d} | |YnXW5|jd|}|dd|dd||XdS)z# Execute a 'ATLAS' request z7Removing user and group variables from the QGIS project lizmap_userNlizmap_user_groupsREQUESTZgetcapabilitiesZgetprintz3Adding user and group variables in the QGIS projectzOInvalid REQUEST parameter: must be one of GetCapabilities, GetPrint, found '{}'Unhandled exception: {}#Internal 'atlasprint' service error) parametersr5infocustomVariablespopsetCustomVariablesgetlowerget_capabilitiesheadersrrlist get_printrr$r, Exceptionr# traceback format_exc) r%r?rr@paramsZ custom_varZ request_paramrQrAZ lizmap_grouperrrrrexecuteRequestXsB         z AtlasPrintService.executeRequest)rWrr@rcCs0||f}d}d|t|ddd}t||dS)z7 Get atlas capabilities based on metadata file atlasprintZsuccessr:)r9r:)r(metadataN)r r)rWrr@r6Z plugin_namer+rrrrPs z"AtlasPrintService.get_capabilities)rWrr@rAlizmap_user_grouprc Cs.|d}|dd}|d}|d} t|d|d} z|sNtd|rtt|} | rttd | |r| rtd |rz t|}Wntk rtd YnX| rzd d | dD} Wntk rtdYnXdd| D} |r|| d<d || d<t f||d| || |d| } Wnhtk rp}zt dd|W5d}~XYn6tk r|jdtt ddYnXt| }|st dd| j|d| j|dz|||Wn,tk r(|jd| jYnXdS)z Get print document TEMPLATE EXP_FILTERNSCALESCALESFORMATr$zTEMPLATE is requiredzExpression is invalid: {}z*SCALE and SCALES can not be used together.zInvalid number in SCALE.cSsg|] }t|qSr)r0).0scalerrr sz/AtlasPrintService.get_print..,zInvalid number in SCALES.cSs"i|]\}}|dkr||qS))r]r^r_r`raMAPrCSERVICEZDPIZ EXCEPTIONSZLAYERZLIZMAP_OVERRIDE_FILTERZ TRANSPARENTVERSION LIZMAP_USERLIZMAP_USER_GROUPS)upper)rbkvrrr s z/AtlasPrintService.get_print..rArB)r@Z layout_name output_formatrcscalesfeature_filterrEz8ATLAS - Error from the user while generating the PDF: {}rFrGrHizATLAS {} not foundrrz$Error occurred while reading {} file)rNr r rZhasParserErrorr$ZparserErrorStringr0 ValueErrorsplititemsjoinr rrTr5r#rUrVrexistsr9rvaluerr read_bytesunlink)r%rWrr@rAr\templaterqrcrpro expressionZadditional_paramsZ output_pathepathrrrrSst           zAtlasPrintService.get_print)F)r-r.r/boolr"r1r9r:rMethodr>rrrY staticmethodrrPtuplerSr2rrr&rr3>s" 0 r3)r)"__doc__rrUpathlibrtypingr qgis.corerr qgis.serverrrr qgis.utilsr corer r r r5rtoolsrr __copyright__ __license__ __email__r1r0rrTrr3rrrrs