U /e @s~UddlmZddlZddlmZddlmZddlmZmZm Z erRddl m Z e dZ de d <Gd d d eeefZdS) ) annotationsN)Iterator)MutableMapping) TYPE_CHECKINGAnyLiteral) TypeAlias)rwxarFileModecseZdZUdZded<ded<ded<d0dd d fd d Zed dddZdddddZddddddZ ddddZ ddddd Z dddd!d"Z d#dd$d%Z ddd&d'Zddd(d)Zddd*d+Zd,dd-d.d/ZZS)1Zipa9Mutable Mapping interface to a Zip file Keys must be strings, values must be bytes Parameters ---------- filename: string mode: string, ('r', 'w', 'a'), defaults to 'a' Notes ----- None of this class is thread-safe - not even normally trivial methods such as ``__len__ `` or ``__contains__``. Examples -------- >>> z = Zip('myfile.zip') # doctest: +SKIP >>> z['x'] = b'123' # doctest: +SKIP >>> z['x'] # doctest: +SKIP b'123' >>> z.flush() # flush and write metadata to disk # doctest: +SKIP strfilenamezFileMode | Literal['closed']modezzipfile.ZipFile | None_filer r )rrcs t||_||_d|_dSN)super__init__rrr)selfrr __class__,/tmp/pip-unpacked-wheel-z3s6s24u/zict/zip.pyr+s z Zip.__init__zzipfile.ZipFile)returncCs:|jdkrtd|jr |jjs4tj|j|jd|_|jS)Nclosedz File closed)r)rOSErrorrfpzipfileZipFilerrrrrfile1s  zZip.filebytes)keyrcCst|tst||j|Sr) isinstancerKeyErrorr"readrr$rrr __getitem__9s zZip.__getitem__zbytes | bytearray | memoryviewNone)r$valuercCsLt|tst|t|tttfs*t|||kr:td|j||dSNzNot supported by stdlib zipfile) r%r TypeErrorr# bytearray memoryviewNotImplementedErrorr"writestr)rr$r+rrr __setitem__>s zZip.__setitem__z Iterator[str]cCsdd|jjDS)Ncss|] }|jVqdSr)r).0Zzirrr HszZip.__iter__..)r"filelistr!rrr__iter__Gsz Zip.__iter__objectboolcCs<t|tsdSz|j|WdStk r6YdSXdS)NFT)r%rr"getinfor&r(rrr __contains__Js  zZip.__contains__cCs tddSr,)r0r(rrr __delitem__SszZip.__delitem__intcCs t|jjSr)lenr"r5r!rrr__len__Vsz Zip.__len__cCs.|jr*|jjr|jj|jd|_dSr)rrflushcloser!rrrr?Ys   z Zip.flushcCs|d|_dS)Nr)r?rr!rrrr@`sz Zip.closecCs|Srrr!rrr __enter__dsz Zip.__enter__r)argsrcGs |dSr)r@)rrBrrr__exit__gsz Zip.__exit__)r )__name__ __module__ __qualname____doc____annotations__rpropertyr"r)r2r6r:r;r>r?r@rArC __classcell__rrrrrs"   r) __future__rrcollections.abcrtypingrrrrZtyping_extensionsrr rHrr#rrrrrs