U ꑫja@sddlmZddlZddlZddlZddlZddlmZzddlZdZ Wne k r`dZ YnXGdddZ d dd Z e d kree dS) ) annotationsN)BytesIOTFc@speZdZdZdddedfddZeddZd d Zd d Z d dZ eddZ eddZ ddZ ddZdS) Compressor)ZjpgZjpegZpngZgifZwebpzipgztgzbz2tbzxzbrZswfZflvZwoffZwoff2Z3gpZ3gppZasfZaviZm4vZmovZmp4ZmpegZmpgZwebmZwmvNTFcCs8|dkr|j}|||_||_|o&t|_|s4||_dSN)SKIP_COMPRESS_EXTENSIONSget_extension_re extension_reuse_gzipbrotli_installed use_brotlilog)self extensionsrrrquietr7/tmp/pip-unpacked-wheel-0x3iekhh/whitenoise/compress.py__init__6s  zCompressor.__init__c Cs4|stdStddttj|tjSdS)Nz^$z\.({})$|)recompileformatjoinmapescape IGNORECASE)rrrrrAs  zCompressor.get_extension_recCs|j| Sr )rsearch)rfilenamerrrshould_compressJszCompressor.should_compresscCsdSr r)rmessagerrrrMszCompressor.logc cst|d}t|}|}W5QRXt|}|jrl||}|d|||rh| ||d|VndS|j r| |}|d|||r| ||d|VdS)NrbZBrotliz.brZGzipz.gz) openosfstatfilenoreadlenrcompress_brotliis_compressed_effectively write_datar compress_gzip)rpathf stat_resultdatasize compressedrrrcompressPs   zCompressor.compressc Cs8t}tjdd|ddd}||W5QRX|S)Nwb r)r#modefileobj compresslevelmtime)rgzipGzipFilewritegetvalue)r4outputZgz_filerrrr0aszCompressor.compress_gzipcCs t|Sr )brotlir7)r4rrrr-lszCompressor.compress_brotlic Csft|}|dkrd}n||}|dk}|rJ|d|||d|dn|d|d|d|S) NrFgffffff?z{} compressed {} ({}K -> {}K)iz Skipping z (z compression not effective))r,rr)rZ encoding_namer1Z orig_sizer4Zcompressed_sizeZ is_effectiveZratiorrrr.ps z$Compressor.is_compressed_effectivelyc Cs@||}t|d}||W5QRXt||j|jf|S)Nr9)r'rAr(utimest_atimest_mtime)rr1r4suffixr3r#r2rrrr/s  zCompressor.write_data)__name__ __module__ __qualname__r printr staticmethodrr$rr7r0r-r.r/rrrrrs$$    rc Cstjdd}|jddddd|jdd d d d |jd dd dd |jddddtj}|jdddd|dtjd||}t|j|j|j |j d}t |j D]<\}}}|D],}||rt j||} || D]} qqqdS)NzSearch for all files inside *not* matching and produce compressed versions with '.gz' and '.br' suffixes (as long as this results in a smaller file)) descriptionz-qz--quietzDon't produce log output store_true)helpactionz --no-gzipzDon't produce gzip '.gz' files store_falser)rPrQdestz --no-brotliz Don't produce brotli '.br' filesrrootz(Path root from which to search for files)rPz, r*z,File extensions to exclude from compression z (default: ))nargsrPdefault)rrrrr)argparseArgumentParser add_argumentrrr parse_argsrrrrr(walkrTr$r1r7) argvparserZdefault_excludeargsZ compressordirpathZ_dirsfilesr#r1Z _compressedrrrmainsZ    rc__main__)N) __future__rrYr?r(riorrDr ImportErrorrrcrI SystemExitrrrrs   x 3