U dZ3@sdddlmZmZddlmZddlmZmZmZddl Z ej Z e j Z e j Z e j Z e jZe jZe jZddZdekr|eZn dstddad ad d Zd d ZddZeeddZddZddZddZddZddZddZ ddZ!d d!Z"d"d#Z#d$d%Z$Gd&d'd'e%Z&d(d)Z'd*d+Z(d,d-Z)d.d/Z*Gd0d1d1e+Z,d2d3Z-Gd4d5d5e.Z/dS)6)divisionprint_function)escape)bytes text_type text_typesNc Csddl}zYnXt|tkr| tj tj d}t |td}t|tsttj d}tj d}||}t|dtst|d |t}g}g}|r|||dt|ft|dkr(||fS|ddD]} t| tsHtt|tsXt| |d} t| dkr| d} t| tst|| t|tjdt| fq4| \} } | |t} | r|| t|tjt| f| r4|| t|dt| fq4t|} | |fS)z: Return (encoded byte string, character set rle). rreplacerrNr )r"typer translaterZDEC_SPECIAL_CHARMAPNotImplementedErrorzipZDEC_SPECIAL_CHARSZALT_DEC_SPECIAL_CHARSr)SOSIcodecsr$r# isinstancerAssertionErrorsplitappendlenrle_append_modifyZDEC_TAGjoin)scZaltr.r/ZsisZsis0ZsoutZcoutZsnZslZsinZsonZoutstrrrrapply_target_encodingmsX            r:cCs to tdkS)zi Return True if python is able to convert non-ascii unicode strings to the current encoding. r)r#rrrrsupports_unicodesr;c Cs~|}d}}|dkrHt||||\}}||krHd}t||||d\}}|||} t|||| \} }|| krrd}|| ||fS)a  Calculate the result of trimming text. start_offs -- offset into text to treat as screen column 0 end_offs -- offset into text to treat as the end of the line start_col -- screen column to trim at the left end_col -- screen column to trim at the right Returns (start, end, pad_left, pad_right), where: start -- resulting start offset end -- resulting end offset pad_left -- 0 for no pad or 1 for one space to be added pad_right -- 0 for no pad or 1 for one space to be added rr ) calc_text_pos) textZ start_offsZend_offs start_colend_colspospad_left pad_rightZscrunposrrrcalc_trim_texts rEc Cst|dt|||\}}}}t|||} t|||} |r^t||d} t| | dft| d|rt||} t| | dft| dt||||t|| | fS)z< Return ( trimmed text, trimmed attr, trimmed cs ). rr )Nr )rEr5 rle_subseg rle_get_atrle_append_beginning_modifyr6rrjust) r=attrZcsr>r?r@eposrArBZattrtrZcstralrrrtrim_text_attr_css.       rMcCs>d}|dkrdS|D]$\}}|||kr0|S||7}qdS)z- Return the attribute at offset pos. rNr)rlerDxarCrrrrGs   rGcCsg}d}|D]p\}}|rF||kr2||8}||7}q ||7}||8}d}||krRq~|||krf||}||7}|||fq |S)z$Return a sub segment of an rle list.rr4)rNstartendlrOrPrCrrrrFs$  rFcCs:d}|D],}t|tks$tt||\}}||7}q|S)z] Return the number of characters covered by a run length encoded attribute list. r)r*tupler2repr)rNrCvrPrrrrrle_lens  rYcCs\|\}}|s ||fg|dd<n8|d\}}||krF|||f|d<n||fg|dd<dS)z Append (a, r) (unpacked from *a_r*) to BEGINNING of rle. Merge with first run when possible MODIFIES rle parameter contents. Returns None. Nrr)rNa_rrPrXrLrCrrrrHs rHcCsN|\}}|r|dd|kr.|||fdS|d\}}|||f|d<dS)z Append (a, r) (unpacked from *a_r*) to the rle list rle. Merge with last run when possible. MODIFIES rle parameter contents. Returns None. rNrQ)rNrZrPrXZlaZlrrrrr60s  r6cCs*|sdSt||d||dd7}dS)z Append attribute list rle2 to rle. Merge last run of rle with first run of rle2 when possible. MODIFIES attr parameter contents. Returns None. Nrr r6)rNrle2rrrrle_join_modify>sr^c Csd}}|r|sgS|d\}}|d\}}g}|r|rt||} t|||f| f|| 8}|dkr|t|kr||\}}|d7}|| 8}|dkr0|t|kr0||\}}|d7}q0|S)a Merge the runs of rle1 and rle2 like this: eg. rle1 = [ ("a", 10), ("b", 5) ] rle2 = [ ("Q", 5), ("P", 10) ] rle_product: [ (("a","Q"), 5), (("a","P"), 5), (("b","P"), 5) ] rle1 and rle2 are assumed to cover the same total run. r r)minr6r5) rle1r]Zi1Zi2a1Zr1a2Zr2rTrXrrr rle_productJs$       rccCs>g}g}|D](\\}}}t|||ft|||fq ||fS)z! Inverse of rle_product. r\)rNr`r]rarbrXrrr rle_factorhs rdc@s eZdZdS)TagMarkupExceptionN)__name__ __module__ __qualname__rrrrretsrecCsFt|d\}}|ddd|}|r>|dddkr>|d=||fS)z:Return (text string, attribute list) for tagmarkup passed.Nrr[)_tagmarkup_recurser7)tmtlrLr=rrrdecompose_tagmarkupvs rlc Cst|tkr~g}g}|D]\}t||\}}|rd|d\}}|d\} } || krd| || f|d<|d=||7}||7}q||fSt|tkrt|dkrtd|f|\}}t||St|ttfstd||g|t|fgfS)zsReturn (text list, attribute list) for tagmarkup passed. tm -- tagmarkup attr -- current attribute or Noner[rz5Tuples must be in the form (attribute, tagmarkup): %rzInvalid markup element: %r) r*listrirUr5rer1rr) rjrJZrtlZralelementrkrLZ last_attrZlast_runZtop_attrZtop_runrrrris,        ricCs*t|tko(t|dko(|dddkS)NrZmouse)r*rUr5findZevrrris_mouse_eventsrscCs|ddkS)NZpressr)rqrrrrris_mouse_presssrtcs eZdZdZfddZZS) MetaSuperzadding .__supercsBtt||||t|d|r*tdt|d|t|dS)Nz _%s__superz/Class has same name as one of its super classes)superru__init__hasattrAttributeErrorsetattr)clsnamebasesd __class__rrrwszMetaSuper.__init__)rfrgrh__doc__rw __classcell__rrrrrusrucCs0t||dd|d}|dd}||S)aW Scale val in the range [0, val_range-1] to an integer in the range [0, out_range-1]. This implementation uses the "round-half-up" rounding method. >>> "%x" % int_scale(0x7, 0x10, 0x10000) '7777' >>> "%x" % int_scale(0x5f, 0x100, 0x10) '6' >>> int_scale(2, 6, 101) 40 >>> int_scale(1, 3, 4) 2 r rm)int)valZ val_rangeZ out_rangeZnumZdemrrr int_scales rc@s(eZdZdZddZddZddZdS) StoppingContextzContext manager that calls ``stop`` on a given object on exit. Used to make the ``start`` method on `MainLoop` and `BaseScreen` optionally act as context managers. cCs ||_dSN)_wrapped)selfwrappedrrrrwszStoppingContext.__init__cCs|Srr)rrrr __enter__szStoppingContext.__enter__cGs|jdSr)rstop)rexc_inforrr__exit__szStoppingContext.__exit__N)rfrgrhrrwrrrrrrrsr)0Z __future__rrZurwidrZ urwid.compatrrrr0r!r<Z calc_widthZ is_wide_charZmove_next_charZmove_prev_charZwithin_double_byterlocalsrr2r#r"r'r(r:r;rErMrGrFrYrHr6r^rcrd Exceptionrerlrirsrtr*rurobjectrrrrrsN   " =     '