1 # -*- coding: utf-8 -*-
3 from PyQt4.QtCore import *
4 from PyQt4.QtGui import *
6 from decimal import Decimal
8 class MyLineEdit(QLineEdit):
11 def setFrozen(self, b):
16 class AmountEdit(MyLineEdit):
18 def __init__(self, base_unit, is_int = False, parent=None):
19 QLineEdit.__init__(self, parent)
20 self.base_unit = base_unit
21 self.textChanged.connect(self.numbify)
23 self.is_shortcut = False
24 self.help_palette = QPalette()
26 def decimal_point(self):
30 text = unicode(self.text()).strip()
32 self.is_shortcut = True
33 pos = self.cursorPosition()
35 if not self.is_int: chars +='.'
36 s = ''.join([i for i in text if i in chars])
41 s = s[:p] + '.' + s[p:p+self.decimal_point()]
43 self.setCursorPosition(pos)
45 def paintEvent(self, event):
46 QLineEdit.paintEvent(self, event)
48 panel = QStyleOptionFrameV2()
49 self.initStyleOption(panel)
50 textRect = self.style().subElementRect(QStyle.SE_LineEditContents, panel, self)
51 textRect.adjust(2, 0, -10, 0)
52 painter = QPainter(self)
53 painter.setPen(self.help_palette.brush(QPalette.Disabled, QPalette.Text).color())
54 painter.drawText(textRect, Qt.AlignRight | Qt.AlignVCenter, self.base_unit())
58 class BTCAmountEdit(AmountEdit):
60 def __init__(self, decimal_point, is_int = False, parent=None):
61 AmountEdit.__init__(self, self._base_unit, is_int, parent)
62 self.decimal_point = decimal_point
65 p = self.decimal_point()
73 raise Exception('Unknown base unit')
77 x = Decimal(str(self.text()))
80 p = pow(10, self.decimal_point())
83 def setAmount(self, amount):
88 p = pow(10, self.decimal_point())
89 x = amount / Decimal(p)