a Sic8@sndZddlZddlmZddlmZmZddlmZm Z dZ ddZ d d Z Gd d d Z d dZddZdS)a Cycler ====== Cycling through combinations of values, producing dictionaries. You can add cyclers:: from cycler import cycler cc = (cycler(color=list('rgb')) + cycler(linestyle=['-', '--', '-.'])) for d in cc: print(d) Results in:: {'color': 'r', 'linestyle': '-'} {'color': 'g', 'linestyle': '--'} {'color': 'b', 'linestyle': '-.'} You can multiply cyclers:: from cycler import cycler cc = (cycler(color=list('rgb')) * cycler(linestyle=['-', '--', '-.'])) for d in cc: print(d) Results in:: {'color': 'r', 'linestyle': '-'} {'color': 'r', 'linestyle': '--'} {'color': 'r', 'linestyle': '-.'} {'color': 'g', 'linestyle': '-'} {'color': 'g', 'linestyle': '--'} {'color': 'g', 'linestyle': '-.'} {'color': 'b', 'linestyle': '-'} {'color': 'b', 'linestyle': '--'} {'color': 'b', 'linestyle': '-.'} N)reduce)productcycle)muladdz0.10.0cCs`|durtt|ni}|dur,tt|ni}t|}t|}||@rXtd||BS)a Helper function to compose cycler keys. Parameters ---------- left, right : iterable of dictionaries or None The cyclers to be composed. Returns ------- keys : set The keys in the composition of the two cyclers. Nz"Can not compose overlapping cycles)nextitersetkeys ValueError)leftrightZl_peekZr_peekZl_keyr_keyrB/var/www/html/django/DPS/env/lib/python3.9/site-packages/cycler.py _process_keys4s  rcsZ|j|jkr.tdj|j|j@|j|jAd||ttfdd|jDS)aS Concatenate `Cycler`\s, as if chained using `itertools.chain`. The keys must match exactly. Examples -------- >>> num = cycler('a', range(3)) >>> let = cycler('a', 'abc') >>> num.concat(let) cycler('a', [0, 1, 2, 'a', 'b', 'c']) Returns ------- `Cycler` The concatenated cycler. zBKeys do not match: Intersection: {both!r} Disjoint: {just_one!r})bothZjust_onec3s$|]}t|||VqdSN_cycler.0k_l_rrr ezconcat..)r r formatby_keyrr)r r rrrconcatKs   r c@seZdZdZddZd+ddZddZed d Zd d Z e d dZ ddZ ddZ ddZddZddZddZddZddZdd Zd!d"ZdZd#d$Zd%d&Zd'd(ZeZd)d*ZeZdS),Cyclera Composable cycles. This class has compositions methods: ``+`` for 'inner' products (zip) ``+=`` in-place ``+`` ``*`` for outer products (`itertools.product`) and integer multiplication ``*=`` in-place ``*`` and supports basic slicing via ``[]``. Parameters ---------- left, right : Cycler or None The 'left' and 'right' cyclers. op : func or None Function which composes the 'left' and 'right' cyclers. cCst|Sr)rselfrrr__call__szCycler.__call__NcCst|tr t|j|j|j|_n |dur:dd|D|_nd|_t|tr`t|j|j|j|_n |durzdd|D|_nd|_t|j|j|_||_dS)zf Semi-private init. Do not use this directly, use `cycler` function instead. NcSsg|]}t|qSrcopyrvrrr rz#Cycler.__init__..cSsg|]}t|qSrr%r'rrrr)r) isinstancer!_left_right_opr_keys)r#r r oprrr__init__s  zCycler.__init__cCs ||jvSr)r.)r#rrrr __contains__szCycler.__contains__cCs t|jS)z!The keys this Cycler knows about.)r r.r"rrrr sz Cycler.keyscskr dS|jvr(tdjd|jvrDtdjd|j|j|jdur|jjvr|jn4t |j t r|j nfdd|j D|_ dS)a Change a key in this cycler to a new name. Modification is performed in-place. Does nothing if the old key is the same as the new key. Raises a ValueError if the new key is already a key. Raises a KeyError if the old key isn't a key. Nz6Can't replace {old} with {new}, {new} is already a key)oldnewz2Can't replace {old} with {new}, {old} is not a keycsg|]}|iqSrr)rentryr3r2rrr)rz%Cycler.change_key..) r.r rKeyErrorremoverr,r change_keyr*r+r!)r#r2r3rr5rr8s&      zCycler.change_keycs,|d}tfdd|D|_h|_|S)a Class method to create 'base' Cycler objects that do not have a 'right' or 'op' and for which the 'left' object is not another Cycler. Parameters ---------- label : str The property key. itr : iterable Finite length iterable of the property values. Returns ------- `Cycler` New 'base' cycler. Nc3s|]}|iVqdSrrr'labelrrrrz$Cycler._from_iter..)listr+r.)clsr:itrretrr9r _from_iterszCycler._from_itercs:ttr.|}ttfdd|DStddS)Nc3s |]\}}t||VqdSrrrrr(keyrrrrz%Cycler.__getitem__..z+Can only use slices with Cycler.__getitem__)r*slicerrritemsr )r#rBtransrrAr __getitem__s zCycler.__getitem__ccs^|jdur"|jD]}t|Vqn8||j|jD]&\}}i}|||||Vq2dSr)r,r+dictr-update)r#r aboutrrr__iter__s    zCycler.__iter__cCs8t|t|kr,tdt|dt|t||tS)z Pair-wise combine two equal length cyclers (zip). Parameters ---------- other : Cycler z&Can only add equal length cycles, not z and )lenr r!zipr#otherrrr__add__szCycler.__add__csLttrt|tSttrD|}ttfdd|DStSdS)z Outer product of two cyclers (`itertools.product`) or integer multiplication. Parameters ---------- other : Cycler or int c3s |]\}}t||VqdSrrr@rPrrrrz!Cycler.__mul__..N) r*r!rintrrrrDNotImplemented)r#rPrErrRr__mul__s   zCycler.__mul__cCs||SrrrOrrr__rmul__szCycler.__rmul__cCsDtttti}|jdur t|jSt|j}t|j}||j||Sr)rNminrrr,rMr+r-)r#op_dictZl_lenZr_lenrrr__len__s      zCycler.__len__cCsLt|tstdt|}t|||_||_t|_t|j|j |j|_ |S)z In-place pair-wise combine two equal length cyclers (zip). Parameters ---------- other : Cycler z"Cannot += with a non-Cycler object) r*r! TypeErrorr&rr.r+rNr-r,r#rPZold_selfrrr__iadd__#s   zCycler.__iadd__cCsLt|tstdt|}t|||_||_t|_t|j|j |j|_ |S)z In-place outer product of two cyclers (`itertools.product`). Parameters ---------- other : Cycler z"Cannot *= with a non-Cycler object) r*r!rZr&rr.r+rr-r,r[rrr__imul__5s   zCycler.__imul__cCs<t|t|krdS|j|jAr$dStddt||DS)NFcss|]\}}||kVqdSrr)rrIrJrrrrLrz Cycler.__eq__..)rMr allrNrOrrr__eq__Gs  z Cycler.__eq__cCs ||k SrrrOrrr__ne__Nsz Cycler.__ne__csrtdtdi}|jdurH|jtfdd|D}dd|dS||jd}d }|j|j ||jd SdS) N+*c3s|]}|VqdSrrr'labrrrWrz"Cycler.__repr__..zcycler(z, )?z({left!r} {op} {right!r}))r r/r ) rNrr,r popr;getr-rr+)r#op_mapr=r/msgrrcr__repr__Ss   zCycler.__repr__cCstd}t|jtd}|D]}|d|d7}qt|D]2}|d7}|D]}|d||d7}qD|d7}q4|d 7}|S) NzrAzzzzz
z
z
)sortedr reprr)r#output sorted_keysrBdrrrr _repr_html_^s  zCycler._repr_html_cCs>|j}dd|D}|D] }|D]}||||q q|S)a Values by key. This returns the transposed values of the cycler. Iterating over a `Cycler` yields dicts with a single value for each key, this method returns a `dict` of `list` which are the values for the given key. The returned value can be used to create an equivalent `Cycler` using only `+`. Returns ------- transpose : dict dict of lists of the values for each key. cSsi|] }|tqSr)r;rrrr rz!Cycler.by_key..)r append)r#r rKrprrrrrls z Cycler.by_keycCs |}ttdd|DS)z Simplify the cycler into a sum (but no products) of cyclers. Returns ------- simple : Cycler css|]\}}t||VqdSrrr@rrrrrz"Cycler.simplify..)rrrrD)r#rErrrsimplifys zCycler.simplify)NN)__name__ __module__ __qualname____doc__r$r0r1propertyr r8 classmethodr?rFrLrQrUrVrYr\r]r_r`__hash__rkrqr _transposertr rrrrr!hs4  $    r!cOs|r|rtdt|dkr>t|dts2tdt|dSt|dkrRt|St|dkrftd|rttdd|DStd d S) a Create a new `Cycler` object from a single positional argument, a pair of positional arguments, or the combination of keyword arguments. cycler(arg) cycler(label1=itr1[, label2=iter2[, ...]]) cycler(label, itr) Form 1 simply copies a given `Cycler` object. Form 2 composes a `Cycler` as an inner product of the pairs of keyword arguments. In other words, all of the iterables are cycled simultaneously, as if through zip(). Form 3 creates a `Cycler` from a label and an iterable. This is useful for when the label cannot be a keyword argument (e.g., an integer or a name that has a space in it). Parameters ---------- arg : Cycler Copy constructor for Cycler (does a shallow copy of iterables). label : name The property key. In the 2-arg form of the function, the label can be any hashable object. In the keyword argument form of the function, it must be a valid python identifier. itr : iterable Finite length iterable of the property values. Can be a single-property `Cycler` that would be like a key change, but as a shallow copy. Returns ------- cycler : Cycler New `Cycler` for the given property zBcyl() can only accept positional OR keyword arguments -- not both.rzDIf only one positional argument given, it must be a Cycler instance.zdOnly a single Cycler can be accepted as the lone positional argument. Use keyword arguments instead.css|]\}}t||VqdSrrr@rrrrrzcycler..z4Must have at least a positional OR keyword argumentsN)rZrMr*r!rrrrD)argskwargsrrrcyclers&    rcsNt|trB|j}t|dkr(d}t||fdd|D}t||S)aD Create a new `Cycler` object from a property name and iterable of values. Parameters ---------- label : hashable The property key. itr : iterable Finite length iterable of the property values. Returns ------- cycler : Cycler New `Cycler` for the given property r}z2Can not create Cycler from a multi-property Cyclerc3s|]}|VqdSrrr'rcrrrrz_cycler..)r*r!r rMr rgr?)r:r=r rjrrcrrs  r)rxr& functoolsr itertoolsrroperatorrr __version__rr r!rrrrrrs+ 9;