Added base unit 'bits'.

[electrum-nvc.git] / gui / qt / amountedit.py

diff --git a/gui/qt/amountedit.py b/gui/qt/amountedit.py
index 6380404..c1fb082 100644 (file)
--- a/gui/qt/amountedit.py
+++ b/gui/qt/amountedit.py
@@ -6,48 +6,25 @@ from PyQt4.QtGui import *
 from decimal import Decimal
 
 class MyLineEdit(QLineEdit):
+    frozen = pyqtSignal()
 
     def setFrozen(self, b):
         self.setReadOnly(b)
         self.setFrame(not b)
+        self.frozen.emit()
 
 class AmountEdit(MyLineEdit):
 
-    def __init__(self, decimal_point, is_int = False, parent=None):
+    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()
@@ -61,8 +38,54 @@ class AmountEdit(MyLineEdit):
             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))