Added base unit 'bits'.

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

diff --git a/gui/qt/amountedit.py b/gui/qt/amountedit.py
index d7b0a92..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):
 
@@ -21,6 +23,9 @@ class AmountEdit(MyLineEdit):
         self.is_shortcut = False
         self.help_palette = QPalette()
 
+    def decimal_point(self):
+        return 8
+
     def numbify(self):
         text = unicode(self.text()).strip()
         if text == '!':
@@ -33,7 +38,7 @@ 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)
 
@@ -58,8 +63,14 @@ class BTCAmountEdit(AmountEdit):
 
     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):
         try: