o Afj@sddlZddlZddlZddlZddlZddlZddlZddlZddl Zddl Zddl m Z m Z mZdZdZedZde_GdddeZdS) N)DatabaseDATABASE_VERSION_LATEST __version__zlocation.db.xz)z&https://location.ipfire.org/databases/zlocation.downloaderc@sLeZdZedfddZddZdifddZdd Zdd d Zdd d Z dS) DownloaderNcCs.||_t|pt|_t|j||_dSN)versionlistMIRRORSmirrorsrandomZshuffle _get_proxiesproxies)selfrr r8/usr/lib/python3.10/site-packages/location/downloader.py__init__*s zDownloader.__init__cCs0i}dD]}tjd|d}|r|||<q|S)N)ZhttpsZhttpz%s_proxy)osenvironget)rrprotocolproxyrrrr 6szDownloader._get_proxiescCsp|r tj||}tjj|dd}|ddti|D] }||||q|jD] }| |j||q*|S)NZGET)methodz User-Agentz location/%s) urllibparseZurljoinrequestZRequestupdaterZ add_headerrZ set_proxy)rurlbaseurlheadersreqheaderrrrr _make_requestAs zDownloader._make_requestc Kstd|j|jftd|jtd|D] \}}td||fqz tjj|fi|}Wn/tj j ye}z!td|j td|j D]}td||j |fqP|d}~wwtd|j td| D] \}}td||fqw|S)NzHTTP %s Request to %sz URL: %sz Headers:z %s: %szHTTP Response: %s)logdebugrZhostZfull_urlZ header_itemsrrZurlopenerror HTTPErrorcoderZ getheaders)rr kwargskvreser!rrr _send_requestVs(    zDownloader._send_requestc Ksd|jtf}i}|rtdt||d<tj|dd}||jD]}|j|||d} z5| | "} t } | d} | sAn | | } | rM|| q9Wdn1sXwY|WnGt jy|} z td | WYd} ~ q#d} ~ wtjjy} z | jd krtd |n td || f|WYd} ~ q#d} ~ ww||||std |q#t|jtjtjBtjB|WdSWdn1swYt |jt!|)Nz%s/%s%a, %d %b %Y %H:%M:%S GMTzIf-Modified-SinceF)dirdelete)rrTiz(Could not decompress downloaded file: %si0z9%s is serving an outdated database. Trying next mirror...z%s reported: %sz6Downloaded database is outdated. Trying next mirror...)"rDATABASE_FILENAMEtimestrftimegmtimetempfileZNamedTemporaryFiler r"r-lzmaZLZMADecompressorread decompresswriteflushZ LZMAErrorr#Zwarningrr%r&r'truncate_check_databaserchmodnamestatS_IRUSRS_IRGRPS_IROTHunlinkFileNotFoundError)r public_key timestampZtmpdirr(rrtZmirrorr r+Z decompressorZbufr,rrrdownloadss\           9zDownloader.downloadc Cstd|jt|j}|r|j|krdStdtdt|jt |d}| |s>t d WddSWddS1sIwYdS) za Checks the downloaded database if it can be opened, verified and if it is recent enough z!Opening downloaded database at %sFzDownloaded new database from %sr.rzCould not verify databaseNT) r#r$r>rZ created_atinfor2r3r4openZverifyr%)rfrErFZdbrrrr<s"       zDownloader._check_database)NNr) __name__ __module__ __qualname__rrr r"r-rHr<rrrrr)s   Gr)Zloggingr6rr r?r5r2Z urllib.errorrZ urllib.parseZurllib.requestZ _locationrrrr1r Z getLoggerr#Z propagateobjectrrrrrs