o uX!0@sddlZddlZddlZddlmZddlmZmZddlm Z m Z m Z m Z dgZ ddgZdd iZd+d d Zd dZdaddZddZddZddZddZddZddZddZdd ZGd!d"d"eZd,d#d$ZGd%d&d&eZd'd(Z d-d)d*Z!dS).N)UnsupportedOperation)UnknownExtension UnknownFormat)Image ImageFileStringIO string_typesPNGGIFJPEGz.jpgTcKst|t|||SN) save_imager)imgformat autoconvertoptionsrA/var/www/jaguar/venv/lib/python3.10/site-packages/pilkit/utils.py img_to_fobjsrcCs|dt|S)Nr)seekropen)targetrrr open_images  rcCstdkr tdadSdS)zLoads the standard PIL file format drivers. Returns True if ``preinit()`` was called (and there's a potential that more drivers were loaded) or False if there is no possibility that new drivers were loaded. rTF) _pil_initrpreinitrrrr _preinit_pils rcCs"ttdkrtdadSdS)zLoads all PIL file format drivers. Returns True if ``init()`` was called (and there's a potential that more drivers were loaded) or False if there is no possibility that new drivers were loaded. TF)rrrinitrrrr _init_pil*s rcCstj|Sr )r EXTENSIONgetlower) extensionrrr_extension_to_format9sr$cCsR|r'|}|tvrt|}|tjvr|StjD] \}}||kr&|SqdSr )upperDEFAULT_EXTENSIONSrr items)rextkvrrr_format_to_extension=s r+cCs:zd|pd}t|d}W|Styd}Y|Sw)Nza%sr) mimetypes guess_type IndexError)r(filenamemimetyperrrextension_to_mimetypeMs  r2cCs tt|Sr )r2format_to_extension)rrrrformat_to_mimetypeV r4cCs<t|}|s tr t|}|strt|}|st||S)zDReturns the format that corresponds to the provided extension. )r$rrr)r#rrrrextension_to_formatZs  r6cCsDd}|rt|}|strt|}|strt|}|s t||S)zCReturns the first extension that matches the provided format. N)r+rrr)rr#rrrr3hs  r3cCstj|d}zt|}Wn ty|}Y|Sw||kr'|}|Szt|}Wn ty9|}Y|Sw||krF|}|S|}|S)Nr)ospathsplitextr3rr"r6r)nameroriginal_extensionsuggested_extensionr#original_formatrrrsuggest_extensionxs*     r>cs<eZdZfddZddZddZddZd d ZZS) FileWrappercstt|d|dS)N_wrapped)superr? __setattr__)selfwrapped __class__rr__init__szFileWrapper.__init__cCs z|jWStytwr )r@filenorAttributeErrorrCrrrrHs   zFileWrapper.filenocC t|j|Sr )getattrr@)rCr:rrr __getattr__r5zFileWrapper.__getattr__cCst|j||Sr )setattrr@)rCr:valuerrrrBszFileWrapper.__setattr__cCrKr )delattrr@)rCkeyrrr __delattr__r5zFileWrapper.__delattr__) __name__ __module__ __qualname__rGrHrMrBrR __classcell__rrrErr?s  r?c s:pi|rt\}||zdWn ty$Ynwfdd}tfddtDr;}nt}z||WnDtytdvrWt dnddj ddd j dj d }|t j krst j } |t _ z ||W| t _ n| t _ wYnwzdWStyYSw) aR Wraps PIL's ``Image.save()`` method. There are two main benefits of using this function over PIL's: 1. It gracefully handles the infamous "Suspension not allowed here" errors. 2. It prepares the image for saving using ``prepare_image()``, which will do some common-sense processing given the target format. rcs@tj|fiWddS1swYdSr )quietsave)fp)rrrrrrXs"zsave_image..savec3s|]}t|VqdSr ) isinstance).0t)outfilerr szsave_image..exifr) prepare_imageupdaterrIanyr r?IOErrormaxlensizerMAXBLOCK) rr]rrr save_kwargsrXwrapper new_maxblock old_maxblockr)rrrr]rrsJ           rc@s eZdZdZddZddZdS)rWz A context manager for suppressing the stderr activity of PIL's C libraries. Based on http://stackoverflow.com/a/978264/155370 cCstztj|_Wn tyYdSwz ttjtj|_ Wn t y(YdSwt |j|_ t |j |jdSr )sys __stderr__rH stderr_fdrIr7rdevnullO_RDWRnull_fdOSErrordupolddup2rJrrr __enter__s  zquiet.__enter__cOsLt|ddsdSt|ddsdSt|j|jt|jt|jdS)Nrtrw)rLr7rxrwrqclosert)rCargskwargsrrr__exit__s   zquiet.__exit__N)rSrTrU__doc__ryr}rrrrrWs rWcCs2d}i}|}|jdkrB|tvrnj|tvr?||d}t|dd}|djdtj dd }| d|d|d <n=d }n:|jdkrk|tvr^z |j d |d <Wn't y]Ynw|tvrh|d}nd }n|d}|d kr||jdtj d }|rddl m}||d}|dkrd |d<||fS)a Prepares the image for saving to the provided format by doing some common-sense conversions. This includes things like preserving transparency and quantizing. This function is used automatically by ``save_image()`` immediately before saving unless you specify ``autoconvert=False``. It is provided as a utility for those doing their own processing. :param img: The image to prepare for saving. :param format: The format that the image will be saved to. FRGBAcSs|dkrdSdS)Nrr)arrr4szprepare_image..RGBPr)palettecolors transparencyTr )rr) MakeOpaquer optimize)r%modeRGBA_TRANSPARENCY_FORMATSPALETTE_TRANSPARENCY_FORMATSloadsplitrevalconvertADAPTIVEpasteinfoKeyError processorsrprocess)rr make_opaquerkalphamaskrrrrrc sF          rccCsRddlm}|j}||p g|}|p|jp|pd}|pi}t|||fi|S)Nr)ProcessorPipeliner )rrrrr)rrrrrrr=rrr process_imagebs r)T)NT)NNTN)"r7r-roior exceptionsrrlibrrrr rrr&rrrrrr$r+r2r4r6r3r>objectr?rrWrcrrrrrs6    G W