o Df!@sdZddlZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmZdd lmZdd lmZmZd ed e efd dZdeee fd e efddZdeee fdededed ef ddZ   d/deee fde ede ede ed ef ddZded efddZdeee fd efd d!Zd"d#Z d0d%d&Z d1d eee ffd'd(Z!     ) * $ d2de ede ede ede ededed+ed,e ed e"fd-d.Z#dS)3z 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_dircoordsreturncCszt|\}}tt|t|t|t|dWSty Ynwd}|D]-}t|}|dur2|}q%t|j|j|_t|j|j|_t|j|j|_t|j |j |_ q%|S)z/ Get the bounding box from coordinates )westsoutheastnorthN) zipr minmax ValueError_bbox_from_coordsrrrr)r xxxyyy coord_bbox coord_setbboxrD/home/ubuntu/webapp/venv/lib/python3.10/site-packages/pyproj/sync.pyrs  " rgeomcCsd|vsd|vr dS|d}|ddkrt|Sd}d}g}|D]}t|}|dur+q |jdkr3d}n|jdkr:d}||q d}|D]<}|rY|rY|jdkrYd|_|jd 7_|dur`|}qDt|j|j|_t|j|j|_t|j|j|_t|j|j|_qD|S) z4 Get the bounding box from geojson geometry coordinatestypeN MultiPolygonFLTh)rrrappendrrrr)rrfound_minus_180found_plus_180bboxescoordinate_setr grid_bboxrrr_bbox_from_geom*s:    r+featurer spatial_testinclude_world_coveragecCs|d}|durRt|}|durdS|j|jdkr3|j|jdkr3|s'dStd |_td|_n|jdkrK|jdkrK|jd 8_|jd 8_t|||SdS) zC Filter by the bounding box. Designed to use with 'filter' geometryNFiginfr#r"r$)getr+rrrrfloatgetattr)r,rr-r.r geom_bboxrrr _filter_bboxNs   r6 source_id area_of_usefilenamec Cs~|d}|s dS|d}|d}|r|sdS|dup ||v}|dup,||ddv}|dup4||v} |r=|r=| r=dSdS) zA Filter by the properties. Designed to use with 'filter' propertiesFnamer7Nr8T)r2) r,r7r8r9r: p_filename p_source_idsource_id__matchedarea_of_use__matchedfilename__matchedrrr_filter_propertieshs    rB grid_namecCs@tt|r dSttjD] }t||rdSqdS)zX Run through all of the PROJ directories to see if the file already exists. FT)rr existsr splitospathsep)rCdata_dirrrr_is_download_neededsrIcCs,|d}|s dS|d}|sdSt|S)zM Filter grids so only those that need to be downloaded are included. r:Fr;)r2rI)r,r:r9rrr_filter_download_neededs  rJcsbt}t|dtfdddD]}||qWd|S1s(wY|S)z7 Return sha256 checksum of file given by path. rbcs dS)Ni)readrfilerrs z_sha256sum..N)hashlibsha256openiterupdate hexdigest) input_filehasherchunkrrMr _sha256sums   rZFcCs|r td|t||d}z2t|||dur(|t|kr(td||t||Wzt|WdStyCYdSwzt|WwtyTYww)z. Download resource file from PROJ url z Downloading: z.partNzSHA256 mismatch: ) printrrrZ RuntimeErrorreplacerFremoveFileNotFoundError)file_url short_name directoryverboserRtmp_pathrrr_download_resource_files$   recCsh|durtd}t|d}|r tt|jjdkr+t t dd|dt |j ddS) zR Returns ------- dict[str, Any]: The PROJ grid data list. NTz files.geojsonrz/files.geojson)r`rarbzutf-8)encoding)r rrDrutcnow fromtimestampstatst_mtimedaysrerjsonloads read_text)target_directory local_pathrrr_load_grid_geojsons  rq intersectsTinclude_already_downloadedroc Cst|dd}|dure|jdkr%|j|jkr%|jd8_|jd8_n5|jdkr?|j|jkr?|jd7_|jd7_nt|jdkrZt|jdkrZ|j|jkrZ|jd7_ttt|||d|}ttt|||d|}|rvt|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. )rofeaturesNr#r$r")rr-r.)r7r8r9) rqrrabsfilterrr6rBtuplerJ) r7r8r9rr-r.rsrortrrrget_transform_grid_lists>#(  rx)NNN)FN)N)NNNNrrTFN)$__doc__rQrlrFr functoolsrpathlibrtypingrrurllib.requestr pyproj._syncr pyproj.aoir pyproj.datadirr r listrdictstrr+boolr6rBrIrJrZrerqrwrxrrrrs      $