B DVW>@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`|dk rtt|ni}|dk r,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_keyZr_keyr%lib/python3.7/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|S)N)r)selfrrr__call__mszCycler.__call__NcCst|tr t|j|j|j|_n |dk r:dd|D|_nd|_t|tr`t|j|j|j|_n |dk rzdd|D|_nd|_t|j|j|_||_dS)z\Semi-private init Do not use this directly, use `cycler` function instead. NcSsg|]}t|qSr)copy).0vrrr zsz#Cycler.__init__..cSsg|]}t|qSr)r)rrrrrrs) isinstancer_left_right_opr_keys)rrroprrr__init__ps  zCycler.__init__cCs t|jS)z2 The keys this Cycler knows about )rr!)rrrrrsz Cycler.keyscskr dS|jkr(tdf|jkrDtdf|j|j|jdk r|jjkr|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. Nz-Can't replace %s with %s, %s is already a keyz)Can't replace %s with %s, %s is not a keycsg|]}|iqSrr)rentry)newoldrrrsz%Cycler.change_key..) r!rKeyErrorremover rr change_keyrrr)rr&r%r)r%r&rr)s      zCycler.change_keyccsBx<||j|jD](\}}t}|||||VqWdS)z Compose the 'left' and 'right' components of this cycle with the proper operation (zip or product as of now) N)r rrdictupdate)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|]}|iVqdS)Nr)rr)labelrr sz$Cycler._from_iter..)listrrr!)clsr0itrZretr)r0r _from_iters zCycler._from_itercs<ttr0|}ttfddt|DStddS)Nc3s |]\}}t||VqdS)N)_cycler)rkr)keyrrr1sz%Cycler.__getitem__..z+Can only use slices with Cycler.__getitem__)rsliceby_keyr r six iteritemsr)rr8transr)r8r __getitem__s  zCycler.__getitem__cCs&|jdkrtdd|jDS|S)Ncss|]}t|VqdS)N)r*)rlrrrr1sz"Cycler.__iter__..)rr rr/)rrrr__iter__s zCycler.__iter__cCs4t|t|kr(tdt|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})lenrformatrr)rotherrrr__add__s zCycler.__add__csNttrt|tSttrF|}ttfddt|DSt SdS)z Outer product of two cycles (`itertools.product`) or integer multiplication. Parameters ---------- other : Cycler or int The second Cycler or integer c3s |]\}}t||VqdS)N)r6)rr7r)rCrrr1sz!Cycler.__mul__..N) rrrintr:r r r;r<NotImplemented)rrCr=r)rCr__mul__s   zCycler.__mul__cCs||S)Nr)rrCrrr__rmul__szCycler.__rmul__cCsDtttti}|jdkr t|jSt|j}t|j}||j||S)N)rminrr rrArr )rZop_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 cycles (zip) Parameters ---------- other : Cycler The second Cycler z"Cannot += with a non-Cycler object) rr TypeErrorrrr!rrr r)rrCold_selfrrr__iadd__s   zCycler.__iadd__cCsLt|tstdt|}t|||_||_t|_t|j|j |j|_ |S)z In-place outer product of two cycles (`itertools.product`) Parameters ---------- other : Cycler The second Cycler z"Cannot *= with a non-Cycler object) rrrKrrr!rrr r)rrCrLrrr__imul__&s   zCycler.__imul__cCs<t|t|krdS|j|jAr$dStddt||DS)z Check equality Fcss|]\}}||kVqdS)Nr)rr,r-rrrr1Bsz Cycler.__eq__..)rArallr)rrCrrr__eq__9s  z Cycler.__eq__csntdtdi}|jdkrD|jtfdd|D}dj|dS||jd}d}|j|j ||jd SdS) N+*c3s|]}|VqdS)Nr)rr)labrrr1Hsz"Cycler.__repr__..zcycler({lab!r}, {itr!r}))rSr4?z({left!r} {op} {right!r}))rr"r) rrrrpopr2rBgetr r)rZop_mapr4r"msgr)rSr__repr__Ds   zCycler.__repr__cCsd}t|jtd}x|D]}|dj|d7}qWxBt|D]6}|d7}x |D]}|dj||d7}qLW|d7}q:W|d7}|S) Nz)r8zzz)valzz
{key!r}
{val!r}
)sortedrreprrBr )routputZ sorted_keysr8dr7rrr _repr_html_Os   zCycler._repr_html_cCsJ|j}tdd|D}x,|D]$}x|D]}||||q(WqW|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|]}|tfVqdS)N)r2)rr7rrrr1ssz Cycler.by_key..)rr*append)rrr.r]r7rrrr:]s   z 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 summationcss|]\}}t||VqdS)N)r6)rr7rrrrr1sz"Cycler.simplify..)r:r r r;r<)rr=rrrsimplify}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` )concat)rrCrrrrasz Cycler.concat)NN)__name__ __module__ __qualname____doc__rr#propertyrr)r/ classmethodr5r>r@rDrGrHrJrMrNrPrXr^r:Z _transposer`rarrrrrMs,  #    rcsj|j|jkr>ddddgj|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})ZbothZjust_onec3s$|]}t|||VqdS)N)r6)rr7)_l_rrrr1szconcat..)rjoinrBrr:r r )rrrWr)rhrirras  racOs|r|rtdt|dkr>t|dts2tdt|dSt|dkrRt|St|dkrftd|rttddt|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.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.css|]\}}t||VqdS)N)r6)rr7rrrrr1 szcycler..z4Must have at least a positional OR keyword argumentsN) rKrArrr6r r r;r<)argskwargsrrrcyclers(    rocsNt|trB|j}t|dkr(d}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 rkz2Can not create Cycler from a multi-property Cyclerc3s|]}|VqdS)Nr)rr)rSrrr1,sz_cycler..)rrrrArrUr5)r0r4rrWr)rSrr6s  r6)reZ __future__rrrrr; itertoolsrrZ six.movesrr operatorr r r __version__robjectrraror6rrrr)sa(=