def create_tor(): from ansi_management import (warning, success, error, info, clear_screen, bold, muted, yellow, blue) from config import Config # ---------------------------------------------- # Test Tor # ---------------------------------------------- with yaspin(text="Testing Tor", color="cyan") as spinner: from connections import test_tor tor = test_tor() if tor['status']: logging.info(success("Tor Connected")) spinner.ok("✅ ") spinner.text = success(" Tor Connected [Success]") print("") return (tor) else: logging.error(error("Could not connect to Tor")) spinner.fail("💥 ") spinner.text = warning(" Tor NOT connected [ERROR]") print(error(" Could not connect to Tor.")) print( info( " [i] If you have a Tor Browser installed try opening (leave it running) and try again." )) print( info(" [i] If you are running Linux try the command: ") + yellow('service tor start')) print( info( " or download Tor at: https://www.torproject.org/download/" )) print("")
def create_tor(): # ---------------------------------------------- # Test Tor # ---------------------------------------------- with yaspin(text="Testing Tor", color="cyan") as spinner: tor = test_tor() if tor['status']: logging.info(success("Tor Connected")) spinner.ok("✅ ") spinner.write(success(" Tor Connected [Success]")) print("") return (tor) else: logging.error(error("Could not connect to Tor")) spinner.fail("💥 ") spinner.write(warning(" Tor NOT connected [ERROR]")) print( error( " Could not connect to Tor. WARden requires Tor to run. Quitting..." )) print( info( " Download Tor at: https://www.torproject.org/download/" )) print("") exit()
def data_large_price(): from node_warden import load_config config = load_config(quiet=True) ft_config = config['MAIN'] font = ft_config.get('large_text_font') btc = btc_price_data() try: btc_price = cleanfloat(btc['DISPLAY']['BTC']['USD']['PRICE']) except Exception: return (error(' >> Error getting price data. Retrying...')) custom_fig = pyfiglet.Figlet(font=font) return_fig = custom_fig.renderText('$ ' + jformat(btc_price, 0)) return_fig = yellow(return_fig) chg_str = btc['DISPLAY']['BTC']['USD']['CHANGEPCTDAY'] chg = cleanfloat(chg_str) msg = '\n' if chg >= 0: msg += success(f'24hr Change: +{chg_str}%\n') if chg > 5: msg += (info("[NgU] ") + muted(f"Looks like Bitcoin is pumping ") + emoji.emojize(":rocket:")) if chg < 0: msg += error(f'24hr Change: {chg_str}%\n') if chg < -5: msg += muted( f"Bitcoin dropping? Buy the dip!\nTime to stack some sats. ") return_fig = muted(return_fig) return_fig += msg return (return_fig)
def check_for_pump(_loop, _data): try: btc = btc_price_data() btc_price = btc['DISPLAY']['BTC']['USD']['PRICE'] chg_str = btc['DISPLAY']['BTC']['USD']['CHANGEPCTDAY'] chg = cleanfloat(chg_str) if chg > 5: logging.info( info("[NgU] ") + muted(f"Looks like Bitcoin is pumping ") + emoji.emojize(":rocket:") + yellow(f' {btc_price}') + success(f' +{chg_str}%')) if chg < -5: logging.info( info("[NgU] ") + muted( f"Looks like Bitcoin is dropping. Time to stack some sats. " ) + yellow(f' {btc_price}') + error(f' {chg_str}%')) except Exception: pass main_loop.set_alarm_in(300, check_for_pump)
def data_random_satoshi(): from node_warden import load_config config = load_config(quiet=True) url = config['QUOTES'].get('url') try: quotes = tor_request(url).json() except Exception: return (error(' >> Error contacting server. Retrying... ')) quote = quotes[randrange(len(quotes))] return_str = info(f"Satoshi Quotes | Subject: {quote['category']}\n") return_str += muted(f"{quote['date']} on {quote['medium']}\n") return_str += yellow(f"{quote['text']} \n\n") return_str += muted("Source: Nakamoto Institute") return (return_str)
def data_logger(): from node_warden import debug_file lines = 8 log_txt = tail(debug_file, lines) return_str = [] for element in log_txt: if 'INFO' in element: return_str.append(info(element)) elif 'ERROR' in log_txt: return_str.append(error(element)) elif 'WARN' in log_txt: return_str.append(warning(element)) else: return_str.append((element)) return_str = ''.join(return_str) return return_str
def data_logger(): from node_warden import debug_file try: lines = 8 log_txt = tail(debug_file, lines) return_str = [] for element in log_txt: if 'INFO' in element: return_str.append(info(element)) elif 'ERROR' in log_txt: return_str.append(error(element)) elif 'WARN' in log_txt: return_str.append(warning(element)) else: return_str.append((element)) return_str = ''.join(return_str) except Exception: return yellow('>> Error Loading Log Messages. Retying...') return return_str
def data_btc_price(): from node_warden import launch_logger launch_logger() from node_warden import load_config config = load_config(quiet=True) fx_config = config['CURRENCIES'] currencies = ast.literal_eval(fx_config.get('fx_list')) primary_fx = ast.literal_eval(fx_config.get('primary_fx')) price_data = multiple_price_grab('BTC', ','.join(currencies)) # Get prices in different currencies tabs = [] btc_usd_price = 0 for fx in currencies: try: price_str = price_data['DISPLAY']['BTC'][fx]['PRICE'] chg_str = price_data['DISPLAY']['BTC'][fx]['CHANGEPCTDAY'] high = price_data['DISPLAY']['BTC'][fx]['HIGHDAY'] low = price_data['DISPLAY']['BTC'][fx]['LOWDAY'] market = muted(price_data['DISPLAY']['BTC'][fx]['LASTMARKET']) try: chg = float(chg_str) if chg >= 0: chg_str = success('+' + chg_str + '%') elif chg < 0: chg_str = error(chg_str + '%') except Exception: chg_str = muted(chg_str + '%') if fx == 'USD': btc_usd_price = cleanfloat(price_str) if fx == primary_fx: fx = info(fx) tabs.append( [u' ' + fx, price_str, chg_str, low + ' - ' + high, market]) except Exception as e: tabs.append(['error: ' + str(e)]) tabs = tabulate( tabs, headers=['Fiat', 'Price', '% change', '24h Range', 'Source'], colalign=["center", "right", "right", "center", "right"]) # GBTC gbtc_config = config['STOCKS'] try: if gbtc_config.getboolean('GBTC_enabled'): tabs += '\n\n' gbtc_url = 'https://www.alphavantage.co/query?function=GLOBAL_QUOTE&symbol=GBTC&apikey=DDC232JDH' gbtc_data = tor_request(gbtc_url).json()['Global Quote'] gbtc_tabs = [] GBTC_shares = gbtc_config.getfloat('gbtc_shares') fairvalue, premium = (GBTC_premium(float(gbtc_data['05. price']), btc_usd_price, GBTC_shares)) if premium * 1 > 0: premium = success('+' + jformat(premium, 2, 0.01) + '%') elif premium * 1 < 0: premium = error(jformat(premium, 2, 0.01) + '%') fairvalue = jformat(fairvalue, 2) chg_str = gbtc_data['10. change percent'] try: chg = cleanfloat(chg_str) if chg > 0: chg_str = success('+' + jformat(chg, 2) + ' %') elif chg < 0: chg_str = error(jformat(chg, 2) + ' %') except Exception: chg_str = muted(chg_str) gbtc_tabs.append([ 'GBTC', gbtc_data['05. price'], chg_str, gbtc_data['04. low'] + ' - ' + gbtc_data['03. high'], premium, fairvalue, gbtc_data['07. latest trading day'] ]) gbtc_tabs = tabulate(gbtc_tabs, headers=[ 'Ticker', 'Price', '% change', '24h Range', 'Premium', 'Fair Value', 'Last Update' ], colalign=[ "center", "right", "right", "center", "right", "right", "right" ]) tabs += gbtc_tabs except Exception as e: er_st = error(f' Error getting GBTC data: {e}') tabs += er_st tabs += ( f"\n\n Last Refresh on: {info(datetime.now().strftime('%H:%M:%S'))}") return tabs
def data_mempool(): from node_warden import load_config from node_warden import launch_logger launch_logger() config = load_config(quiet=True) mp_config = config['MEMPOOL'] url = mp_config.get('url') tabs = [] # Get recommended fees mp_fee = tor_request(url + '/api/v1/fees/recommended').json() tabs = list(mp_fee.values()) tabs = [[str(x) + ' sats/Vb' for x in tabs]] tabs = tabulate(tabs, headers=["Fastest Fee", "30 min fee", "1 hour fee"], colalign=["center", "center", "center"]) try: block_height = tor_request(url + '/api/blocks/tip/height').json() except Exception: return (error(f' >> Error getting data from {url}. Retrying...')) # Save the latest block height saved_block = pickle_it(action='load', filename='block.pkl') if (saved_block != block_height) and ( config['MEMPOOL'].getboolean('block_found_sound')): # Block found play sound try: engine = pyttsx3.init() engine.setProperty('rate', 270) engine.say(config['MEMPOOL'].get('block_found_txt')) engine.runAndWait() except Exception: pass logging.info( info('[MEMPOOL] ') + success("A new Bitcoin Block was just found. ") + yellow("'Tick. Tock. Next block.'")) pickle_it(action='save', filename='block.pkl', data=block_height) block_txt = success(f' Block Height: {jformat(block_height, 0)}\n\n') tabs = block_txt + info(' Mempool Fee Estimates: \n') + tabs try: mp_blocks = tor_request(url + '/api/blocks').json() except Exception: return (error(" >> Error getting Mempool data. Retrying...")) mp_tabs = [] gradient_color = 0 for block in mp_blocks: mp_tabs.append([ time_ago(block['timestamp']), jformat(block['height'], 0), jformat(block['tx_count'], 0), jformat(block['size'], 2, 1000000) + ' MB' ]) gradient_color += 1 mp_tabs = tabulate(mp_tabs, headers=[" Time", "Height", "Tx Count", "Size"], colalign=["right", "center", "center", "right"]) tabs += info('\n\n Latest Blocks: \n') + mp_tabs tabs += muted(f"\n\n Source: {url} {success('[Tor Request]')}\n") return tabs
def data_btc_price(): from node_warden import launch_logger launch_logger() from node_warden import load_config config = load_config(quiet=True) fx_config = config['CURRENCIES'] currencies = ast.literal_eval(fx_config.get('fx_list')) primary_fx = ast.literal_eval(fx_config.get('primary_fx')) price_data = multiple_price_grab('BTC', ','.join(currencies)) # Get prices in different currencies tabs = [] btc_usd_price = 0 for fx in currencies: try: price_str = price_data['DISPLAY']['BTC'][fx]['PRICE'] chg_str = price_data['DISPLAY']['BTC'][fx]['CHANGEPCTDAY'] high = price_data['DISPLAY']['BTC'][fx]['HIGHDAY'] low = price_data['DISPLAY']['BTC'][fx]['LOWDAY'] market = muted(price_data['DISPLAY']['BTC'][fx]['LASTMARKET']) try: chg = float(chg_str) if chg >= 0: chg_str = success('+' + chg_str + '%') elif chg < 0: chg_str = error(chg_str + '%') except Exception: chg_str = muted(chg_str + '%') if fx == 'USD': btc_usd_price = cleanfloat(price_str) if fx == primary_fx: fx = info(fx) tabs.append( [u' ' + fx, price_str, chg_str, low + ' - ' + high, market]) except Exception as e: tabs.append(['error: ' + str(e)]) if tabs == []: return ( error(f' >> Error getting data from CryptoCompare. Retrying...')) try: tabs = tabulate( tabs, headers=['Fiat', 'Price', '% change', '24h Range', 'Source'], colalign=["center", "right", "right", "center", "right"]) except Exception: return ( error(f' >> Error getting data from CryptoCompare. Retrying...')) # GBTC gbtc_config = config['STOCKS'] try: if gbtc_config.getboolean('GBTC_enabled'): tabs += '\n\n' gbtc_url = 'https://finnhub.io/api/v1/quote?symbol=GBTC&token=bvfhuqv48v6rhdtvnks0' gbtc_data = tor_request(gbtc_url).json() gbtc_tabs = [] GBTC_shares = gbtc_config.getfloat('gbtc_shares') fairvalue, premium = (GBTC_premium((gbtc_data['c']), btc_usd_price, GBTC_shares)) if premium * 1 > 0: premium = success('+' + jformat(premium, 2, 0.01) + '%') elif premium * 1 < 0: premium = error(jformat(premium, 2, 0.01) + '%') fairvalue = jformat(fairvalue, 2) chg_str = gbtc_data['c'] / gbtc_data['pc'] try: chg = (chg_str) if chg > 0: chg_str = success('+' + jformat(chg, 2) + ' %') elif chg < 0: chg_str = error(jformat(chg, 2) + ' %') except Exception: chg_str = muted(chg_str) gbtc_tabs.append([ 'GBTC', jformat(gbtc_data['c'], 2), chg_str, jformat(gbtc_data['l'], 2) + ' - ' + jformat(gbtc_data['h'], 2), premium, fairvalue, time_ago(gbtc_data['t']) ]) gbtc_tabs = tabulate(gbtc_tabs, headers=[ 'Ticker', 'Price', '% change', '24h Range', 'Premium', 'Fair Value', 'Last Update' ], colalign=[ "center", "right", "right", "center", "right", "right", "right" ]) tabs += gbtc_tabs except Exception as e: er_st = error(f' Error getting GBTC data: {e}') tabs += er_st tabs += ( f"\n\n Last Refresh on: {info(datetime.now().strftime('%H:%M:%S'))}") return tabs
def main_dashboard(config, tor, spinner): try: refresh_interval = int(config['MAIN'].get('refresh')) except Exception: refresh_interval = 5 running_jobs = { 'btc': { 'workers': 0 }, 'tor': { 'workers': 0 }, 'login': { 'workers': 0 }, 'mp': { 'workers': 0 }, 'logger': { 'workers': 0 } } palette = [('titlebar', 'dark green', ''), ('refresh button', 'dark green,bold', ''), ('quit button', 'dark green', ''), ('getting quote', 'dark blue', ''), ('headers', 'white,bold', ''), ('change ', 'dark green', ''), ('change negative', 'dark red', '')] def exit_on_q(key): if key in ('q', 'Q'): raise urwid.ExitMainLoop() def update_header(layout, message=None, message_type=None): # Create Header refresh_time = datetime.now().strftime('%H:%M:%S') txt = u' WARden Node Edition (Version: ' + version( ) + ') | Last Refresh on: ' + refresh_time if message: txt += ' | ' + message header_text = urwid.Text(txt) header = urwid.AttrMap(header_text, 'titlebar') layout.header = header # Draw empty dashboard spinner.stop() menu = urwid.Text([u'Press (', ('quit button', u'Q'), u') to quit.']) # Class to Create URWID box window to receive data class Box: def __init__(self, loader_text=None, valign='top', top=1, bottom=1, left=1, right=1, height=None): self.loader_text = loader_text self.valign = valign self.top = top self.bottom = bottom self.left = left self.right = right self.height = height self.text = urwid.Text(self.loader_text) self.filler = urwid.Filler(self.text, valign=self.valign, top=self.top, bottom=self.bottom) self.v_padding = urwid.Padding(self.filler, left=self.left, right=self.right) self.line_box = urwid.LineBox(self.v_padding) self.box_size = self.height # Create the BTC price box quote_box_size = len(ast.literal_eval( config['CURRENCIES'].get('fx_list'))) + 10 quote_box = Box(loader_text='Loading Prices...', height=quote_box_size).line_box # Create the TOR Box tor_box_size = 9 tor_box = Box(loader_text='Checking Tor Status...', height=tor_box_size).line_box # Create user login Box login_box_size = 12 login_box = Box(loader_text='Loading User Logins...', height=login_box_size).line_box # Create MemPool Box mp_box_size = 24 mp_box = Box(loader_text='Loading Mempool...', height=mp_box_size).line_box # Create Logger Box logger_box_size = 10 logger_box = Box(loader_text='Loading Message Log...', height=logger_box_size).line_box # Create the Satoshi Quotes Box satoshi_box_size = 20 satoshi_box = Box(loader_text='Loading Satoshi Wisdom...', height=satoshi_box_size).line_box # Assemble the widgets header = 'Loading...' log_tor = urwid.Columns([mp_box, urwid.Pile([login_box, tor_box])]) log_tor_size = max(mp_box_size, login_box_size, tor_box_size) bottom_box_size = max(satoshi_box_size, logger_box_size) bottom_box = urwid.Columns([logger_box, satoshi_box]) body_widget = urwid.Pile([(quote_box_size, quote_box), (log_tor_size, log_tor), (bottom_box_size, bottom_box)]) layout = urwid.Frame(header=header, body=body_widget, footer=menu) update_header(layout) # Handle key presses def handle_input(key): if key == 'R' or key == 'r': refresh(main_loop, '') if key == 'Q' or key == 'q': raise urwid.ExitMainLoop() def check_for_pump(_loop, _data): try: btc = btc_price_data() btc_price = btc['DISPLAY']['BTC']['USD']['PRICE'] chg_str = btc['DISPLAY']['BTC']['USD']['CHANGEPCTDAY'] chg = cleanfloat(chg_str) if chg > 5: logging.info( info("[NGU] ") + muted(f"Looks like Bitcoin is pumping ") + emoji.emojize(":rocket:") + yellow(f' {btc_price}') + success(f' +{chg_str}%')) if chg < -5: logging.info( info("[NGU] ") + muted( f"Looks like Bitcoin is dropping. Time to stack some sats. " ) + yellow(f' {btc_price}') + error(f' {chg_str}%')) except Exception: pass main_loop.set_alarm_in(300, check_for_pump) def get_quote(_loop, _data): quote = translate_text_for_urwid(data_random_satoshi()) satoshi_box.base_widget.set_text(quote) main_loop.set_alarm_in(60, get_quote) def refresh(_loop, _data): # Add Background Tasks update_header(layout) main_loop.draw_screen() # Add Background Updates max_workers = 1 # Max number of threads **KEEP AT 1*** # Beware of increasing the max_workers - need to make sure # processes are killed for all workers - CODE IS NOT DOING THAT # right now and will lead to CPU usage blowing up # UPDATER FUNCTIONS - ONE NEEDED PER UPDATE # These run on background as watch pipes def update_btc(read_data): read_data = translate_text_for_urwid(read_data) quote_box.base_widget.set_text(read_data) main_loop.remove_watch_pipe = True running_jobs['btc']['workers'] = 0 for pipe in running_jobs['btc']['pipe']: if pipe != []: pipe.kill() gc.collect() def update_tor(read_data): read_data = translate_text_for_urwid(read_data) tor_box.base_widget.set_text(read_data) running_jobs['tor']['workers'] = 0 for pipe in running_jobs['tor']['pipe']: if pipe != []: pipe.kill() gc.collect() def update_login(read_data): read_data = translate_text_for_urwid(read_data) login_box.base_widget.set_text(read_data) running_jobs['login']['workers'] = 0 for pipe in running_jobs['login']['pipe']: if pipe != []: pipe.kill() gc.collect() def update_mp(read_data): read_data = translate_text_for_urwid(read_data) mp_box.base_widget.set_text(read_data) main_loop.remove_watch_pipe = True running_jobs['mp']['workers'] = 0 for pipe in running_jobs['mp']['pipe']: if pipe != []: pipe.kill() gc.collect() def update_logger(read_data): read_data = translate_text_for_urwid(read_data) logger_box.base_widget.set_text(read_data) main_loop.remove_watch_pipe = True running_jobs['logger']['workers'] = 0 for pipe in running_jobs['logger']['pipe']: if pipe != []: pipe.kill() gc.collect() # Job List Dictionaty job_list = { 'btc': { 'max_workers': 1, 'subprocess': 'python3 data.py data_btc_price', 'updater': update_btc }, 'tor': { 'max_workers': 1, 'subprocess': 'python3 data.py data_tor', 'updater': update_tor }, 'login': { 'max_workers': 1, 'subprocess': 'python3 data.py data_login', 'updater': update_login }, 'mp': { 'max_workers': 1, 'subprocess': 'python3 data.py data_mempool', 'updater': update_mp }, 'logger': { 'max_workers': 1, 'subprocess': 'python3 data.py data_logger', 'updater': update_logger } } for job in job_list.keys(): if running_jobs[job]['workers'] < job_list[job]['max_workers']: running_jobs[job]['workers'] += 1 stdout = main_loop.watch_pipe(job_list[job]['updater']) stderr = main_loop.watch_pipe(job_list[job]['updater']) launch_process = subprocess.Popen(job_list[job]['subprocess'], shell=True, stdout=stdout, stderr=stderr) # Store or create a list to store running_jobs[job].setdefault('pipe', []).append(launch_process) main_loop.set_alarm_in(refresh_interval, refresh) try: main_loop = urwid.MainLoop(layout, palette, unhandled_input=handle_input) main_loop.set_alarm_in(30, check_for_pump) main_loop.set_alarm_in(10, get_quote) main_loop.set_alarm_in(0, refresh) main_loop.run() except Exception as e: # Catch some timeouts - only once logging.error(info('[MAIN] ') + muted('Error: ') + yellow(str(e))) update_header(layout, message=f'Error: {e}') main_dashboard(config, tor, spinner)