class PluginObject(object):
tabName = 'Exchange Rates'
maxVersion = '0.92'
#############################################################################
def __init__(self, main):
self.main = main
##########################################################################
##### Display the conversion values based on the Coinbase API
self.lastPriceFetch = 0
self.lblHeader = QRichLabel(tr("""<b>Tracking buy and sell prices on
Coinbase every 60 seconds</b>"""), doWrap=False)
self.lblLastTime = QRichLabel('', doWrap=False)
self.lblSellLabel = QRichLabel(tr('Coinbase <b>Sell</b> Price (USD):'), doWrap=False)
self.lblBuyLabel = QRichLabel(tr('Coinbase <b>Buy</b> Price (USD):'), doWrap=False)
self.lblSellPrice = QRichLabel('<Not Available>')
self.lblBuyPrice = QRichLabel('<Not Available>')
self.lastSellStr = ''
self.lastBuyStr = ''
self.btnUpdate = QPushButton(tr('Check Now'))
self.main.connect(self.btnUpdate, SIGNAL('clicked()'), self.checkUpdatePrice)
##########################################################################
##### A calculator for converting prices between USD and BTC
lblCalcTitle = QRichLabel(tr("""Convert between USD and BTC using
Coinbase sell price"""), hAlign=Qt.AlignHCenter, doWrap=False)
self.edtEnterUSD = QLineEdit()
self.edtEnterBTC = QLineEdit()
self.lblEnterUSD1 = QRichLabel('$')
self.lblEnterUSD2 = QRichLabel('USD')
self.lblEnterBTC = QRichLabel('BTC')
btnClear = QPushButton('Clear')
self.main.connect(self.edtEnterUSD, SIGNAL('textEdited(QString)'), self.updateCalcBTC)
self.main.connect(self.edtEnterBTC, SIGNAL('textEdited(QString)'), self.updateCalcUSD)
def clearCalc():
self.edtEnterUSD.setText('')
self.edtEnterBTC.setText('')
self.main.connect(btnClear, SIGNAL('clicked()'), clearCalc)
frmCalcMid = makeHorizFrame( [self.lblEnterUSD1,
self.edtEnterUSD,
self.lblEnterUSD2,
'Stretch',
self.edtEnterBTC,
self.lblEnterBTC])
frmCalcClear = makeHorizFrame(['Stretch', btnClear, 'Stretch'])
frmCalc = makeVertFrame([lblCalcTitle, frmCalcMid, frmCalcClear], STYLE_PLAIN)
##########################################################################
##### A table showing you the total balance of each wallet in USD and BTC
lblWltTableTitle = QRichLabel(tr("Wallet balances converted to USD"),
doWrap=False, hAlign=Qt.AlignHCenter)
numWallets = len(self.main.walletMap)
self.wltTable = QTableWidget(self.main)
self.wltTable.setRowCount(numWallets)
self.wltTable.setColumnCount(4)
self.wltTable.horizontalHeader().setStretchLastSection(True)
self.wltTable.setMinimumWidth(600)
##########################################################################
##### Setup the main layout for the tab
mainLayout = QGridLayout()
i=0
mainLayout.addWidget(self.lblHeader, i,0, 1,3)
i+=1
mainLayout.addItem(QSpacerItem(15,15), i,0)
mainLayout.addWidget(self.lblSellLabel, i,1)
mainLayout.addWidget(self.lblSellPrice, i,2)
i+=1
mainLayout.addItem(QSpacerItem(15,15), i,0)
mainLayout.addWidget(self.lblBuyLabel, i,1)
mainLayout.addWidget(self.lblBuyPrice, i,2)
i+=1
mainLayout.addWidget(self.lblLastTime, i,0, 1,2)
mainLayout.addWidget(self.btnUpdate, i,2)
i+=1
mainLayout.addItem(QSpacerItem(20,20), i,0)
i+=1
mainLayout.addWidget(frmCalc, i,0, 1,3)
i+=1
mainLayout.addItem(QSpacerItem(30,30), i,0)
i+=1
mainLayout.addWidget(lblWltTableTitle, i,0, 1,3)
i+=1
mainLayout.addWidget(self.wltTable, i,0, 1,3)
mainLayout.setColumnStretch(0,0)
mainLayout.setColumnStretch(1,1)
mainLayout.setColumnStretch(2,1)
tabWidget = QWidget()
tabWidget.setLayout(mainLayout)
frmH = makeHorizFrame(['Stretch', tabWidget, 'Stretch'])
frm = makeVertFrame(['Space(20)', frmH, 'Stretch'])
# Now set the scrollarea widget to the layout
self.tabToDisplay = QScrollArea()
self.tabToDisplay.setWidgetResizable(True)
self.tabToDisplay.setWidget(frm)
#############################################################################
def getTabToDisplay(self):
return self.tabToDisplay
#############################################################################
def addCommasToPrice(self, pstr):
dispStr = pstr.strip().split('.')[0]
dispStr = ','.join([dispStr[::-1][3*i:3*(i+1)][::-1] \
for i in range((len(dispStr)-1)/3+1)][::-1])
if '.' in pstr:
dispStr = dispStr + '.' + pstr.split('.')[1]
return dispStr
#############################################################################
def fetchFormattedPrice(self, url):
sock = urllib2.urlopen(url)
value = ast.literal_eval(sock.read())['subtotal']['amount']
return self.addCommasToPrice(value)
#############################################################################
def checkUpdatePrice(self):
urlBase = 'http://coinbase.com/api/v1/prices/'
urlSell = urlBase + 'sell'
urlBuy = urlBase + 'buy'
try:
self.lastSellStr = self.fetchFormattedPrice(urlSell)
self.lastBuyStr = self.fetchFormattedPrice(urlBuy)
self.lblSellPrice.setText('<b><font color="%s">$%s</font> / BTC</b>' % \
(htmlColor('TextBlue'), self.lastSellStr))
self.lblBuyPrice.setText( '<b><font color="%s">$%s</font> / BTC</b>' % \
(htmlColor('TextBlue'), self.lastBuyStr))
self.lastPriceFetch = RightNow()
self.updateLastTimeStr()
self.updateWalletTable()
self.updateCalcUSD(self.edtEnterBTC.text())
except:
#LOGEXCEPT('Failed to fetch price data from %s' % urlBase)
pass
#############################################################################
def updateLastTimeStr(self):
secs = RightNow() - self.lastPriceFetch
tstr = 'Less than 1 min'
if secs > 60:
tstr = secondsToHumanTime(secs)
self.lblLastTime.setText(tr("""<font color="%s">Last updated:
%s ago</font>""") % (htmlColor('DisableFG'), tstr))
#############################################################################
def injectGoOnlineFunc(self, topBlock):
self.checkUpdatePrice()
#############################################################################
def injectHeartbeatAlwaysFunc(self):
# Check the price every 60 seconds, update widgets
self.updateLastTimeStr()
if RightNow() < self.lastPriceFetch+60:
return
self.lastPriceFetch = RightNow()
self.checkUpdatePrice()
#############################################################################
def updateCalcUSD(self, newBTCVal):
try:
convertVal = float(self.lastSellStr.replace(',',''))
usdVal = convertVal * float(newBTCVal.replace(',',''))
self.edtEnterUSD.setText(self.addCommasToPrice('%0.2f' % usdVal))
except:
self.edtEnterUSD.setText('')
#############################################################################
def updateCalcBTC(self, newUSDVal):
try:
convertVal = float(self.lastSellStr.replace(',',''))
btcVal = float(newUSDVal.replace(',','')) / convertVal
self.edtEnterBTC.setText(self.addCommasToPrice('%0.8f' % btcVal))
except:
self.edtEnterBTC.setText('')
#############################################################################
def updateWalletTable(self):
numWallets = len(self.main.walletMap)
self.wltTable.setRowCount(numWallets)
self.wltTable.setColumnCount(4)
row = 0
for wltID,wltObj in self.main.walletMap.iteritems():
wltValueBTC = '(...)'
wltValueUSD = '(...)'
if TheBDM.getBDMState()=='BlockchainReady':
convertVal = float(self.lastSellStr.replace(',',''))
wltBal = wltObj.getBalance('Total')
wltValueBTC = coin2str(wltBal, maxZeros=2)
wltValueUSD = '$' + self.addCommasToPrice('%0.2f' % (wltBal*convertVal/1e8))
rowItems = []
rowItems.append(QTableWidgetItem(wltID))
rowItems.append(QTableWidgetItem(wltObj.labelName))
rowItems.append(QTableWidgetItem(wltValueBTC))
rowItems.append(QTableWidgetItem(wltValueUSD))
rowItems[-2].setTextAlignment(Qt.AlignRight)
rowItems[-1].setTextAlignment(Qt.AlignRight)
rowItems[-2].setFont(GETFONT('Fixed', 10))
rowItems[-1].setFont(GETFONT('Fixed', 10))
for i,item in enumerate(rowItems):
self.wltTable.setItem(row, i, item)
item.setFlags(Qt.NoItemFlags)
self.wltTable.setHorizontalHeaderItem(0, QTableWidgetItem(tr('Wallet ID')))
self.wltTable.setHorizontalHeaderItem(1, QTableWidgetItem(tr('Wallet Name')))
self.wltTable.setHorizontalHeaderItem(2, QTableWidgetItem(tr('BTC Balance')))
self.wltTable.setHorizontalHeaderItem(3, QTableWidgetItem(tr('USD ($) Value')))
self.wltTable.verticalHeader().hide()
row += 1
class VerifyOfflinePackageDialog(ArmoryDialog):
def __init__(self, parent, main):
super(VerifyOfflinePackageDialog, self).__init__(parent, main)
self.main = main
layout = QVBoxLayout(self)
load = QGroupBox(tr("Load Signed Package"), self)
layout.addWidget(load)
layoutload = QVBoxLayout()
load.setLayout(layoutload)
self.loadFileButton = QPushButton(tr("Select file to verify..."), load)
layoutload.addWidget(self.loadFileButton)
self.connect(self.loadFileButton, SIGNAL('clicked()'), self.load)
self.lblVerified = QRichLabel('', hAlign=Qt.AlignHCenter, doWrap=False)
layout.addWidget(self.lblVerified)
save = QGroupBox(tr("Save Verified Package"), self)
layout.addItem(QSpacerItem(10, 10))
layout.addWidget(save)
layoutsave = QVBoxLayout()
save.setLayout(layoutsave)
self.saveFileButton = QPushButton(tr("Select file to save to..."),
load)
self.saveFileButton.setEnabled(False)
layoutsave.addWidget(self.saveFileButton)
self.connect(self.saveFileButton, SIGNAL('clicked()'), self.save)
self.setWindowTitle('Verify Signed Package')
def load(self):
self.fileData = None
#self.fromfile = QFileDialog.getOpenFileName(self, tr("Load file to verify"), "", tr("Armory Signed Packages (*.signed)"))
self.fromfile = self.main.getFileLoad(tr('Load file to Verify'),\
['Armory Signed Packages (*.signed)'])
if len(self.fromfile) == 0:
return
df = open(self.fromfile, "rb")
allfile = df.read()
df.close()
magicstart = "START_OF_SIGNATURE_SECTION"
magicend = "END_OF_SIGNATURE_SECTION"
if 0 != allfile.find(magicstart, 0,
1024 * 1024 * 4): # don't search past 4MiB
QMessageBox.warning(self, tr("Invalid File"),
tr("This file is not a signed package"))
return
end = allfile.find(magicend, 0,
1024 * 1024 * 4) # don't search past 4MiB
if -1 == end: # don't search past 4MiB
QMessageBox.warning(
self, tr("Invalid File"),
tr("The end of the signature in the file could not be found"))
signatureData = allfile[len(magicstart):end]
fileData = allfile[end + len(magicend):]
print "All:", end, end + len(magicend), len(fileData), len(allfile)
allsigs = downloadLinkParser(filetext=signatureData).downloadMap
res = binary_to_hex(sha256(fileData))
good = False
url = None
print "Hash of package file: ", res
# simply check if any of the hashes match
for pack in allsigs.itervalues():
for packver in pack.itervalues():
for packos in packver.itervalues():
for packosver in packos.itervalues():
for packosarch in packosver.itervalues():
okhash = packosarch[1]
if okhash == res:
url = packosarch[0]
good = True
if good:
self.saveFileButton.setEnabled(True)
self.fileData = fileData
self.fileName = os.path.basename(url)
self.lblVerified.setText(
tr("""<font color="%s"><b>Signature is
Valid!</b></font>""") % htmlColor('TextGreen'))
reply = QMessageBox.warning(self, tr("Signature Valid"), tr("""
The downloaded file has a <b>valid</b> signature from
<font color="%s"><b>Armory Technologies, Inc.</b></font>, and is
safe to install.
<br><br>
Would you like to overwrite the original file with the extracted
installer? If you would like to save it to a new location, click
"No" and then use the "Save Verified Package" button to select
a new save location.""") % htmlColor('TextGreen'), \
QMessageBox.Yes | QMessageBox.No)
if reply == QMessageBox.Yes:
newFile = self.fromfile
if newFile.endswith('.signed'):
newFile = self.fromfile[:-7]
LOGINFO('Saving installer to: ' + newFile)
with open(newFile, 'wb') as df:
df.write(self.fileData)
if os.path.exists(newFile):
LOGINFO('Removing original file: ' + self.fromfile)
os.remove(self.fromfile)
QMessageBox.warning(
self, tr('Saved!'),
tr("""
The installer was successfully extracted and saved to the
following location:
<br><br>
%s""") % newFile, QMessageBox.Ok)
else:
self.saveFileButton.setEnabled(False)
self.lblVerified.setText(
tr("""<font color="%s">Invalid signature
on loaded file!</font>""") % htmlColor('TextRed'))
QMessageBox.warning(self, tr("Signature failure"), \
tr("This file has an invalid signature"))
def save(self):
tofile = QFileDialog.getSaveFileName(self, tr("Save confirmed package"), \
QDir.homePath() + "/" + self.fileName)
if len(tofile) == 0:
return
df = open(tofile, "wb")
df.write(self.fileData)
df.close()
class VerifyOfflinePackageDialog(QDialog):
def __init__(self, parent, main):
super(VerifyOfflinePackageDialog, self).__init__(parent)
self.main = main
layout = QVBoxLayout(self)
load = QGroupBox(tr("Load Signed Package"), self)
layout.addWidget(load)
layoutload = QVBoxLayout()
load.setLayout(layoutload)
self.loadFileButton = QPushButton(tr("Select file to verify..."), load);
layoutload.addWidget(self.loadFileButton)
self.connect(self.loadFileButton, SIGNAL('clicked()'), self.load)
self.lblVerified = QRichLabel('', hAlign=Qt.AlignHCenter, doWrap=False)
layout.addWidget(self.lblVerified)
save = QGroupBox(tr("Save Verified Package"), self)
layout.addItem(QSpacerItem(10,10))
layout.addWidget(save)
layoutsave = QVBoxLayout()
save.setLayout(layoutsave)
self.saveFileButton = QPushButton(tr("Select file to save to..."), load);
self.saveFileButton.setEnabled(False)
layoutsave.addWidget(self.saveFileButton)
self.connect(self.saveFileButton, SIGNAL('clicked()'), self.save)
def load(self):
self.fileData = None
#self.fromfile = QFileDialog.getOpenFileName(self, tr("Load file to verify"), "", tr("Armory Signed Packages (*.signed)"))
self.fromfile = self.main.getFileLoad(tr('Load file to Verify'),\
['Armory Signed Packages (*.signed)'])
if len(self.fromfile)==0:
return
df = open(self.fromfile, "rb")
allfile = df.read()
df.close()
magicstart="START_OF_SIGNATURE_SECTION"
magicend="END_OF_SIGNATURE_SECTION"
if 0 != allfile.find(magicstart, 0, 1024*1024*4): # don't search past 4MiB
QMessageBox.warning(self, tr("Invalid File"), tr("This file is not a signed package"))
return
end = allfile.find(magicend, 0, 1024*1024*4) # don't search past 4MiB
if -1 == end: # don't search past 4MiB
QMessageBox.warning(self, tr("Invalid File"), tr("The end of the signature in the file could not be found"))
signatureData = allfile[len(magicstart):end]
fileData = allfile[end+len(magicend):]
print "All:",end, end+len(magicend), len(fileData), len(allfile)
allsigs = downloadLinkParser(filetext=signatureData).downloadMap
res = binary_to_hex(sha256(fileData))
good=False
url=None
print "Hash of package file: ", res
# simply check if any of the hashes match
for pack in allsigs.itervalues():
for packver in pack.itervalues():
for packos in packver.itervalues():
for packosver in packos.itervalues():
for packosarch in packosver.itervalues():
okhash = packosarch[1]
if okhash == res:
url = packosarch[0]
good=True
if good:
self.saveFileButton.setEnabled(True)
self.fileData = fileData
self.fileName = os.path.basename(url)
self.lblVerified.setText(tr("""<font color="%s"><b>Signature is
Valid!</b></font>""") % htmlColor('TextGreen'))
reply = QMessageBox.warning(self, tr("Signature Valid"), tr("""
The downloaded file has a <b>valid</b> signature from
<font color="%s"><b>Armory Technologies, Inc.</b></font>, and is
safe to install.
<br><br>
Would you like to overwrite the original file with the extracted
installer? If you would like to save it to a new location, click
"No" and then use the "Save Verified Package" button to select
a new save location.""") % htmlColor('TextGreen'), \
QMessageBox.Yes | QMessageBox.No)
if reply==QMessageBox.Yes:
newFile = self.fromfile
if newFile.endswith('.signed'):
newFile = self.fromfile[:-7]
LOGINFO('Saving installer to: ' + newFile)
with open(newFile, 'wb') as df:
df.write(self.fileData)
if os.path.exists(newFile):
LOGINFO('Removing original file: ' + self.fromfile)
os.remove(self.fromfile)
QMessageBox.warning(self, tr('Saved!'), tr("""
The installer was successfully extracted and saved to the
following location:
<br><br>
%s""") % newFile, QMessageBox.Ok)
else:
self.saveFileButton.setEnabled(False)
self.lblVerified.setText(tr("""<font color="%s">Invalid signature
on loaded file!</font>""") % htmlColor('TextRed'))
QMessageBox.warning(self, tr("Signature failure"), \
tr("This file has an invalid signature"))
def save(self):
tofile = QFileDialog.getSaveFileName(self, tr("Save confirmed package"), \
QDir.homePath() + "/" + self.fileName)
if len(tofile)==0:
return
df = open(tofile, "wb")
df.write(self.fileData)
df.close()
