Added base unit 'bits'.

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

diff --git a/gui/qt/amountedit.py b/gui/qt/amountedit.py
index f0fa040..c1fb082 100644 (file)
--- a/gui/qt/amountedit.py
+++ b/gui/qt/amountedit.py
@@ -6,10 +6,12 @@ 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):
 
@@ -19,6 +21,10 @@ class AmountEdit(MyLineEdit):
         self.textChanged.connect(self.numbify)
         self.is_int = is_int
         self.is_shortcut = False
+        self.help_palette = QPalette()
+
+    def decimal_point(self):
+        return 8
 
     def numbify(self):
         text = unicode(self.text()).strip()
@@ -32,45 +38,53 @@ 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.palette().brush(QPalette.Disabled, QPalette.Text).color())
-             painter.drawText(textRect, Qt.AlignRight | Qt.AlignVCenter, 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):
-        QLineEdit.__init__(self, parent)
+        AmountEdit.__init__(self, self._base_unit, is_int, parent)
         self.decimal_point = decimal_point
-        self.textChanged.connect(self.numbify)
-        self.is_int = is_int
-        self.is_shortcut = False
 
-    def base_unit(self):
+    def _base_unit(self):
         p = self.decimal_point()
-        assert p in [5,8]
-        return "BTC" if p == 8 else "mBTC"
+        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):
-        x = unicode( self.text() )
-        if x in['.', '']: 
+        try:
+            x = Decimal(str(self.text()))
+        except:
             return None
         p = pow(10, self.decimal_point())
-        return int( p * Decimal(x) )
+        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))