o o_*@sTddlmZddlmZddlZGdddeZGdddeZGdd d eZdS) )Edit)DecimalNcs6eZdZdZdZd fdd ZddZdd ZZS) NumEditaNumEdit - edit numerical types based on the characters in 'allowed' different numerical types can be edited: + regular int: 0123456789 + regular float: 0123456789. + regular oct: 01234567 + regular hex: 0123456789abcdef $0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZTcs"tt|||||_||_dS)N)superr__init___allowedtrimLeadingZeros)selfallowedcaptiondefaultr  __class__//usr/lib/python3/dist-packages/urwid/numedit.pyr(s zNumEdit.__init__cCst|dko ||jvS)z5 Return true for allowed characters. )lenupperr)r chrrr valid_char-szNumEdit.valid_charcCs|\}t||f|}|s>|jr>|jdkr>|jdddkr>||jd||jdd|jdkr>|jdddks|S)aC Handle editing keystrokes. Remove leading zeros. >>> e, size = NumEdit("0123456789", u"", "5002"), (10,) >>> e.keypress(size, 'home') >>> e.keypress(size, 'delete') >>> assert e.edit_text == "002" >>> e.keypress(size, 'end') >>> assert e.edit_text == "2" >>> # binary only >>> e, size = NumEdit("01", u"", ""), (10,) >>> assert e.edit_text == "" >>> e.keypress(size, '1') >>> e.keypress(size, '0') >>> e.keypress(size, '1') >>> assert e.edit_text == "101" rNr0)rkeypressr edit_pos edit_text set_edit_pos set_edit_text)r sizekeymaxcol unhandledrrrr3szNumEdit.keypress)T) __name__ __module__ __qualname____doc__ALLOWEDrrr __classcell__rrrrrs  rcs*eZdZdZd fdd ZddZZS) IntegerEditzEdit widget for integer valuesN cs||_d}|jd|j}|durSt|tttfstdt|tr>t|r>d|}t |t|t j s=td||nt|trO| ddkrOtdt|}t t|j||||jd kd dS) a caption -- caption markup default -- default edit value >>> IntegerEdit(u"", 42) >>> e, size = IntegerEdit(u"", "5002"), (10,) >>> e.keypress(size, 'home') >>> e.keypress(size, 'delete') >>> assert e.edit_text == "002" >>> e.keypress(size, 'end') >>> assert e.edit_text == "2" >>> e.keypress(size, '9') >>> e.keypress(size, '0') >>> assert e.edit_text == "290" >>> e, size = IntegerEdit("", ""), (10,) >>> assert e.value() is None >>> # binary >>> e, size = IntegerEdit(u"", "1010", base=2), (10,) >>> e.keypress(size, 'end') >>> e.keypress(size, '1') >>> assert e.edit_text == "10101" >>> assert e.value() == Decimal("21") >>> # HEX >>> e, size = IntegerEdit(u"", "10", base=16), (10,) >>> e.keypress(size, 'end') >>> e.keypress(size, 'F') >>> e.keypress(size, 'f') >>> assert e.edit_text == "10Ff" >>> assert e.keypress(size, 'G') == 'G' # unhandled key >>> assert e.edit_text == "10Ff" >>> # keep leading 0's when not base 10 >>> e, size = IntegerEdit(u"", "10FF", base=16), (10,) >>> assert e.edit_text == "10FF" >>> assert e.value() == Decimal("4351") >>> e.keypress(size, 'home') >>> e.keypress(size, 'delete') >>> e.keypress(size, '0') >>> assert e.edit_text == "00FF" >>> # test exception on incompatable value for base >>> e, size = IntegerEdit(u"", "10FG", base=16), (10,) Traceback (most recent call last): ... ValueError: invalid value: 10FG for base 16 >>> # test exception on float init value >>> e, size = IntegerEdit(u"", 10.0), (10,) Traceback (most recent call last): ... ValueError: default: Only 'str', 'int', 'long' or Decimal input allowed >>> e, size = IntegerEdit(u"", Decimal("10.0")), (10,) Traceback (most recent call last): ... ValueError: not an 'integer Decimal' instance r(N;default: Only 'str', 'int', 'long' or Decimal input allowedz^[{}]+$zinvalid value: {} for base {}rz!not an 'integer Decimal' instancer))r )baser% isinstanceintstrr ValueErrorrformatrematch IGNORECASEas_tuplerr'r)r r r r,val allowed_chars validation_rerrrrUs(7   zIntegerEdit.__init__cCs|jr tt|j|jSdS)z Return the numeric value of self.edit_text. >>> e, size = IntegerEdit(), (10,) >>> e.keypress(size, '5') >>> e.keypress(size, '1') >>> assert e.value() == 51 N)rrr.r,)r rrrvalues zIntegerEdit.value)r(Nr)r!r"r#r$rr9r&rrrrr'RsSr'cs.eZdZdZ  d fdd Zdd ZZS) FloatEditzEdit widget for float values.r(NT.csd|_||_|dvrtd|d}|durG|dkrGt|tttfs'tdt|tr8t|r8t |t|}|rCt | d|_t|}t t ||jdd|||dS) a caption -- caption markup default -- default edit value preserveSignificance -- return value has the same signif. as default decimalSeparator -- use '.' as separator by default, optionally a ',' >>> FloatEdit(u"", "1.065434") >>> e, size = FloatEdit(u"", "1.065434"), (10,) >>> e.keypress(size, 'home') >>> e.keypress(size, 'delete') >>> assert e.edit_text == ".065434" >>> e.keypress(size, 'end') >>> e.keypress(size, 'backspace') >>> assert e.edit_text == ".06543" >>> e, size = FloatEdit(), (10,) >>> e.keypress(size, '5') >>> e.keypress(size, '1') >>> e.keypress(size, '.') >>> e.keypress(size, '5') >>> e.keypress(size, '1') >>> assert e.value() == Decimal("51.51") >>> e, size = FloatEdit(decimalSeparator=":"), (10,) Traceback (most recent call last): ... ValueError: invalid decimalSeparator: : >>> e, size = FloatEdit(decimalSeparator=","), (10,) >>> e.keypress(size, '5') >>> e.keypress(size, '1') >>> e.keypress(size, ',') >>> e.keypress(size, '5') >>> e.keypress(size, '1') >>> assert e.edit_text == "51,51" >>> e, size = FloatEdit("", "3.1415", preserveSignificance=True), (10,) >>> e.keypress(size, 'end') >>> e.keypress(size, 'backspace') >>> e.keypress(size, 'backspace') >>> assert e.edit_text == "3.14" >>> assert e.value() == Decimal("3.1400") >>> e.keypress(size, '1') >>> e.keypress(size, '5') >>> e.keypress(size, '9') >>> assert e.value() == Decimal("3.1416") >>> e, size = FloatEdit("", ""), (10,) >>> assert e.value() is None >>> e, size = FloatEdit(u"", 10.0), (10,) Traceback (most recent call last): ... ValueError: default: Only 'str', 'int', 'long' or Decimal input allowed N)r<,zinvalid decimalSeparator: {}r(r*r+rr)) significance_decimalSeparatorr0r1r-r.r/rrfloatabsr5rr;rr%)r r r preserveSignificancedecimalSeparatorr6rrrrs&4zFloatEdit.__init__cCs|jrPd}|jrdj|jd}dd|j|jDdgdd\}}|jrEtt||jkrEt|dtt||j}tt |}t |j|t|d Sd S) z= Return the numeric value of self.edit_text. z {ip}.{fp}z{{ip}}.{{fp:<0{sig}d}})sigcSsg|]}|qSrr).0vrrr sz#FloatEdit.value..rr+r))ipfpN) rr>r1splitr?rr/r@r.roundr)r fmtrHrIrrrr9s   zFloatEdit.value)r(NTr<r:rrrrr;s Mr;)urwidrdecimalrr2rr'r;rrrrs  6e