o eW>@sdZddlmZmZmZmZddlZddlmZm Z ddl m Z m Z ddl mZmZddlZdZdd ZGd d d eZd d ZddZddZdS)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': '-.'} )absolute_importdivisionprint_functionunicode_literalsN)productcycle)zipreduce)muladdz0.10.0cCs`|dur tt|ni}|durtt|ni}t|}t|}||@r,td||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)leftrightl_peekr_peekl_keyr_keyr5/var/www/myenv/lib/python3.10/site-packages/cycler.py _process_keys7s   rc@seZdZdZddZd+ddZeddZd d Zd d Z e d dZ ddZ ddZ ddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&ZeZd'd(Zd)d*ZdS),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 : Cycler or None The 'left' cycler right : Cycler or None The 'right' cycler op : func or None Function which composes the 'left' and 'right' cyclers. cCst|SN)rselfrrr__call__mzCycler.__call__NcCst|trt|j|j|j|_n|durdd|D|_nd|_t|tr0t|j|j|j|_n|dur=dd|D|_nd|_t|j|j|_||_dS)z\Semi-private init Do not use this directly, use `cycler` function instead. NcSg|]}t|qSrcopy.0vrrr zz#Cycler.__init__..cSr rr!r#rrrr&r') isinstancer_left_right_opr_keys)rrroprrr__init__ps   zCycler.__init__cCs t|jS)z2 The keys this Cycler knows about )rr,rrrrrs z Cycler.keyscskrdS|jvrtdf|jvr"tdf|j|j|jdurB|jjvrB|jdSt|j t rQ|j dSfdd|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. Nz-Can't replace %s with %s, %s is already a keyz)Can't replace %s with %s, %s is not a keycsg|]}|iqSrr)r$entrynewoldrrr&sz%Cycler.change_key..) r,rKeyErrorremover r*r change_keyr(r)r)rr2r1rr0rr5s"      zCycler.change_keyccs@||j|jD]\}}t}|||||Vq dS)z Compose the 'left' and 'right' components of this cycle with the proper operation (zip or product as of now) N)r+r)r*dictupdate)raboutrrr_composes  zCycler._composecs0|d}tfdd|D|_tg|_|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 : Cycler New 'base' `Cycler` Nc3s|]}|iVqdSrrr#labelrr z$Cycler._from_iter..)listr)rr,)clsr=itrretrr<r _from_iters zCycler._from_itercs8ttr|}ttfddt|DStd)Nc3s"|] \}}t||VqdSr_cyclerr$kr%keyrrr>z%Cycler.__getitem__..z+Can only use slices with Cycler.__getitem__)r(sliceby_keyr r six iteritemsr)rrJtransrrIr __getitem__s zCycler.__getitem__cCs&|jdurtdd|jDS|S)Ncss|]}t|VqdSr)r6)r$lrrrr>r?z"Cycler.__iter__..)r*r r)r;rrrr__iter__s zCycler.__iter__cCs4t|t|krtdt|t|t||tS)z Pair-wise combine two equal length cycles (zip) Parameters ---------- other : Cycler The second Cycler z1Can only add equal length cycles, not {0} and {1})lenrformatrrrotherrrr__add__s   zCycler.__add__csJttr t|tSttr#|}ttfddt|DSt S)z Outer product of two cycles (`itertools.product`) or integer multiplication. Parameters ---------- other : Cycler or int The second Cycler or integer c3s"|] \}}t||VqdSrrErGrWrrr>rKz!Cycler.__mul__..) r(rrintrMr r rNrONotImplemented)rrWrPrrYr__mul__s  zCycler.__mul__cCs||SrrrVrrr__rmul__rzCycler.__rmul__cCsDtttti}|jdurt|jSt|j}t|j}||j||Sr)rminrr r*rTr)r+)rop_dictl_lenr_lenrrr__len__ s     zCycler.__len__cCLt|ts tdt|}t|||_||_t|_t|j|j |j|_ |S)z In-place pair-wise combine two equal length cycles (zip) Parameters ---------- other : Cycler The second Cycler z"Cannot += with a non-Cycler object) r(r TypeErrorr"rr,r)rr+r*rrWold_selfrrr__iadd__   zCycler.__iadd__cCrc)z In-place outer product of two cycles (`itertools.product`) Parameters ---------- other : Cycler The second Cycler z"Cannot *= with a non-Cycler object) r(rrdr"rr,r)rr+r*rerrr__imul__&rhzCycler.__imul__cCs<t|t|kr dS|j|jArdStddt||DS)z Check equality Fcss|] \}}||kVqdSrr)r$r8r9rrrr>Bsz Cycler.__eq__..)rTrallrrVrrr__eq__9s  z Cycler.__eq__csjtdtdi}|jdur"|jtfdd|D}dj|dS||jd}d}|j|j ||jd S) N+*c3|]}|VqdSrrr#labrrr>Hr?z"Cycler.__repr__..zcycler({lab!r}, {itr!r}))rprB?z({left!r} {op} {right!r}))rr-r) rrr*rpopr@rUgetr+r))rop_maprBr-msgrror__repr__Ds   zCycler.__repr__cCstd}t|jtd}|D] }|dj|d7}q t|D]}|d7}|D] }|dj||d7}q"|d7}q|d7}|S) NzrIzzz)valzz
{key!r}
{val!r}
)sortedrreprrUr )routput sorted_keysrJdrHrrr _repr_html_Os  zCycler._repr_html_cCsB|j}tdd|D}|D]}|D] }||||qq|S)aValues 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. css|]}|tfVqdSr)r@r$rHrrrr>ssz Cycler.by_key..)rr6append)rrr:r|rHrrrrM]sz Cycler.by_keycCs"|}ttddt|DS)zSimplify the Cycler Returned as a composition using only sums (no multiplications) Returns ------- simple : Cycler An equivalent cycler using only summationcs|] \}}t||VqdSrrErGrrrr>z"Cycler.simplify..)rMr r rNrO)rrPrrrsimplify}szCycler.simplifycCs t||S)aFConcatenate this cycler and an other. The keys must match exactly. This returns a single Cycler which is equivalent to `itertools.chain(self, other)` Examples -------- >>> num = cycler('a', range(3)) >>> let = cycler('a', 'abc') >>> num.concat(let) cycler('a', [0, 1, 2, 'a', 'b', 'c']) Parameters ---------- other : `Cycler` The `Cycler` to concatenate to this one. Returns ------- ret : `Cycler` The concatenated `Cycler` )concatrVrrrrs z Cycler.concat)NN)__name__ __module__ __qualname____doc__rr.propertyrr5r; classmethodrDrQrSrXr\r]rbrgrirkrvr}rM _transposerrrrrrrMs2  #     rcsh|j|jkrdgdj|j|j@|j|jAd}t|||ttfdd|jDS)aConcatenate two cyclers. The keys must match exactly. This returns a single Cycler which is equivalent to `itertools.chain(left, right)` Examples -------- >>> num = cycler('a', range(3)) >>> let = cycler('a', 'abc') >>> num.concat(let) cycler('a', [0, 1, 2, 'a', 'b', 'c']) Parameters ---------- left, right : `Cycler` The two `Cycler` instances to concatenate Returns ------- ret : `Cycler` The concatenated `Cycler` z )zKeys do not match:zIntersection: {both!r}zDisjoint: {just_one!r})bothjust_onec3s&|]}t|||VqdSrrEr~_l_rrrr>s$zconcat..)rjoinrUrrMr r )rrrurrrrs    rcOs|r|rtdt|dkrt|dtstdt|dSt|dkr)t|St|dkr3td|rBttddt|DStd ) 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.rzEIf 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.csrrrErGrrrr> rzcycler..z4Must have at least a positional OR keyword arguments) rdrTr(rrFr r rNrO)argskwargsrrrcyclers(    rcsNt|tr!|j}t|dkrd}t||fdd|D}t||S)aI 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 rz2Can not create Cycler from a multi-property Cyclerc3rnrrr#rorrr>,r?z_cycler..)r(rrrTrrrrD)r=rBrrurrorrFs   rF)r __future__rrrrrN itertoolsrr six.movesrr operatorr r r" __version__robjectrrrrFrrrrs*a( =