/~hd1ddlZddlZddlZddlmZddlmZmZddlm Z m Z m Z m Z m Z ddgZddgZd d d Zdd Zd ZdadZdZdZdZdZdZdZdZdZGddeZddZGddeZ dZ!d dZ"y)!N)UnsupportedOperation)UnknownExtension UnknownFormat)Image ImageFileStringIO string_types getattrsafePNGWEBPGIFz.jpgz.png)JPEGr c 0t|t|||SN) save_imager )imgformat autoconvertoptionss U/var/www/peopleoo.sandbox-dev.co.uk/venv/lib/python3.12/site-packages/pilkit/utils.py img_to_fobjrs c8:vw DDcN|jdtj|S)Nr)seekropen)targets r open_imagers KKN ::f rcDtdkrtjdayy)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_initrpreinitrr _preinit_pilr#s1}   rcXttdkrtjdayy)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)r#r rinitr"rr _init_pilr'+s$N1}   rc\tjj|jSr)r EXTENSIONgetlower) extensions r_extension_to_formatr-:s ??  y0 11rc|rd|j}|tvrt|}|tjvr|Stjj D]\}}||k(s |cSyr)upperDEFAULT_EXTENSIONSrr)items)rextkvs r_format_to_extensionr5>sg  ' '$V,Ceoo% OO))+ DAqF{  rcn d|xsdz}tj|d}|S#t$rd}Y|SwxYw)Nza%sr) mimetypes guess_type IndexError)r2filenamemimetypes rextension_to_mimetyper=NsLCI2&''1!4 O  Os !% 44c*tt|Sr)r=format_to_extension)rs rformat_to_mimetyper@Ws !4V!< ==rct|}|str t|}|str t|}|s t||S)zDReturns the format that corresponds to the provided extension. )r-r#r'r)r,rs rextension_to_formatrB[sC") ,F ln%i0 ik%i0 y)) Mrcd}|r9t|}|str t|}|str t|}|s t||S)zCReturns the first extension that matches the provided format. N)r5r#r'r)rr,s rr?r?isLI (0 \^,V4IY[,V4I F## rcZtjj|d} t|}|j |j k(r|}|S t |}|j |j k(r|}|S|}|S#t $r|}Y|SwxYw#t$r|}Y|SwxYw)Nr)ospathsplitextr?r+rBrr)nameroriginal_extensionsuggested_extensionr,original_formats rsuggest_extensionrLys))$/241&9  $ $ &*<*B*B*D D*I  4"56H"I <<>_%:%:%<< 2I !4I $ 0/   0 '&  !'s# B B BB B*)B*c6eZdZfdZdZdZdZdZxZS) FileWrapperc.tt| d|y)N_wrapped)superrN __setattr__)selfwrapped __class__s r__init__zFileWrapper.__init__s k4,ZArc^ |jjS#t$rtwxYwr)rPfilenorAttributeErrorrSs rrXzFileWrapper.filenos/ !=='') )# !  !s,c.t|j|Sr)getattrrP)rSrHs r __getattr__zFileWrapper.__getattr__st}}d++rc0t|j||Sr)setattrrP)rSrHvalues rrRzFileWrapper.__setattr__st}}dE22rc.t|j|Sr)delattrrP)rSkeys r __delattr__zFileWrapper.__delattr__st}}c**r) __name__ __module__ __qualname__rVrXr]rRrd __classcell__)rUs@rrNrNsB! ,3+rrNcxsi|r"t\}|j| jdfd}t fdt Dr}n t } || jdS#t$rYUwxYw#t$rtdvrtdnddzjddzdjdzjdz}|tjkrtj} |t_ ||| t_ n#| t_ wxYwYwxYw#t$rYSwxYw) 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. rclt5j|fidddy#1swYyxYwr)quietsave)fprrrs rrlzsave_image..saves2 W , CHHR +7 + , , ,s*3c36K|]}t|ywr) isinstance).0toutfiles r zsave_image..s 8a:gq ! 8sexifr) prepare_imageupdaterrYanyr rNIOErrormaxlensizerMAXBLOCK) rrrrrr save_kwargsrlwrapper new_maxblock old_maxblocks ```` rrrsbmG(f5[ 7#  Q, 8< 88g&. W 0  Q N]    $ .%+w%6S !A B HHQK!O  Ochhqk ) ),, ,  )) )  . M!-I I  +.2   N sNB /B8E BBB E %D9- E 9 EE  E  EEceZdZdZdZdZy)rkz A context manager for suppressing the stderr activity of PIL's C libraries. Based on http://stackoverflow.com/a/978264/155370 c tjj|_ t jt jt j|_ t j|j|_ t j|j|jy#t$rYyt $rYywxYw#t$rYywxYwr)sys __stderr__rX stderr_fdrYrrErdevnullO_RDWRnull_fdOSErrordupolddup2rZs r __enter__zquiet.__enter__s  ^^224DN 772::ryy9DL 66$..)  dnn-  #       s(#B,7C, C7CC CCc t|ddsyt|ddsytj|j|jtj |j tj |jy)Nrr)r\rErrrcloser)rSargskwargss r__exit__zquiet.__exit__sVtY- tUD)  $..)  rN)rerfrg__doc__rrr"rrrkrks .(rrkcd}i}|j}|jdk(r|tvrn|tvr|j |j d}t j|d}tt dd}|jdjd|d }|jd |d |d <nd }n|jdk(r9|tvr |jd |d <nW|tvr|jd}n=d }n:|jd}|d k(r$tt dd}|jd|}|r+ddl m}|j|jd}|dk(rd |d<||fS#t$rYFwxYw)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. FRGBAc|dkrdSdS)Nrr")as rzprepare_image..9sa3hsArzPalette.ADAPTIVEADAPTIVERGBPr)palettecolors transparencyTr)rr) MakeOpaqueroptimize)r/modeRGBA_TRANSPARENCY_FORMATSPALETTE_TRANSPARENCY_FORMATSloadsplitrevalr convertpasteinfoKeyError processorsrprocess)rr make_opaqueralphamaskrrs rrxrxsKK \\^F xx6 . .  3 3& HHJIIKOE::e%EFD!%);ZHG++e$,,S'#,NC IIc4 *-K 'K S 1 1 .1hh~.F N+0 0++f%CKkk%  U?!%);ZHG++c7+3C*l""3'//6 "& J  5  sE<< FFcddlm}|j}||xsgj|}|xs|jxs|xsd}|xsi}t |||fi|S)Nr)ProcessorPipeliner)rrrrr)rrrrrrrKs r process_imagerhs_-jjO J," - 5 5c :C  >szz >_ >FmG sFK ;7 ;;r)T)NT)NNTN)#rEr8rior exceptionsrrlibrrr r r rrr0rrr r#r'r-r5r=r@rBr?rLobjectrNrrkrxrr"rrrs  #7FF#FO %u~   E    2  >   .+&+*DN!F!HUp