from decimal import Decimal
-class AmountEdit(QLineEdit):
+class MyLineEdit(QLineEdit):
+ frozen = pyqtSignal()
- def __init__(self, decimal_point, is_int = False, parent=None):
+ def setFrozen(self, b):
+ self.setReadOnly(b)
+ self.setFrame(not b)
+ self.frozen.emit()
+
+class AmountEdit(MyLineEdit):
+
+ def __init__(self, base_unit, is_int = False, parent=None):
QLineEdit.__init__(self, parent)
- self.decimal_point = decimal_point
+ self.base_unit = base_unit
self.textChanged.connect(self.numbify)
self.is_int = is_int
self.is_shortcut = False
+ self.help_palette = QPalette()
- def base_unit(self):
- p = self.decimal_point()
- assert p in [5,8]
- return "BTC" if p == 8 else "mBTC"
-
- def get_amount(self):
- x = unicode( self.text() )
- if x in['.', '']:
- return None
- p = pow(10, self.decimal_point())
- return int( p * Decimal(x) )
-
- def setAmount(self, amount):
- p = pow(10, self.decimal_point())
- x = amount / Decimal(p)
- self.setText(str(x))
-
- def paintEvent(self, event):
- QLineEdit.paintEvent(self, event)
- if self.decimal_point:
- panel = QStyleOptionFrameV2()
- self.initStyleOption(panel)
- textRect = self.style().subElementRect(QStyle.SE_LineEditContents, panel, self)
- textRect.adjust(2, 0, -10, 0)
- painter = QPainter(self)
- painter.setPen(self.palette().brush(QPalette.Disabled, QPalette.Text).color())
- painter.drawText(textRect, Qt.AlignRight | Qt.AlignVCenter, self.base_unit())
-
+ def decimal_point(self):
+ return 8
def numbify(self):
text = unicode(self.text()).strip()
if '.' in s:
p = s.find('.')
s = s.replace('.','')
- s = s[:p] + '.' + s[p:p+8]
+ s = s[:p] + '.' + s[p:p+self.decimal_point()]
self.setText(s)
self.setCursorPosition(pos)
+
+ def paintEvent(self, event):
+ QLineEdit.paintEvent(self, event)
+ if self.base_unit:
+ panel = QStyleOptionFrameV2()
+ self.initStyleOption(panel)
+ textRect = self.style().subElementRect(QStyle.SE_LineEditContents, panel, self)
+ textRect.adjust(2, 0, -10, 0)
+ painter = QPainter(self)
+ painter.setPen(self.help_palette.brush(QPalette.Disabled, QPalette.Text).color())
+ painter.drawText(textRect, Qt.AlignRight | Qt.AlignVCenter, self.base_unit())
+
+
+
+class BTCAmountEdit(AmountEdit):
+
+ def __init__(self, decimal_point, is_int = False, parent=None):
+ AmountEdit.__init__(self, self._base_unit, is_int, parent)
+ self.decimal_point = decimal_point
+
+ def _base_unit(self):
+ p = self.decimal_point()
+ assert p in [2, 5, 8]
+ if p == 8:
+ return 'BTC'
+ if p == 5:
+ return 'mBTC'
+ if p == 2:
+ return 'bits'
+ raise Exception('Unknown base unit')
+
+ def get_amount(self):
+ try:
+ x = Decimal(str(self.text()))
+ except:
+ return None
+ p = pow(10, self.decimal_point())
+ return int( p * x )
+
+ def setAmount(self, amount):
+ if amount is None:
+ self.setText("")
+ return
+
+ p = pow(10, self.decimal_point())
+ x = amount / Decimal(p)
+ self.setText(str(x))
+