419 lines
17 KiB
Python
419 lines
17 KiB
Python
#!/usr/bin/env python
|
|
#
|
|
# Electrum - lightweight bitcoinprivate client
|
|
# Copyright (C) 2015 Thomas Voegtlin
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person
|
|
# obtaining a copy of this software and associated documentation files
|
|
# (the "Software"), to deal in the Software without restriction,
|
|
# including without limitation the rights to use, copy, modify, merge,
|
|
# publish, distribute, sublicense, and/or sell copies of the Software,
|
|
# and to permit persons to whom the Software is furnished to do so,
|
|
# subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be
|
|
# included in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
|
|
# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
|
# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
# SOFTWARE.
|
|
|
|
import webbrowser
|
|
import datetime
|
|
|
|
from electrum_bitcoinprivate.wallet import AddTransactionException, TX_HEIGHT_LOCAL
|
|
from .util import *
|
|
from electrum_bitcoinprivate.i18n import _
|
|
from electrum_bitcoinprivate.util import block_explorer_URL, profiler
|
|
|
|
try:
|
|
from electrum_bitcoinprivate.plot import plot_history, NothingToPlotException
|
|
except:
|
|
plot_history = None
|
|
|
|
# note: this list needs to be kept in sync with another in kivy
|
|
TX_ICONS = [
|
|
"unconfirmed.png",
|
|
"warning.png",
|
|
"unconfirmed.png",
|
|
"offline_tx.png",
|
|
"clock1.png",
|
|
"clock2.png",
|
|
"clock3.png",
|
|
"clock4.png",
|
|
"clock5.png",
|
|
"confirmed.png",
|
|
]
|
|
|
|
|
|
class HistoryList(MyTreeWidget, AcceptFileDragDrop):
|
|
filter_columns = [2, 3, 4] # Date, Description, Amount
|
|
|
|
def __init__(self, parent=None):
|
|
MyTreeWidget.__init__(self, parent, self.create_menu, [], 3)
|
|
AcceptFileDragDrop.__init__(self, ".txn")
|
|
self.refresh_headers()
|
|
self.setColumnHidden(1, True)
|
|
self.setSortingEnabled(True)
|
|
self.sortByColumn(0, Qt.AscendingOrder)
|
|
self.start_timestamp = None
|
|
self.end_timestamp = None
|
|
self.years = []
|
|
self.create_toolbar_buttons()
|
|
self.wallet = None
|
|
|
|
def format_date(self, d):
|
|
return str(datetime.date(d.year, d.month, d.day)) if d else _('None')
|
|
|
|
def refresh_headers(self):
|
|
headers = ['', '', _('Date'), _('Description'), _('Amount'), _('Balance')]
|
|
fx = self.parent.fx
|
|
if fx and fx.show_history():
|
|
headers.extend(['%s '%fx.ccy + _('Value')])
|
|
self.editable_columns |= {6}
|
|
if fx.get_history_capital_gains_config():
|
|
headers.extend(['%s '%fx.ccy + _('Acquisition price')])
|
|
headers.extend(['%s '%fx.ccy + _('Capital Gains')])
|
|
else:
|
|
self.editable_columns -= {6}
|
|
self.update_headers(headers)
|
|
|
|
def get_domain(self):
|
|
'''Replaced in address_dialog.py'''
|
|
return self.wallet.get_addresses()
|
|
|
|
def on_combo(self, x):
|
|
s = self.period_combo.itemText(x)
|
|
x = s == _('Custom')
|
|
self.start_button.setEnabled(x)
|
|
self.end_button.setEnabled(x)
|
|
if s == _('All'):
|
|
self.start_timestamp = None
|
|
self.end_timestamp = None
|
|
self.start_button.setText("-")
|
|
self.end_button.setText("-")
|
|
else:
|
|
try:
|
|
year = int(s)
|
|
except:
|
|
return
|
|
start_date = datetime.datetime(year, 1, 1)
|
|
end_date = datetime.datetime(year+1, 1, 1)
|
|
self.start_timestamp = time.mktime(start_date.timetuple())
|
|
self.end_timestamp = time.mktime(end_date.timetuple())
|
|
self.start_button.setText(_('From') + ' ' + self.format_date(start_date))
|
|
self.end_button.setText(_('To') + ' ' + self.format_date(end_date))
|
|
self.update()
|
|
|
|
def create_toolbar_buttons(self):
|
|
self.period_combo = QComboBox()
|
|
self.start_button = QPushButton('-')
|
|
self.start_button.pressed.connect(self.select_start_date)
|
|
self.start_button.setEnabled(False)
|
|
self.end_button = QPushButton('-')
|
|
self.end_button.pressed.connect(self.select_end_date)
|
|
self.end_button.setEnabled(False)
|
|
self.period_combo.addItems([_('All'), _('Custom')])
|
|
self.period_combo.activated.connect(self.on_combo)
|
|
|
|
def get_toolbar_buttons(self):
|
|
return self.period_combo, self.start_button, self.end_button
|
|
|
|
def on_hide_toolbar(self):
|
|
self.start_timestamp = None
|
|
self.end_timestamp = None
|
|
self.update()
|
|
|
|
def save_toolbar_state(self, state, config):
|
|
config.set_key('show_toolbar_history', state)
|
|
|
|
def select_start_date(self):
|
|
self.start_timestamp = self.select_date(self.start_button)
|
|
self.update()
|
|
|
|
def select_end_date(self):
|
|
self.end_timestamp = self.select_date(self.end_button)
|
|
self.update()
|
|
|
|
def select_date(self, button):
|
|
d = WindowModalDialog(self, _("Select date"))
|
|
d.setMinimumSize(600, 150)
|
|
d.date = None
|
|
vbox = QVBoxLayout()
|
|
def on_date(date):
|
|
d.date = date
|
|
cal = QCalendarWidget()
|
|
cal.setGridVisible(True)
|
|
cal.clicked[QDate].connect(on_date)
|
|
vbox.addWidget(cal)
|
|
vbox.addLayout(Buttons(OkButton(d), CancelButton(d)))
|
|
d.setLayout(vbox)
|
|
if d.exec_():
|
|
if d.date is None:
|
|
return None
|
|
date = d.date.toPyDate()
|
|
button.setText(self.format_date(date))
|
|
return time.mktime(date.timetuple())
|
|
|
|
def show_summary(self):
|
|
h = self.summary
|
|
if not h:
|
|
self.parent.show_message(_("Nothing to summarize."))
|
|
return
|
|
start_date = h.get('start_date')
|
|
end_date = h.get('end_date')
|
|
format_amount = lambda x: self.parent.format_amount(x.value) + ' ' + self.parent.base_unit()
|
|
d = WindowModalDialog(self, _("Summary"))
|
|
d.setMinimumSize(600, 150)
|
|
vbox = QVBoxLayout()
|
|
grid = QGridLayout()
|
|
grid.addWidget(QLabel(_("Start")), 0, 0)
|
|
grid.addWidget(QLabel(self.format_date(start_date)), 0, 1)
|
|
grid.addWidget(QLabel(_("End")), 1, 0)
|
|
grid.addWidget(QLabel(self.format_date(end_date)), 1, 1)
|
|
grid.addWidget(QLabel(_("Initial balance")), 2, 0)
|
|
grid.addWidget(QLabel(format_amount(h['start_balance'])), 2, 1)
|
|
grid.addWidget(QLabel(str(h.get('start_fiat_balance'))), 2, 2)
|
|
grid.addWidget(QLabel(_("Final balance")), 4, 0)
|
|
grid.addWidget(QLabel(format_amount(h['end_balance'])), 4, 1)
|
|
grid.addWidget(QLabel(str(h.get('end_fiat_balance'))), 4, 2)
|
|
grid.addWidget(QLabel(_("Income")), 5, 0)
|
|
grid.addWidget(QLabel(format_amount(h.get('income'))), 5, 1)
|
|
grid.addWidget(QLabel(str(h.get('fiat_income'))), 5, 2)
|
|
grid.addWidget(QLabel(_("Expenditures")), 6, 0)
|
|
grid.addWidget(QLabel(format_amount(h.get('expenditures'))), 6, 1)
|
|
grid.addWidget(QLabel(str(h.get('fiat_expenditures'))), 6, 2)
|
|
grid.addWidget(QLabel(_("Capital gains")), 7, 0)
|
|
grid.addWidget(QLabel(str(h.get('capital_gains'))), 7, 2)
|
|
grid.addWidget(QLabel(_("Unrealized gains")), 8, 0)
|
|
grid.addWidget(QLabel(str(h.get('unrealized_gains', ''))), 8, 2)
|
|
vbox.addLayout(grid)
|
|
vbox.addLayout(Buttons(CloseButton(d)))
|
|
d.setLayout(vbox)
|
|
d.exec_()
|
|
|
|
def plot_history_dialog(self):
|
|
if plot_history is None:
|
|
self.parent.show_message(
|
|
_("Can't plot history.") + '\n' +
|
|
_("Perhaps some dependencies are missing...") + " (matplotlib?)")
|
|
return
|
|
try:
|
|
plt = plot_history(self.transactions)
|
|
plt.show()
|
|
except NothingToPlotException as e:
|
|
self.parent.show_message(str(e))
|
|
|
|
@profiler
|
|
def on_update(self):
|
|
self.wallet = self.parent.wallet
|
|
fx = self.parent.fx
|
|
r = self.wallet.get_full_history(domain=self.get_domain(), from_timestamp=self.start_timestamp, to_timestamp=self.end_timestamp, fx=fx)
|
|
self.transactions = r['transactions']
|
|
self.summary = r['summary']
|
|
if not self.years and self.transactions:
|
|
from datetime import date
|
|
start_date = self.transactions[0].get('date') or date.today()
|
|
end_date = self.transactions[-1].get('date') or date.today()
|
|
self.years = [str(i) for i in range(start_date.year, end_date.year + 1)]
|
|
self.period_combo.insertItems(1, self.years)
|
|
item = self.currentItem()
|
|
current_tx = item.data(0, Qt.UserRole) if item else None
|
|
self.clear()
|
|
if fx: fx.history_used_spot = False
|
|
for tx_item in self.transactions:
|
|
tx_hash = tx_item['txid']
|
|
height = tx_item['height']
|
|
conf = tx_item['confirmations']
|
|
timestamp = tx_item['timestamp']
|
|
value = tx_item['value'].value
|
|
balance = tx_item['balance'].value
|
|
label = tx_item['label']
|
|
status, status_str = self.wallet.get_tx_status(tx_hash, height, conf, timestamp)
|
|
has_invoice = self.wallet.invoices.paid.get(tx_hash)
|
|
icon = self.icon_cache.get(":icons/" + TX_ICONS[status])
|
|
v_str = self.parent.format_amount(value, True, whitespaces=True)
|
|
balance_str = self.parent.format_amount(balance, whitespaces=True)
|
|
entry = ['', tx_hash, status_str, label, v_str, balance_str]
|
|
fiat_value = None
|
|
if value is not None and fx and fx.show_history():
|
|
fiat_value = tx_item['fiat_value'].value
|
|
value_str = fx.format_fiat(fiat_value)
|
|
entry.append(value_str)
|
|
# fixme: should use is_mine
|
|
if value < 0:
|
|
entry.append(fx.format_fiat(tx_item['acquisition_price'].value))
|
|
entry.append(fx.format_fiat(tx_item['capital_gain'].value))
|
|
item = SortableTreeWidgetItem(entry)
|
|
item.setIcon(0, icon)
|
|
item.setToolTip(0, str(conf) + " confirmation" + ("s" if conf != 1 else ""))
|
|
item.setData(0, SortableTreeWidgetItem.DataRole, (status, conf))
|
|
if has_invoice:
|
|
item.setIcon(3, self.icon_cache.get(":icons/seal"))
|
|
for i in range(len(entry)):
|
|
if i>3:
|
|
item.setTextAlignment(i, Qt.AlignRight | Qt.AlignVCenter)
|
|
item.setFont(i, QFont(MONOSPACE_FONT))
|
|
if value and value < 0:
|
|
item.setForeground(3, QBrush(QColor("#BC1E1E")))
|
|
item.setForeground(4, QBrush(QColor("#BC1E1E")))
|
|
if fiat_value and not tx_item['fiat_default']:
|
|
item.setForeground(6, QBrush(QColor("#1E1EFF")))
|
|
if tx_hash:
|
|
item.setData(0, Qt.UserRole, tx_hash)
|
|
self.insertTopLevelItem(0, item)
|
|
if current_tx == tx_hash:
|
|
self.setCurrentItem(item)
|
|
|
|
def on_edited(self, item, column, prior):
|
|
'''Called only when the text actually changes'''
|
|
key = item.data(0, Qt.UserRole)
|
|
text = item.text(column)
|
|
# fixme
|
|
if column == 3:
|
|
self.parent.wallet.set_label(key, text)
|
|
self.update_labels()
|
|
self.parent.update_completions()
|
|
elif column == 6:
|
|
self.parent.wallet.set_fiat_value(key, self.parent.fx.ccy, text)
|
|
self.on_update()
|
|
|
|
def on_doubleclick(self, item, column):
|
|
if self.permit_edit(item, column):
|
|
super(HistoryList, self).on_doubleclick(item, column)
|
|
else:
|
|
tx_hash = item.data(0, Qt.UserRole)
|
|
tx = self.wallet.transactions.get(tx_hash)
|
|
self.parent.show_transaction(tx)
|
|
|
|
def update_labels(self):
|
|
root = self.invisibleRootItem()
|
|
child_count = root.childCount()
|
|
for i in range(child_count):
|
|
item = root.child(i)
|
|
txid = item.data(0, Qt.UserRole)
|
|
label = self.wallet.get_label(txid)
|
|
item.setText(3, label)
|
|
|
|
def update_item(self, tx_hash, height, conf, timestamp):
|
|
if self.wallet is None:
|
|
return
|
|
status, status_str = self.wallet.get_tx_status(tx_hash, height, conf, timestamp)
|
|
icon = self.icon_cache.get(":icons/" + TX_ICONS[status])
|
|
items = self.findItems(tx_hash, Qt.UserRole|Qt.MatchContains|Qt.MatchRecursive, column=1)
|
|
if items:
|
|
item = items[0]
|
|
item.setIcon(0, icon)
|
|
item.setData(0, SortableTreeWidgetItem.DataRole, (status, conf))
|
|
item.setText(2, status_str)
|
|
|
|
def create_menu(self, position):
|
|
self.selectedIndexes()
|
|
item = self.currentItem()
|
|
if not item:
|
|
return
|
|
column = self.currentColumn()
|
|
tx_hash = item.data(0, Qt.UserRole)
|
|
if not tx_hash:
|
|
return
|
|
if column is 0:
|
|
column_title = "ID"
|
|
column_data = tx_hash
|
|
else:
|
|
column_title = self.headerItem().text(column)
|
|
column_data = item.text(column)
|
|
tx_URL = block_explorer_URL(self.config, 'tx', tx_hash)
|
|
height, conf, timestamp = self.wallet.get_tx_height(tx_hash)
|
|
tx = self.wallet.transactions.get(tx_hash)
|
|
is_relevant, is_mine, v, fee = self.wallet.get_wallet_delta(tx)
|
|
is_unconfirmed = height <= 0
|
|
pr_key = self.wallet.invoices.paid.get(tx_hash)
|
|
menu = QMenu()
|
|
if height == TX_HEIGHT_LOCAL:
|
|
menu.addAction(_("Remove"), lambda: self.remove_local_tx(tx_hash))
|
|
menu.addAction(_("Copy {}").format(column_title), lambda: self.parent.app.clipboard().setText(column_data))
|
|
for c in self.editable_columns:
|
|
menu.addAction(_("Edit {}").format(self.headerItem().text(c)),
|
|
lambda bound_c=c: self.editItem(item, bound_c))
|
|
menu.addAction(_("Details"), lambda: self.parent.show_transaction(tx))
|
|
if pr_key:
|
|
menu.addAction(self.icon_cache.get(":icons/seal"), _("View invoice"), lambda: self.parent.show_invoice(pr_key))
|
|
if tx_URL:
|
|
menu.addAction(_("View on block explorer"), lambda: webbrowser.open(tx_URL))
|
|
menu.exec_(self.viewport().mapToGlobal(position))
|
|
|
|
def remove_local_tx(self, delete_tx):
|
|
to_delete = {delete_tx}
|
|
to_delete |= self.wallet.get_depending_transactions(delete_tx)
|
|
question = _("Are you sure you want to remove this transaction?")
|
|
if len(to_delete) > 1:
|
|
question = _(
|
|
"Are you sure you want to remove this transaction and {} child transactions?".format(len(to_delete) - 1)
|
|
)
|
|
answer = QMessageBox.question(self.parent, _("Please confirm"), question, QMessageBox.Yes, QMessageBox.No)
|
|
if answer == QMessageBox.No:
|
|
return
|
|
for tx in to_delete:
|
|
self.wallet.remove_transaction(tx)
|
|
self.wallet.save_transactions(write=True)
|
|
# need to update at least: history_list, utxo_list, address_list
|
|
self.parent.need_update.set()
|
|
|
|
def onFileAdded(self, fn):
|
|
try:
|
|
with open(fn) as f:
|
|
tx = self.parent.tx_from_text(f.read())
|
|
self.parent.save_transaction_into_wallet(tx)
|
|
except IOError as e:
|
|
self.parent.show_error(e)
|
|
|
|
def export_history_dialog(self):
|
|
d = WindowModalDialog(self, _('Export History'))
|
|
d.setMinimumSize(400, 200)
|
|
vbox = QVBoxLayout(d)
|
|
defaultname = os.path.expanduser('~/electrum-bitcoinprivate-history.csv')
|
|
select_msg = _('Select file to export your wallet transactions to')
|
|
hbox, filename_e, csv_button = filename_field(self, self.config, defaultname, select_msg)
|
|
vbox.addLayout(hbox)
|
|
vbox.addStretch(1)
|
|
hbox = Buttons(CancelButton(d), OkButton(d, _('Export')))
|
|
vbox.addLayout(hbox)
|
|
#run_hook('export_history_dialog', self, hbox)
|
|
self.update()
|
|
if not d.exec_():
|
|
return
|
|
filename = filename_e.text()
|
|
if not filename:
|
|
return
|
|
try:
|
|
self.do_export_history(self.wallet, filename, csv_button.isChecked())
|
|
except (IOError, os.error) as reason:
|
|
export_error_label = _("Electrum-bitcoinprivate was unable to produce a transaction export.")
|
|
self.parent.show_critical(export_error_label + "\n" + str(reason), title=_("Unable to export history"))
|
|
return
|
|
self.parent.show_message(_("Your wallet history has been successfully exported."))
|
|
|
|
def do_export_history(self, wallet, fileName, is_csv):
|
|
history = self.transactions
|
|
lines = []
|
|
for item in history:
|
|
if is_csv:
|
|
lines.append([item['txid'], item.get('label', ''), item['confirmations'], item['value'], item['date']])
|
|
else:
|
|
lines.append(item)
|
|
with open(fileName, "w+", encoding='utf-8') as f:
|
|
if is_csv:
|
|
import csv
|
|
transaction = csv.writer(f, lineterminator='\n')
|
|
transaction.writerow(["transaction_hash","label", "confirmations", "value", "timestamp"])
|
|
for line in lines:
|
|
transaction.writerow(line)
|
|
else:
|
|
from electrum_bitcoinprivate.util import json_encode
|
|
f.write(json_encode(history))
|