o pf>@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZddlmZdZGdddejZejed d d Zed d ZeejddejdddddZeejddejdddejdejdejddejddejdejddejdejddddZeddZeejddejdddejdd ee ddd!d"ed#ed$ed%ed&ed'ed(ed)ed*d+d,Z!ed-d.Z"e"ejddejddded'ed(ed)ed*ejd/d0d1d2d3Z#e"ejdejddded'ed(ed)ed*ejd/d0d1d4d5Z$e"ejdejddded'ed(ed)ed*ejddejdd6ejd/d0d1d7d8Z%e"ejdejddded'ed(ed)ed*ejd/d0d1d9d:Z&eejddejdddejd;dd?dejd@dAejdd ee ddd!d"ed#ed$ed%ed&ejdBdCd1ejd/d0d1dDdEZ'eejddejdddejd;dd?dejd@dAejdd ee ddd!d"ed#ed$ed%ed&ed'ed(ed)ed*ejdBdCd1dGdHZ(eejd;dd?dejd@dAejdd ee ddd!d"ed'ed(ed)ed*ejdBdCd1dIdJZ)eejd;dKddFd"ejd>edejd@dAed'ed(ed)ed*dLdMZ*dS)NN)Tile)Tileset)Storage)Kilni'c@s$eZdZdejdeefddZdS) OrderCommandsctxreturncCs t|jSN)listcommands)selfrr R/home/ertert/spirit/venv/lib/python3.10/site-packages/tilekiln/scripts/__init__.py list_commandss zOrderCommands.list_commandsN)__name__ __module__ __qualname__clickContextr strrr r r rrsr)clscCsdSr r r r r rclisrcCdS)z,Commands to work with and check config filesNr r r r rconfig"srz--configT)exists)requiredtypecCst|dS)zTests a config for validity. The process will exit with exit code 0 if tilekiln can load the config. This is intended for build and CI scripts used by configs. N)tilekiln load_config)rr r rtest(s rz--layer)rz--zoomz-z)rrz-xz-yc Cst|}|dur!|t|||D]}t|qtddS|jD]9}|j |kr]| t|||}|durStjd|d|j d|j d|d dd td t|tdq$tjd |d dd td dS) zPrint the SQL for a tile or layer. Prints the SQL that would be issued to generate a particular tile layer, or if no layer is given, the entire tile. This allows manual debugging of a tile query. NrzZoom z not between min zoom z and max zoom z for layer .T)errzLayer 'z' not found in configuration) rr layer_queriesrrechosysexitlayersid render_sqlminzoommaxzoom)rlayerzoomxycsqllcr r rr14s,       r1cCr)zCommands for tile generation. All tile generation commands run queries against the source database which has the geospatial data Nr r r r rgenerateWsr3z-nz --num-threadszNumber of worker processes.)default show_defaulthelpz--source-dbnamez --source-hostz --source-portz--source-usernamez--storage-dbnamez--storage-hostz--storage-portz--storage-usernamec Cst|} ddtjD} t|t| } tdt| d| dtj |||| dd} t | }t || }t j||||d}t| |}| D] }||}|||qGd S) zGenerate specific tiles. A list of z/x/y tiles is read from stdin and those tiles are generated and saved to storage. cSh|]}t|qSr r from_string.0tr r r wztiles..z Rendering z tiles over z threadsdbnamehostportuserkwargs)r@rArBusernameN)rrr%stdinminlenrr$ psycopg_poolNullConnectionPoolrr from_configpsycopgconnectrrender save_tile)r num_threads source_dbname source_host source_portsource_usernamestorage_dbname storage_host storage_portstorage_usernamer0tilesthreadspoolstoragetilesetgen_connkilntilemvtr r rrZas&     rZcCr)zzCommands working with tile storage. These commands allow creation and manipulation of the tile storage database. Nr r r r rr]sr]z--idzOverride YAML config ID)r6c CsPt|}tj||||dd}t|}|t||} | | dS)zInitialize storage for tiles. Creates the storage for a tile layer and stores its metadata in the database. If the metadata tables have not yet been created they will also be setup. r?rDN) rrrJrKr create_schemarrLprepare_storageclose) rrVrWrXrYr(r0r\r]r^r r rinits    rfc Csj|dur |dur tdd}|durt|}|j}tj||||dd}t|}||| dS)z Destroy storage for tilesN+Missing one of '--id' or '--config' optionsr?rD) r UsageErrorrrr(rJrKrremove_tilesetre) rrVrWrXrYr(r0r\r]r r rdestroys     rj)rmultiplec Cs|dur |dur tdd}|durt|}|j}tj||||dd}t|} |dkr4| |n| ||| dS)zfMass-delete tiles from a tileset Deletes tiles from a tileset, by zoom, or delete all zooms. Nrgr?rDr ) rrhrrr(rJrKrtruncate_tablesre) rrVrWrXrYr-r(r0r\r]r r rdeletes       rmc Cs|dur |dur tdd}|durt|}|j}tj||||dd}t|}ddtj D} t dt | d| || dS) zDelete specific tiles. A list of z/x/y tiles is read from stdin and those tiles are deleted from storage. The entire list is read before deletion starts. Nrgr?rDcSr7r r8r:r r rr= r>ztiledelete..z Deleting z tiles) rrhrrr(rJrKrr%rGr$rI delete_tiles) rrVrWrXrYr(r0r\r]rZr r r tiledeletes   roz --bind-hostz 127.0.0.1zBind socket to this host.z --bind-porti@zBind socket to this port.)r4r5rr6z --base-urlz;Defaults to http://127.0.0.1:8000 or the bind host and portc Cs|tjtjj<| pt|jtjtjj<|dur |tjtjj<n d|d|tjtjj<|dur6|tjd<|dur?|tjd<|durH|tjd<|durQ|tjd<t j d|||d dS) z$Starts a server for development Nhttp://: PGDATABASEPGHOSTPGPORTPGUSERztilekiln.dev:devrArBworkers) osenvironrdevTILEKILN_CONFIGrr( TILEKILN_ID TILEKILN_URLuvicornrun) r bind_host bind_portrQrRrSrTrUbase_urlr(r r rrzs    rzzBind socket to this host. c Cs|tjtjj<t|tjtjj<| dur| tjtjj<n d|d|tjtjj<|dur2|tjd<|dur;|tjd<|durD|tjd<|durM|tjd<|durV|tjd<| dur_| tjd <| durh| tjd <| durq| tjd <t j d |||d dS)/Starts a server for pre-generated tiles from DBNrprqGENERATE_PGDATABASEGENERATE_PGHOSTGENERATE_PGPORTGENERATE_PGUSERSTORAGE_PGDATABASESTORAGE_PGHOSTSTORAGE_PGPORTSTORAGE_PGUSERztilekiln.server:liverv) rxryrserverr{rTILEKILN_THREADSrzr}r~r) rrrrQrRrSrTrUrVrWrXrYrr r rlive7s,        rcCst|tjtjj<|dur|tjtjj<n d|d|tjtjj<|dur+|tjd<|dur4|tjd<|dur=|tjd<|durF|tjd<tj d|||d dS) rNrprqrrrsrtruztilekiln.server:serverrv) rrxryrrrrzr}r~r)rrrQrVrWrXrYrr r rserveis    rz0.0.0.0c CsTtj||||dd}t|}ddlm}td|d|d||||dd S) z` Run a prometheus exporter which can be accessed for gathering metrics on stored tiles. r?rDr)serve_prometheusz&Running prometheus exporter on http://rq/N)rJrKrtilekiln.prometheusrrr$) rrrVrWrXrYr\r]rr r r prometheuss   r)+rxr%rrMrJr~r tilekiln.devtilekiln.server tilekiln.tilertilekiln.tilesetrtilekiln.storager tilekiln.kilnrPROMETHEUS_PORTGrouprgrouprrcommandoptionPathrSTRINGINTr1r3rIsched_getaffinityrZr]rfrjrmrorzrrrr r r rs&                       $