o o_C4@s`ddlmZmZddlmZddlmZmZmZddl Z ej Z e j Z e j Z e j Z e jZe jZe jZddZdevr>eZnJddad ad d Zd d ZddZeeddZddZddZddZddZddZddZddZ 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_typesNcCsddl}|}zfz1z ||jdWn |jyYnw|dp%dWWz ||j|WS|jy:YSwtyn}z(|jri|jddriWYd}~Wz ||j|WdS|jyhYdSwd}~wwz ||j|Ww|jyYww)Nrzunknown locale)locale getlocale setlocaleLC_ALLError ValueErrorargs startswith)r initialer,/usr/lib/python3/dist-packages/urwid/util.pydetect_encoding)s< rdetected_encodingz It worked!TcCs~|}|dvrtddan|dvrtddantddadaz|r2d ||aWd SWd Sty>Yd Sw) z~ Set the byte encoding to assume when processing strings and the encoding to use when converting unicode strings. )zutf-8utf8utfrF)zeuc-jpzeuc-krzeuc-cnzeuc-twgb2312gbkbig5zcn-gbuhceucjpeuckreuccneuctwcncbwideTnarrowasciirN)lowerstr_utilset_byte_encoding_use_dec_special_target_encodingencode LookupError)encodingrrr set_encodingHs      r.cCstS)z Get the mode Urwid is using when processing text strings. Returns 'narrow' for 8-bit encodings, 'wide' for CJK encodings or 'utf8' for UTF-8 encodings. )r'get_byte_encodingrrrrget_encoding_modejsr0c Cstr2t|tkr2z|tj}Wn!ty1ttjtj D]\}}| |tj |tj }qYnwt|tkrI| tj tj d}t |td}t|tsPJtj d}tj d}||}t|dtsjJ|d |t}g}g}|r|||dt|ft|dkr||fS|ddD]f} t| tsJt|tsJ| |d} t| dkr| d} t| tsJ|| t|tjdt| fq| \} } | |t} | r|| t|tjt| f| r|| t|dt| fqt|} | |fS)z: Return (encoded byte string, character set rle). rreplacer%rNr )r)typer translaterDEC_SPECIAL_CHARMAPNotImplementedErrorzipDEC_SPECIAL_CHARSALT_DEC_SPECIAL_CHARSr1SOSIcodecsr+r* isinstancersplitappendlenrle_append_modifyDEC_TAGjoin)scaltr9r:sissis0soutcoutsnslsinsonoutstrrrrapply_target_encodingss^              rOcCs totdkS)zi Return True if python is able to convert non-ascii unicode strings to the current encoding. r%)r*rrrrsupports_unicodes rPc Cs~|}d}}|dkr$t||||\}}||kr$d}t||||d\}}|||} t|||| \} }|| kr9d}|| ||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) text start_offsend_offs start_colend_colspospad_left pad_rightscrunposrrrcalc_trim_texts  r]c Cst|dt|||\}}}}t|||} t|||} |r/t||d} t| | dft| d|rBt||} t| | dft| dt||||t|| | fS)z< Return ( trimmed text, trimmed attr, trimmed cs ). rr )Nr )r]r? rle_subseg rle_get_atrle_prepend_modifyr@rrjust) rRattrcsrUrVrWeposrXrYattrtrcstralrrrtrim_text_attr_css$      rhcCs>d}|dkrdS|D]\}}|||kr|S||7}q dS)z- Return the attribute at offset pos. rNr)rler\xar[rrrr_s   r_cCsg}d}|D]9\}}|r#||kr||8}||7}q||7}||8}d}||kr*|S|||kr4||}||7}|||fq|S)z$Return a sub segment of an rle list.rr>)ristartendlrjrkr[rrrr^s&  r^cCs:d}|D]}t|tksJt||\}}||7}q|S)z] Return the number of characters covered by a run length encoded attribute list. r)r2tuplerepr)rir[vrkrrrrrle_lens  rtcCs`|\}}|s||fg|dd<dS|d\}}||kr%|||f|d<dS||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)ria_rrkrsrgr[rrrr`$s r`cCsN|\}}|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. rNrl)rirurkrslalrrrrr@6s  r@cCs*|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 r@)rirle2rrrrle_join_modifyDsr{c Csd}}|r|s gS|d\}}|d\}}g}|r^|r^t||} t|||f| f|| 8}|dkrB|t|krB||\}}|d7}|| 8}|dkrZ|t|krZ||\}}|d7}|r^|s|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)minr@r?) rle1rzi1i2a1r1a2r2rorsrrr rle_productPs$       rcCs>g}g}|D]\\}}}t|||ft|||fq||fS)z! Inverse of rle_product. ry)rir}rzrrrsrrr rle_factorns rc@s eZdZdS)TagMarkupExceptionN)__name__ __module__ __qualname__rrrrrzs rcCsFt|d\}}|ddd|}|r|dddur|d=||fS)z:Return (text string, attribute list) for tagmarkup passed.Nrrv)_tagmarkup_recurserB)tmtlrgrRrrrdecompose_tagmarkup|s rc Cst|tkr?g}g}|D].}t||\}}|r2|d\}}|d\} } || kr2| || f|d<|d=||7}||7}q ||fSt|tkr[t|dkrRtd|f|\}}t||St|ttfsitd||g|t|fgfS)zsReturn (text list, attribute list) for tagmarkup passed. tm -- tagmarkup attr -- current attribute or Nonervrz5Tuples must be in the form (attribute, tagmarkup): %rzInvalid markup element: %r) r2listrrpr?rr<rr) rrbrtlralelementrrg last_attrlast_runtop_attrtop_runrrrrs,        rcCs*t|tkot|dko|dddkS)Nrmouse)r2rpr?findevrrris_mouse_events*rcCs|ddkS)Npressr)rrrrris_mouse_pressrcs eZdZdZfddZZS) MetaSuperzadding .__supercsBtt||||t|d|rtdt|d|t|dS)Nz _%s__superz/Class has same name as one of its super classes)superr__init__hasattrAttributeErrorsetattr)clsnamebasesd __class__rrrszMetaSuper.__init__)rrr__doc__r __classcell__rrrrrsrcCs0t||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 r)int)val val_range out_rangenumdemrrr 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)selfwrappedrrrrs zStoppingContext.__init__cCs|Srr)rrrr __enter__szStoppingContext.__enter__cGs|jdSr)rstop)rexc_inforrr__exit__rzStoppingContext.__exit__N)rrrrrrrrrrrrs  r)/ __future__rrurwidr urwid.compatrrrr;r'rQ calc_width is_wide_charmove_next_charmove_prev_charwithin_double_byterlocalsrr*r)r.r0rOrPr]rhr_r^rtr`r@r{rr Exceptionrrrrrr2rrobjectrrrrrsP  " =     '