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()