separate AmountEdit and BTCAmountEdit classes, for exchange rate plugin

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

diff --git a/gui/qt/amountedit.py b/gui/qt/amountedit.py
index 6380404..f0fa040 100644 (file)
--- a/gui/qt/amountedit.py
+++ b/gui/qt/amountedit.py
@@ -13,6 +13,44 @@ class MyLineEdit(QLineEdit):
 
 class AmountEdit(MyLineEdit):
 
+    def __init__(self, base_unit, is_int = False, parent=None):
+        QLineEdit.__init__(self, parent)
+        self.base_unit = base_unit
+        self.textChanged.connect(self.numbify)
+        self.is_int = is_int
+        self.is_shortcut = False
+
+    def numbify(self):
+        text = unicode(self.text()).strip()
+        if text == '!':
+            self.is_shortcut = True
+        pos = self.cursorPosition()
+        chars = '0123456789'
+        if not self.is_int: chars +='.'
+        s = ''.join([i for i in text if i in chars])
+        if not self.is_int:
+            if '.' in s:
+                p = s.find('.')
+                s = s.replace('.','')
+                s = s[:p] + '.' + s[p:p+8]
+        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())
+
+
+
+class BTCAmountEdit(AmountEdit):
+
     def __init__(self, decimal_point, is_int = False, parent=None):
         QLineEdit.__init__(self, parent)
         self.decimal_point = decimal_point
@@ -37,32 +75,3 @@ class AmountEdit(MyLineEdit):
         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 numbify(self):
-        text = unicode(self.text()).strip()
-        if text == '!':
-            self.is_shortcut = True
-        pos = self.cursorPosition()
-        chars = '0123456789'
-        if not self.is_int: chars +='.'
-        s = ''.join([i for i in text if i in chars])
-        if not self.is_int:
-            if '.' in s:
-                p = s.find('.')
-                s = s.replace('.','')
-                s = s[:p] + '.' + s[p:p+8]
-        self.setText(s)
-        self.setCursorPosition(pos)
-
-