3 ® h·8ã@sndZddlZddlmZddlmZmZddlmZm Z dZ dd„Z d d „Z Gd d „d ƒZ d d„Zdd„ZdS)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)ÚproductÚcycle)ÚmulÚaddz0.10.0cCs`|dk rtt|ƒƒni}|dk r,tt|ƒƒni}t|jƒƒ}t|jƒƒ}||@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)ÚnextÚiterÚsetÚkeysÚ ValueError)ÚleftÚrightZl_peekZr_peekZl_keyZr_key©rú(/tmp/pip-build-fibhr3ey/cycler/cycler.pyÚ _process_keys4s  rcsZ|j|jkr.tdj|j|j@|j|jAdƒ‚|jƒ‰|jƒ‰tt‡‡fdd„|jDƒƒS)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})ZbothZjust_onec3s$|]}t|ˆ|ˆ|ƒVqdS)N)Ú_cycler)Ú.0Úk)Ú_lÚ_rrrú eszconcat..)r r ÚformatÚby_keyrr)r r r)rrrÚconcatKs  rc@sÆeZdZdZdd„Zd+dd„Zdd„Zed d „ƒZd d „Z e d d„ƒZ dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd „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|ƒS)N)r)ÚselfrrrÚ__call__„szCycler.__call__NcCsšt|tƒr t|j|j|jƒ|_n |dk r:dd„|Dƒ|_nd|_t|tƒr`t|j|j|jƒ|_n |dk rzdd„|Dƒ|_nd|_t|j|jƒ|_||_dS)zf Semi-private init. Do not use this directly, use `cycler` function instead. NcSsg|]}tj|ƒ‘qSr)Úcopy)rÚvrrrú ’sz#Cycler.__init__..cSsg|]}tj|ƒ‘qSr)r)rrrrrr›s)Ú isinstancerÚ_leftÚ_rightÚ_oprÚ_keys)rr r ÚoprrrÚ__init__‡s  zCycler.__init__cCs ||jkS)N)r$)rrrrrÚ __contains__¢szCycler.__contains__cCs t|jƒS)z!The keys this Cycler knows about.)r r$)rrrrr ¥sz Cycler.keyscsºˆˆkr dSˆ|jkr(tdjˆˆdƒ‚ˆ|jkrDtdjˆˆdƒ‚|jjˆƒ|jjˆƒ|jdk r‚ˆ|jjkr‚|jjˆˆƒn4t |j t ƒrž|j jˆˆƒn‡‡fdd„|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)ÚoldÚnewz2Can't replace {old} with {new}, {old} is not a keycsg|]}ˆ|ˆi‘qSrr)rÚentry)r)r(rrrÌsz%Cycler.change_key..) r$r rÚKeyErrorÚremoverr"r Ú change_keyr r!r)rr(r)r)r)r(rr-ªs        zCycler.change_keycs,|dƒ}t‡fdd„|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|]}ˆ|iVqdS)Nr)rr)Úlabelrrrãsz$Cycler._from_iter..)Úlistr!r$)Úclsr.ÚitrÚretr)r.rÚ _from_iterÎszCycler._from_itercs:tˆtƒr.|jƒ}tt‡fdd„|jƒDƒƒStdƒ‚dS)Nc3s |]\}}t||ˆƒVqdS)N)r)rrr)Úkeyrrrësz%Cycler.__getitem__..z+Can only use slices with Cycler.__getitem__)r ÚslicerrrÚitemsr )rr4Útransr)r4rÚ __getitem__çs zCycler.__getitem__ccsf|jdkr&xV|jD]}t|ƒVqWn.N) r rrÚintrrrr6ÚNotImplemented)rrAr7r)rArÚ__mul__s   zCycler.__mul__cCs||S)Nr)rrArrrÚ__rmul__szCycler.__rmul__cCsDtttti}|jdkr t|jƒSt|jƒ}t|jƒ}||j||ƒS)N)r@Úminrrr"r?r!r#)rZop_dictZl_lenZr_lenrrrÚ__len__s      zCycler.__len__cCsLt|tƒstdƒ‚tj|ƒ}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Ú TypeErrorrrr$r!r@r#r")rrAÚold_selfrrrÚ__iadd__#s   zCycler.__iadd__cCsLt|tƒstdƒ‚tj|ƒ}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 rrIrrr$r!rr#r")rrArJrrrÚ__imul__5s   zCycler.__imul__cCs<t|ƒt|ƒkrdS|j|jAr$dStdd„t||ƒDƒƒS)NFcss|]\}}||kVqdS)Nr)rr;r<rrrrLsz Cycler.__eq__..)r?r Úallr@)rrArrrÚ__eq__Gs  z Cycler.__eq__cCs ||k S)Nr)rrArrrÚ__ne__Nsz Cycler.__ne__csrtdtdi}|jdkrH|jjƒ‰t‡fdd„|Dƒƒ}dˆ›d|›dS|j|jdƒ}d }|j|j ||jd SdS) Nú+Ú*c3s|]}|ˆVqdS)Nr)rr)ÚlabrrrWsz"Cycler.__repr__..zcycler(z, ú)ú?z({left!r} {op} {right!r}))r r%r ) r@rr"r Úpopr/Úgetr#rr!)rZop_mapr1r%Úmsgr)rRrÚ__repr__Ss   zCycler.__repr__cCs€d}t|jtd}x|D]}|d|›d7}qWxBt|ƒD]6}|d7}x |D]}|d||›d7}qLW|d7}q:W|d 7}|S) Nz)r4zzzzz
z
z
)Úsortedr Úreprr)rÚoutputZ sorted_keysr4ÚdrrrrÚ _repr_html_^s   zCycler._repr_html_cCsF|j}dd„|Dƒ}x,|D]$}x|D]}||j||ƒq$WqW|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|] }tƒ|“qSr)r/)rrrrrú ‚sz!Cycler.by_key..)r Úappend)rr r=r\rrrrrls   z Cycler.by_keycCs |jƒ}ttdd„|jƒDƒƒS)z‡ Simplify the cycler into a sum (but no products) of cyclers. Returns ------- simple : Cycler css|]\}}t||ƒVqdS)N)r)rrrrrrršsz"Cycler.simplify..)rrrr6)rr7rrrÚsimplifyŒs zCycler.simplify)NN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr&r'Úpropertyr r-Ú classmethodr3r8r>rBrErFrHrKrLrNrOÚ__hash__rXr]rZ _transposer`rrrrrrhs0  $    rcOsŽ|r|rtdƒ‚t|ƒdkr>t|dtƒs2tdƒ‚t|dƒSt|ƒdkrRt|ŽSt|ƒdkrftdƒ‚|r‚ttdd„|jƒDƒƒStd ƒ‚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||ƒVqdS)N)r)rrrrrrrÕszcycler..z4Must have at least a positional OR keyword argumentsN)rIr?r rrrrr6)ÚargsÚkwargsrrrÚcyclerŸs&    rlcsNt|tƒrB|j}t|ƒdkr(d}t|ƒ‚|jƒ‰‡fdd„|Dƒ}tj||ƒ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 rhz2Can not create Cycler from a multi-property Cyclerc3s|]}|ˆVqdS)Nr)rr)rRrrrósz_cycler..)r rr r?r rUr3)r.r1r rWr)rRrrÚs  r)rdrÚ functoolsrÚ itertoolsrrÚoperatorrrÚ __version__rrrrlrrrrrÚ)s 9;