119 lines
3.5 KiB
Python
119 lines
3.5 KiB
Python
from PyQt4.QtGui import *
|
|
from electrum.plugins import BasePlugin, hook
|
|
from electrum.i18n import _
|
|
|
|
|
|
import datetime
|
|
from electrum.util import format_satoshis
|
|
from electrum.bitcoin import COIN
|
|
|
|
try:
|
|
import matplotlib.pyplot as plt
|
|
import matplotlib.dates as md
|
|
from matplotlib.patches import Ellipse
|
|
from matplotlib.offsetbox import AnchoredOffsetbox, TextArea, DrawingArea, HPacker
|
|
flag_matlib=True
|
|
except:
|
|
flag_matlib=False
|
|
|
|
|
|
class Plugin(BasePlugin):
|
|
|
|
def is_available(self):
|
|
if flag_matlib:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
@hook
|
|
def export_history_dialog(self, window, hbox):
|
|
wallet = window.wallet
|
|
history = wallet.get_history()
|
|
if len(history) > 0:
|
|
b = QPushButton(_("Preview plot"))
|
|
hbox.addWidget(b)
|
|
b.clicked.connect(lambda: self.do_plot(wallet, history))
|
|
else:
|
|
b = QPushButton(_("No history to plot"))
|
|
hbox.addWidget(b)
|
|
|
|
|
|
def do_plot(self, wallet, history):
|
|
balance_Val=[]
|
|
fee_val=[]
|
|
value_val=[]
|
|
datenums=[]
|
|
unknown_trans = 0
|
|
pending_trans = 0
|
|
counter_trans = 0
|
|
balance = 0
|
|
for item in history:
|
|
tx_hash, confirmations, value, timestamp, balance = item
|
|
if confirmations:
|
|
if timestamp is not None:
|
|
try:
|
|
datenums.append(md.date2num(datetime.datetime.fromtimestamp(timestamp)))
|
|
balance_Val.append(1000.*balance/COIN)
|
|
except [RuntimeError, TypeError, NameError] as reason:
|
|
unknown_trans += 1
|
|
pass
|
|
else:
|
|
unknown_trans += 1
|
|
else:
|
|
pending_trans += 1
|
|
|
|
value_val.append(1000.*value/COIN)
|
|
if tx_hash:
|
|
label, is_default_label = wallet.get_label(tx_hash)
|
|
label = label.encode('utf-8')
|
|
else:
|
|
label = ""
|
|
|
|
|
|
f, axarr = plt.subplots(2, sharex=True)
|
|
|
|
plt.subplots_adjust(bottom=0.2)
|
|
plt.xticks( rotation=25 )
|
|
ax=plt.gca()
|
|
x=19
|
|
test11="Unknown transactions = "+str(unknown_trans)+" Pending transactions = "+str(pending_trans)+" ."
|
|
box1 = TextArea(" Test : Number of pending transactions", textprops=dict(color="k"))
|
|
box1.set_text(test11)
|
|
|
|
|
|
box = HPacker(children=[box1],
|
|
align="center",
|
|
pad=0.1, sep=15)
|
|
|
|
anchored_box = AnchoredOffsetbox(loc=3,
|
|
child=box, pad=0.5,
|
|
frameon=True,
|
|
bbox_to_anchor=(0.5, 1.02),
|
|
bbox_transform=ax.transAxes,
|
|
borderpad=0.5,
|
|
)
|
|
|
|
|
|
ax.add_artist(anchored_box)
|
|
|
|
|
|
plt.ylabel('mBTC')
|
|
plt.xlabel('Dates')
|
|
xfmt = md.DateFormatter('%Y-%m-%d')
|
|
ax.xaxis.set_major_formatter(xfmt)
|
|
|
|
|
|
axarr[0].plot(datenums,balance_Val,marker='o',linestyle='-',color='blue',label='Balance')
|
|
axarr[0].legend(loc='upper left')
|
|
axarr[0].set_title('History Transactions')
|
|
|
|
|
|
xfmt = md.DateFormatter('%Y-%m-%d')
|
|
ax.xaxis.set_major_formatter(xfmt)
|
|
axarr[1].plot(datenums,value_val,marker='o',linestyle='-',color='green',label='Value')
|
|
|
|
|
|
axarr[1].legend(loc='upper left')
|
|
# plt.annotate('unknown transaction = %d \n pending transactions = %d' %(unknown_trans,pending_trans),xy=(0.7,0.05),xycoords='axes fraction',size=12)
|
|
plt.show()
|