o ea@sddlmZddlZddlZddlZddlZddlmZzddlZdZ Wn e y/dZ YnwGdddZ d dd Z e d krEee dS) ) annotationsN)BytesIOTFc@speZdZdZdddedfddZeddZd d Zd d Z d dZ eddZ eddZ ddZ ddZdS) Compressor)jpgjpegpnggifwebpzipgztgzbz2tbzxzbrswfflvwoffwoff23gp3gppasfavim4vmovmp4mpegmpgwebmwmvNTFcCs<|dur|j}|||_||_|ot|_|s||_dSdSN)SKIP_COMPRESS_EXTENSIONSget_extension_re extension_reuse_gzipbrotli_installed use_brotlilog)self extensionsr$r&r'quietr+F/var/www/html/venv/lib/python3.10/site-packages/whitenoise/compress.py__init__6s   zCompressor.__init__c Cs0|stdStddttj|tjS)Nz^$z\.({})$|)recompileformatjoinmapescape IGNORECASE)r)r+r+r,r"As  zCompressor.get_extension_recCs|j| Sr )r#search)r(filenamer+r+r,should_compressJszCompressor.should_compresscCsdSr r+)r(messager+r+r,r'MszCompressor.logccst|d}t|}|}Wdn1swYt|}|jrA||}|d|||r?| ||d|VndS|j r\| |}|d|||r^| ||d|VdSdSdS)NrbBrotliz.brGzipz.gz) openosfstatfilenoreadlenr&compress_brotliis_compressed_effectively write_datar$ compress_gzip)r(pathf stat_resultdatasize compressedr+r+r,compressPs"    zCompressor.compresscCsRt}tjdd|ddd}||Wd|S1s wY|S)Nwb r)r7modefileobj compresslevelmtime)rgzipGzipFilewritegetvalue)rJoutputgz_filer+r+r,rFas   zCompressor.compress_gzipcCs t|Sr )brotlirM)rJr+r+r,rCls zCompressor.compress_brotlic Csht|}|dkr d}n||}|dk}|r&|d|||d|d|S|d|d|d|S) NrFgffffff?z{} compressed {} ({}K -> {}K)iz Skipping z (z compression not effective))rBr'r1)r( encoding_namerG orig_sizerJcompressed_size is_effectiveratior+r+r,rDpsz$Compressor.is_compressed_effectivelycCsT||}t|d }||Wdn1swYt||j|jf|S)NrO)r=rWr>utimest_atimest_mtime)r(rGrJsuffixrIr7rHr+r+r,rEs   zCompressor.write_data)__name__ __module__ __qualname__r!printr- staticmethodr"r8r'rMrFrCrDrEr+r+r+r,rs $     rc Cstjdd}|jddddd|jdd d d d |jd dd dd |jddddtj}|jdddd|dtjd||}t|j|j|j |j d}t |j D]\}}}|D]}||rrt j||} || D]} qoq\qUdS)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$)rlrmdestz --no-brotliz Don't produce brotli '.br' filesr&rootz(Path root from which to search for files)rlz, r)*z,File extensions to exclude from compression z (default: ))nargsrldefault)r)r$r&r*r)argparseArgumentParser add_argumentr2rr! parse_argsr)r$r&r*r>walkrpr8rGrM) argvparserdefault_excludeargs compressordirpath_dirsfilesr7rG _compressedr+r+r,mainsX    r__main__r ) __future__rrurUr>r/iorr[r% ImportErrorrrre SystemExitr+r+r+r,s"    x3