2013-04-06 14:34:12 -07:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
|
|
from PyQt4.QtCore import *
|
|
|
|
from PyQt4.QtGui import *
|
|
|
|
|
2014-06-05 03:40:07 -07:00
|
|
|
from decimal import Decimal
|
2013-04-06 14:34:12 -07:00
|
|
|
|
2014-06-05 05:49:32 -07:00
|
|
|
class MyLineEdit(QLineEdit):
|
|
|
|
|
|
|
|
def setFrozen(self, b):
|
|
|
|
self.setReadOnly(b)
|
|
|
|
self.setFrame(not b)
|
|
|
|
|
|
|
|
class AmountEdit(MyLineEdit):
|
2013-04-06 14:34:12 -07:00
|
|
|
|
2014-06-11 05:44:26 -07:00
|
|
|
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
|
2014-06-11 09:52:55 -07:00
|
|
|
self.help_palette = QPalette()
|
2014-06-11 05:44:26 -07:00
|
|
|
|
|
|
|
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:
|
2014-06-11 09:52:55 -07:00
|
|
|
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())
|
2014-06-11 05:44:26 -07:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BTCAmountEdit(AmountEdit):
|
|
|
|
|
2014-06-05 03:40:07 -07:00
|
|
|
def __init__(self, decimal_point, is_int = False, parent=None):
|
2014-06-11 09:52:55 -07:00
|
|
|
AmountEdit.__init__(self, self._base_unit, is_int, parent)
|
2014-06-05 03:40:07 -07:00
|
|
|
self.decimal_point = decimal_point
|
2013-04-06 14:34:12 -07:00
|
|
|
|
2014-06-11 09:52:55 -07:00
|
|
|
def _base_unit(self):
|
2014-06-05 03:40:07 -07:00
|
|
|
p = self.decimal_point()
|
|
|
|
assert p in [5,8]
|
|
|
|
return "BTC" if p == 8 else "mBTC"
|
|
|
|
|
|
|
|
def get_amount(self):
|
2014-06-11 09:17:27 -07:00
|
|
|
try:
|
|
|
|
x = Decimal(str(self.text()))
|
|
|
|
except:
|
2014-06-05 03:40:07 -07:00
|
|
|
return None
|
|
|
|
p = pow(10, self.decimal_point())
|
2014-06-11 09:17:27 -07:00
|
|
|
return int( p * x )
|
2014-06-05 03:40:07 -07:00
|
|
|
|
|
|
|
def setAmount(self, amount):
|
2014-06-14 09:02:45 -07:00
|
|
|
if amount is None:
|
|
|
|
self.setText("")
|
|
|
|
return
|
|
|
|
|
2014-06-05 03:40:07 -07:00
|
|
|
p = pow(10, self.decimal_point())
|
|
|
|
x = amount / Decimal(p)
|
|
|
|
self.setText(str(x))
|
2013-04-06 14:34:12 -07:00
|
|
|
|