handle units in AmountEdit

gui/qt/amountedit.py

@@ -3,27 +3,44 @@
 from PyQt4.QtCore import *
 from PyQt4.QtGui import *
 
+from decimal import Decimal
 
 class AmountEdit(QLineEdit):
 
-    def __init__(self, text_getter, is_int = False, parent=None):
+    def __init__(self, decimal_point, is_int = False, parent=None):
         QLineEdit.__init__(self, parent)
-        self.text_getter = text_getter
+        self.decimal_point = decimal_point
         self.textChanged.connect(self.numbify)
         self.is_int = is_int
         self.is_shortcut = False
 
+    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.text_getter:
+        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.text_getter())
+             painter.drawText(textRect, Qt.AlignRight | Qt.AlignVCenter, self.base_unit())
 
 
     def numbify(self):
@@ -41,3 +58,5 @@ class AmountEdit(QLineEdit):
                 s = s[:p] + '.' + s[p:p+8]
         self.setText(s)
         self.setCursorPosition(pos)
+
+