xqj!dZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZmZd ed edzfd Zdeee fd edzfdZdeee fdededed ef dZ d"deee fdedzdedzdedzd ef dZded efdZdeee fd efdZdZ d#dZd$deezdzd eee ffdZ d%dedzdedzdedzdedzdeded edeezdzd e!fd!Z"y)&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)Any) urlretrieve)get_proj_endpoint)BBox) get_data_dirget_user_data_dircoordsreturnc t|\}}tt|t|t|t|S#t$rYnwxYwd}|D]}t |}|||}t|j |j |_t|j|j|_t|j|j|_t|j|j|_ |S)z/ Get the bounding box from coordinates )westsoutheastnorthN) zipr minmax ValueError_bbox_from_coordsrrrr)r xxxyyy coord_bbox coord_setbboxs G/var/www/html/truck-me/venv/lib/python3.12/site-packages/pyproj/sync.pyrrs <SSSCCQQ    J >  + <   J!*//499=JO":#3#3TZZ@J ":#3#3TZZ@J !*//499=JO > s=A A  A geomcd|vsd|vry|d}|ddk7r t|Sd}d}g}|D]D}t|}||jdk(rd}n|jdk(rd}|j|Fd}|D]}|r-|r+|jdk(rd|_|xjd z c_||}7t |j|j|_t |j |j |_t |j|j|_t |j|j|_|S) z4 Get the bounding box from geojson geometry coordinatestypeN MultiPolygonFLTh)rrrappendrrrr)rr found_minus_180found_plus_180bboxescoordinate_setr grid_bboxs r_bbox_from_geomr,-s;D F$$6}%K F|~% --ON F% 0 <  99 "O YY# !N dI < ~$))t2CDI II I  I ;IN!)//4::>IO!)//4::>IO ;IN < featurer spatial_testinclude_world_coveragec|jd}|t|}|y|j|jz dkDrA|j|j z dkDr%|syt d |_t d|_nH|jdkDr9|jdkr*|xjdzc_|xjdzc_t|||Sy) zC Filter by the bounding box. Designed to use with 'filter' geometryFiginfr$r#r%)getr,rrrrfloatgetattr)r.rr/r0r geom_bboxs r _filter_bboxr9Qs ;;z "D #D)   NNY^^ +c 1)//1C7)#El]IN"5\IN ^^c !dii$&6 NNc !N NNc !N*wt\*955 r- source_id area_of_usefilenamec|jd}|sy|jd}|jd}|r|sy|duxs||v}|duxs||jddv}|duxs||v} |r|r| ryy) zA Filter by the properties. Designed to use with 'filter' propertiesFnamer:Nr;T)r5) r.r:r;r<r> p_filename p_source_idsource_id__matchedarea_of_use__matchedfilename__matcheds r_filter_propertiesrFks\*J 'J..-K ["d*Fi;.F&$.+rB3!D(BH ,B27H r- grid_namectt|jrytj t j D]}t||jsyy)zX Run through all of the PROJ directories to see if the file already exists. FT)rr existsr splitospathsep)rGdata_dirs r_is_download_neededrNsY   +224 N((4 ) $ + + - r-ch|jd}|sy|jd}|syt|S)zM Filter grids so only those that need to be downloaded are included. r>Fr?)r5rN)r.r>r<s r_filter_download_neededrPs7\*J ~~f%H  x ((r-ctj}t|d5tfddD]}|j | ddd|j S#1swY|j SxYw)z7 Return sha256 checksum of file given by path. rbc&jdS)Ni)read)filesrz_sha256sum..s$))E"2r-r-N)hashlibsha256openiterupdate hexdigest) input_filehasherchunkrUs @r _sha256sumr`sr ^^ F j$ !42C8 !E MM%  !!     !    s &A!!A9cl|rtd|t||d} t||||t|k7rt d||j t|| t j|y#t$rYywxYw# t j|w#t$rYwwxYwxYw)z. Download resource file from PROJ url z Downloading: z.partNzSHA256 mismatch: ) printrrr` RuntimeErrorreplacerKremoveFileNotFoundError)file_url short_name directoryverboserXtmp_paths r_download_resource_filerls  hZ()I*U34H Hh'  &Jx,@"@!2:,?@ @i45  IIh       IIh     sBAB 'A== B B  B3B$#B3$ B0-B3/B00B3target_directorycn| td}t|d}|jrOtjtj |j jz jdkDrttdd|tj|jdS)zR Returns ------- dict[str, Any]: The PROJ grid data list. Tz files.geojsonrz/files.geojson)rgrhrizutf-8)encoding)r rrIrnow fromtimestampstatst_mtimedaysrlrjsonloads read_text)rm local_paths r_load_grid_geojsonrys,T2&8J     (001B1K1KL LRRUVV)+,N;&& ::j**G*< ==r-include_already_downloadedct|d}|!|jdkDrD|j|jkDr+|xjdzc_|xjdzc_n|jdkrD|j|jkDr+|xjdz c_|xjdz c_n^t|jdkrFt|jdkr.|j|jkr|xjdz c_t t t ||||}t t t||||}|r t|Stt t|S)a= 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: str | Path, 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. )rmfeaturesr$r%r#)rr/r0)r:r;r<) ryrrabsfilterrr9rFtuplerP) r:r;r<rr/r0rzrmr|s rget_transform_grid_listrs.F"3CDZPH  99s?tyy4994 II I II I YY $))dii"7 II I II I ^c !c$))ns&:tyy499?T II I )'=     #  H"X /: ;;r-)NNN)FN)N)NNNN intersectsTFN)#__doc__rWrurKr functoolsrpathlibrtypingrurllib.requestr pyproj._syncr pyproj.aoir pyproj.datadirr r listrdictstrr,boolr9rFrNrPr`rlryrrr-rrs  &*:dtd{0!$sCx.!TD[!H #s(^#'7:TX 8!"  #s(^TztDj   4 3 4  )T#s(^ ) ) <@*>td):>d3PS8n>,!"$#'',*.B<TzB<tB<DjB< + B<  B< ! B<!%B<Dj4'B< B