a 1$e! @sdZddlZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmZdd lmZdd lmZmZee ed d d Zeee fe edddZeee feeeedddZd*eee fe ee ee eedddZeedddZeee fedddZddZd+d d!Z d,eee fd"d#d$Z!d-e ee ee ee eeeee ee"d' d(d)Z#dS).z Based on the logic in the PROJ projsync CLI program https://github.com/OSGeo/PROJ/blob/9ff543c4ffd86152bc58d0a0164b2ce9ebbb8bec/src/apps/projsync.cpp N)datetime)partial)Path)AnyOptional) urlretrieve)get_proj_endpoint)BBox) get_data_dirget_user_data_dir)coordsreturncCsz.t|\}}tt|t|t|t|dWSty@Yn0d}|D]Z}t|}|durd|}qJt|j|j|_t|j|j|_t|j|j|_t|j |j |_ qJ|S)z/ Get the bounding box from coordinates )westsoutheastnorthN) zipr minmax ValueError_bbox_from_coordsrrrr)r ZxxxZyyyZ coord_bboxZ coord_setbboxrG/var/www/html/django/DPS/env/lib/python3.9/site-packages/pyproj/sync.pyrs " r)geomr cCsd|vsd|vrdS|d}|ddkr0t|Sd}d}g}|D]>}t|}|durVq@|jdkrfd}n|jdkrtd}||q@d}|D]x}|r|r|jdkrd|_|jd 7_|dur|}qt|j|j|_t|j|j|_t|j|j|_t|j|j|_q|S) z4 Get the bounding box from geojson geometry coordinatestypeNZ MultiPolygonFLTh)rrrappendrrrr)rrZfound_minus_180Zfound_plus_180ZbboxesZcoordinate_setrZ grid_bboxrrr_bbox_from_geom*s:    r!)featurer spatial_testinclude_world_coverager cCs|d}|durt|}|dur&dS|j|jdkrf|j|jdkrf|sNdStd |_td|_n0|jdkr|jdkr|jd 8_|jd 8_t|||SdS) zC Filter by the bounding box. Designed to use with 'filter' ZgeometryNFiginfrrr)getr!rrrrfloatgetattr)r"rr#r$rZ geom_bboxrrr _filter_bboxNs$   r*)r" source_id area_of_usefilenamer c Cs~|d}|sdS|d}|d}|r.|s2dS|dup@||v}|dupX||ddv}|duph||v} |rz|rz| rzdSdS) zA Filter by the properties. Designed to use with 'filter' propertiesFnamer+Nr,T)r') r"r+r,r-r.Z p_filenameZ p_source_idZsource_id__matchedZarea_of_use__matchedZfilename__matchedrrr_filter_propertieshs    r1) grid_namer cCs@tt|rdSttjD]}t||r"dSq"dS)zX Run through all of the PROJ directories to see if the file already exists. FT)rr existsr splitospathsep)r2data_dirrrr_is_download_neededs r8)r"r cCs,|d}|sdS|d}|s$dSt|S)zM Filter grids so only those that need to be downloaded are included. r.Fr/)r'r8)r"r.r-rrr_filter_download_neededs  r9cs\t}t|d2tfdddD]}||q&Wdn1sJ0Y|S)z7 Return sha256 checksum of file given by path. rbcs dS)Ni)readrfilerrz_sha256sum..r?N)hashlibsha256openiterupdate hexdigest)Z input_fileZhasherchunkrr<r _sha256sums  *rGFcCs|rtd|t||d}zbt|||durP|t|krPtd||t||Wzt|WqtyYq0n$zt|WntyYn00dS)z. Download resource file from PROJ url z Downloading: z.partNzSHA256 mismatch: ) printrrrG RuntimeErrorreplacer5removeFileNotFoundError)file_url short_name directoryverboserAZtmp_pathrrr_download_resource_files    rQ)r cCsh|durtd}t|d}|r@tt|jjdkrVt t dd|dt |j ddS) zR Returns ------- dict[str, Any]: The PROJ grid data list. NTz files.geojsonrz/files.geojson)rMrNrOzutf-8)encoding)r rr3rutcnow fromtimestampstatst_mtimedaysrQrjsonloads read_text)target_directoryZ local_pathrrr_load_grid_geojsons  r\ intersectsT) r+r,r-rr#r$include_already_downloadedr[r c Cst|dd}|dur|jdkrJ|j|jkrJ|jd8_|jd8_nj|jdkr~|j|jkr~|jd7_|jd7_n6t|jdkrt|jdkr|j|jkr|jd7_ttt|||d|}ttt|||d|}|rt|Sttt |S) aI Get a list of transform grids that can be downloaded. Parameters ---------- source_id: str, optional area_of_use: str, optional filename: str, optional bbox: BBox, optional spatial_test: str, default="intersects" Can be "contains" or "intersects". include_world_coverage: bool, default=True If True, it will include grids with a global extent. include_already_downloaded: bool, default=False If True, it will list grids regardless of if they are downloaded. target_directory: Union[str, Path, None], optional The directory to download the geojson file to. Default is the user writable directory. Returns ------- list[dict[str, Any]]: A list of geojson data of containing information about features that can be downloaded. )r[featuresNrrr)rr#r$)r+r,r-) r\rrabsfilterrr*r1tupler9) r+r,r-rr#r$r^r[r_rrrget_transform_grid_lists>#(  rc)NNN)FN)N)NNNNr]TFN)$__doc__r@rXr5r functoolsrpathlibrtypingrrurllib.requestrZ pyproj._syncrZ pyproj.aoir Zpyproj.datadirr r listrdictstrr!boolr*r1r8r9rGrQr\rbrcrrrrsf      %