1 # -*- coding: utf-8 -*-
3 from PyQt4.QtCore import *
4 from PyQt4.QtGui import *
6 from decimal import Decimal
8 class MyLineEdit(QLineEdit):
10 def setFrozen(self, b):
14 class AmountEdit(MyLineEdit):
16 def __init__(self, base_unit, is_int = False, parent=None):
17 QLineEdit.__init__(self, parent)
18 self.base_unit = base_unit
19 self.textChanged.connect(self.numbify)
21 self.is_shortcut = False
24 text = unicode(self.text()).strip()
26 self.is_shortcut = True
27 pos = self.cursorPosition()
29 if not self.is_int: chars +='.'
30 s = ''.join([i for i in text if i in chars])
35 s = s[:p] + '.' + s[p:p+8]
37 self.setCursorPosition(pos)
39 def paintEvent(self, event):
40 QLineEdit.paintEvent(self, event)
42 panel = QStyleOptionFrameV2()
43 self.initStyleOption(panel)
44 textRect = self.style().subElementRect(QStyle.SE_LineEditContents, panel, self)
45 textRect.adjust(2, 0, -10, 0)
46 painter = QPainter(self)
47 painter.setPen(self.palette().brush(QPalette.Disabled, QPalette.Text).color())
48 painter.drawText(textRect, Qt.AlignRight | Qt.AlignVCenter, self.base_unit())
52 class BTCAmountEdit(AmountEdit):
54 def __init__(self, decimal_point, is_int = False, parent=None):
55 QLineEdit.__init__(self, parent)
56 self.decimal_point = decimal_point
57 self.textChanged.connect(self.numbify)
59 self.is_shortcut = False
62 p = self.decimal_point()
64 return "BTC" if p == 8 else "mBTC"
67 x = unicode( self.text() )
70 p = pow(10, self.decimal_point())
71 return int( p * Decimal(x) )
73 def setAmount(self, amount):
74 p = pow(10, self.decimal_point())
75 x = amount / Decimal(p)