-WIP-electrum-btcp/gui/qt/amountedit.py

105 lines
3.2 KiB
Python
Raw Normal View History

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
from electrum.util import format_satoshis_plain
2013-04-06 14:34:12 -07:00
2014-06-05 05:49:32 -07:00
class MyLineEdit(QLineEdit):
frozen = pyqtSignal()
2014-06-05 05:49:32 -07:00
def setFrozen(self, b):
self.setReadOnly(b)
self.setFrame(not b)
self.frozen.emit()
2014-06-05 05:49:32 -07:00
class AmountEdit(MyLineEdit):
2014-09-07 09:45:06 -07:00
shortcut = pyqtSignal()
2013-04-06 14:34:12 -07:00
def __init__(self, base_unit, is_int = False, parent=None):
QLineEdit.__init__(self, parent)
# This seems sufficient for hundred-BTC amounts with 8 decimals
self.setFixedWidth(140)
self.base_unit = base_unit
self.textChanged.connect(self.numbify)
self.is_int = is_int
self.is_shortcut = False
self.help_palette = QPalette()
2014-06-16 09:38:28 -07:00
def decimal_point(self):
return 8
def numbify(self):
text = unicode(self.text()).strip()
if text == '!':
2014-09-07 09:45:06 -07:00
self.shortcut.emit()
return
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('.','')
2014-06-16 09:38:28 -07:00
s = s[:p] + '.' + s[p:p+self.decimal_point()]
self.setText(s)
# setText sets Modified to False. Instead we want to remember
# if updates were because of user modification.
self.setModified(self.hasFocus())
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.help_palette.brush(QPalette.Disabled, QPalette.Text).color())
painter.drawText(textRect, Qt.AlignRight | Qt.AlignVCenter, self.base_unit())
2014-07-16 06:33:59 -07:00
def get_amount(self):
try:
return (int if self.is_int else Decimal)(str(self.text()))
2014-07-16 06:33:59 -07:00
except:
return None
class BTCAmountEdit(AmountEdit):
2014-06-05 03:40:07 -07:00
def __init__(self, decimal_point, is_int = False, parent=None):
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
def _base_unit(self):
2014-06-05 03:40:07 -07:00
p = self.decimal_point()
2014-06-30 07:40:11 -07:00
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')
2014-06-05 03:40:07 -07:00
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(" ") # Space forces repaint in case units changed
else:
self.setText(format_satoshis_plain(amount, self.decimal_point()))
2015-08-03 22:15:54 -07:00
class BTCkBEdit(BTCAmountEdit):
def _base_unit(self):
return BTCAmountEdit._base_unit(self) + '/kB'