mmgen/mmgen-node-tools
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity
mmgen-node-tools/btc-ticker
philemon fbbf21ef86
 Temporarily disable ticker
2017-02-19 08:59:48 +03:00

1029 lines
29 KiB
Python
Executable file
Raw Permalink Blame History

#!/usr/bin/python
# -*- coding: UTF-8 -*-
#
# mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
# Copyright (C)2013-2016 Philemon <mmgen-py@yandex.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
"""
btc-ticker: ticker and alarm clock for mmgen-node-tools
"""
import sys,os,time,subprocess
import threading as th
from collections import OrderedDict
from decimal import Decimal
from mmgen.share import Opts
from mmgen.util import msg,msg_r,die,die_pause
from mmgen.node_tools.Global import *
from mmgen.node_tools.Util import *
from mmgen.node_tools.Sound import *
from mmgen.node_tools.Term import *
die_pause(1,'Ticker temporarily disabled') # DEBUG
from mmgen.color import *
init_color()
quit = False
sound_vol = float(100)
audio_host,repeat_spec = '','3s:5m,1m:30m,5m:1h,30m:1d'
# repeat_spec = '1:5m' # DEBUG
valid_loglevels = OrderedDict([(1,'info'),(2,'data'),(3,'debug')])
logfile = 'ticker_log.txt'
alrm_clock = 'off'
num_digits = 6
proxy = None
fx_host= 'https://finance.yahoo.com'
sounds_dir = nt.system_data_dir+'/audio'
sounds = {
'lo': [sounds_dir+'/ringtone.wav', 95],
'hi': [sounds_dir+'/Positive.wav', 102],
'conn_err': [sounds_dir+'/Rhodes.wav', 105],
'alrm_clock': [sounds_dir+'/Counterpoint.wav', 105],
}
alrm_names = tuple(sounds.keys())
dlock,llock,alock = th.Lock(),th.Lock(),th.Lock()
data_dir = nt.data_dir+'/lib/ticker/'
try: os.makedirs(data_dir)
except: pass
def update_fx(a,b,debug=False):
if debug:
with open('/tmp/ticker.out') as f: text = f.read()
else:
text = get_url('{}/q?s={}{}'.format(fx_host,a,b),gzip_ok=True,proxy=proxy,debug=debug_curl)
if not text: return False
import re
ret = re.split('yfs_l10_{}{}=.+?>'.format(a,b),text,maxsplit=1)[1].split('<',1)[0][:10]
try:
globals()[a+b] = float(ret)
with open(data_dir+'/{}{}.txt'.format(a,b),'wb') as f: f.write(ret+'\n')
return True
except:
return False
def get_fx(a,b):
try:
with open(data_dir+'/{}{}.txt'.format(a,b)) as f:
d = f.read().rstrip()
globals()[a+b] = float(d)
return True
except:
return update_fx(a,b)
with open('/etc/timezone') as f:
os.environ['TZ'] = f.read().rstrip()
class Xch(object): pass
class Src(object): pass
num_xchgs = 2
xchgs = tuple([Xch() for i in range(num_xchgs)])
sources = OrderedDict([('tc','ticker')])
for g in xchgs:
for k in sources:
setattr(g,k,Src())
setattr(getattr(g,k),'desc',sources[k])
for g in [xchgs[0]]:
g.cur = 'USD'
# g.Desc = 'OKCoin USD'
# g.desc = 'okcoin_usd'
# g.desc_short = 'okc'
# g.tc.url = 'https://www.okcoin.com/api/v1/ticker.do?symbol=btc_usd'
g.Desc = 'Gemini'
g.desc = 'gemini'
g.desc_short = 'gem'
g.tc.url = 'https://api.gemini.com/v1/pubticker/btcusd'
g.poll_secs = 60
g.cur_sign = '$'
g.xcur_sign = '¥'
g.fiat_precision = 2
g.hi_alrm = 999999
g.lo_alrm = 1
g.cc_unit = 'BTC'
for g in [xchgs[1]]:
g.cur = 'CNY'
g.Desc = 'OKCoin CNY'
g.desc = 'okcoin_cny'
g.desc_short = 'okc'
g.tc.url = 'https://www.okcoin.cn/api/v1/ticker.do?symbol=btc_cny'
g.poll_secs = 60
g.cur_sign = '¥'
g.xcur_sign = '$'
g.fiat_precision = 1
g.hi_alrm = 999999
g.lo_alrm = 1
g.cc_unit = 'BTC'
# g.cur = 'USD'
# g.Desc = 'BitFinex'
# g.desc = 'bitfinex'
# g.desc_short = 'bfx'
# g.tc.url = 'https://api.bitfinex.com/v1/pubticker/btcusd'
# Gemini - available symbols: btcusd ethbtc ethusd
# g.cur = 'USD'
# g.Desc = 'Gemini'
# g.desc = 'gemini'
# g.desc_short = 'gem'
# g.tc.url = 'https://api.gemini.com/v1/pubticker/btcusd'
opts_data = {
'prog_name': sys.argv[0].split('/')[-1],
'desc': 'Price alarm for Bitcoin exchange',
'usage': '[opts] [<low price alarm> <high price alarm>]',
#-b, --big-number Print big numbers
'options': """
-h, --help Print this help message
-a, --alarm-clock-only Disable ticker, use as alarm clock only
-c, --cur-ticker-only Display only current exchange's ticker price
-d, --debug Debug mode. Use saved HTTP data from files
-D, --debug-curl Debug curl
-l, --log= l Log all data to file '{lf}' at levels 'l' (comma-separated: {lls})
-n, --num-digits= n Display 'n' number of big digits
-p, --poll-intervals=i1[,i2] Poll servers every 'i' seconds (default: '{pi}')
-P, --proxy= x Connect via proxy 'x' (see PROXY EXAMPLES below for format)
-r, --repeat-spec Sleep interval/duration program for the alarm
(default: '{r}')
(see REPEAT SPEC FORMAT below)
-R, --resize-window Resize window to optimum dimensions
-t, --testing Testing mode. Don't execute shell commands
-u, --utc Show times in UTC rather than local time
-v, --volume= n Adjust sound volume by percentage 'n' (default: {v})
-V, --test-volume Test the alarm volume and exit
-x, --xchgs= x,y[,…] Display only exchanges 'x,y[,…]' (comma-separated
list of integers, see CONFIGURED EXCHANGES below)
""".format(
v=int(sound_vol),
pi = ','.join([str(g.poll_secs) for g in xchgs]),
r=repeat_spec,
lf=logfile,
lls=', '.join(['%s=%s'%(i,j) for i,j in valid_loglevels.items()])
),
'notes': '''
REPEAT SPEC FORMAT:
<interval>:<duration>[,<interval>:<duration>[,…]]
For example, '3s:5m,1m:2h,30m:1d' means:
ring alarm every 3 seconds for 5 minutes,
then every minute for 2 hours,
then every 30 minutes for 1 day
PROXY EXAMPLES:
socks5h://localhost:9050 (for Tor running on localhost)
http://192.168.0.4:8118 (for Privoxy running on host 192.168.0.4)
CONFIGURED EXCHANGES:
{x}
'''.format(
x='\n '.join(['%s - %s'%(n,x.Desc) for n,x in enumerate(xchgs,1)])
)
}
opts,args = Opts.parse_opts(sys.argv,opts_data)[:2]
debug_curl = 'debug_curl' in opts
proxy = opts['proxy'] if 'proxy' in opts else None
# Global var: the currently displayed exchange
if 'alarm_clock_only' in opts:
num_digits = 4
mute_alrms = False
ga = None
xchgs = []
else:
mute_alrms = True
ga = xchgs[0]
fx_pairs = [('usd','cny')]
for a,b in fx_pairs:
msg_r('Getting {}/{} exchange rate...'.format(a.upper(),b.upper()))
if get_fx(a,b):
msg('OK')
else:
die_pause(1,'Unable to get {}/{} exchange rate'.format(a.upper(),b.upper()))
if 'xchgs' in opts:
usr_xchgs = [int(i)-1 for i in opts['xchgs'].split(',')]
xchgs = [xchgs[i] for i in usr_xchgs]
# Initialize some variables for active exchanges
for g in xchgs:
g.retry_interval = 30
g.conn_alrm_timeout = 300
g.tc.bal = 0.0,0.0,0.0
if not hasattr(g,'desc_short'): g.desc_short = g.desc
for k in sources:
s = getattr(g,k)
main_thrd_names = tuple([x.desc+'_ticker' for x in xchgs])
kill_flgs = dict([(k,th.Event()) for k in main_thrd_names + ('alrm','clock','log')])
debug = 'debug' in opts
if debug: msg('Debugging mode. Using saved data from files')
def do_errmsg(s,k='ticker'):
if k == 'ticker':
with dlock:
blank_ticker()
msg_r(CUR_HOME)
msgred_r(s)
time.sleep(5)
def toggle_option(k):
with dlock:
if k in opts: del opts[k]
else: opts[k] = True
def get_thrd_names(): return [i.name for i in th.enumerate()]
def start_alrm(name):
kill_flgs['alrm'].clear()
t = th.Thread(
target = play_sound,
name = name,
kwargs = {
'fn': sounds[name][0], # (fn,vol)
'repeat_spec': repeat_spec,
'remote_host': audio_host,
'vol': sound_vol * sounds[name][1] / 100,
'kill_flg': kill_flgs['alrm'],
'testing': False
}
)
t.daemon = True
t.start()
def start_alrm_maybe(my_thrd_name,kill_list=None):
if mute_alrms: return
# Allow for an empty kill list
klist = kill_list if kill_list != None else alrm_names
with alock: # kill alarm thrds except mine
if any([n in klist and n != my_thrd_name for n in get_thrd_names()]):
kill_flgs['alrm'].set()
if not my_thrd_name: return # if thread name 'None', kill first, then return
if not any([n in alrm_names for n in get_thrd_names()]):
start_alrm(my_thrd_name)
def parse_loglevel_arg(s):
m1 = 'Invalid loglevel argument: %s\n' % s
try: ret = [int(i) for i in s.split(',')]
except:
m2 = 'Loglevels must be comma-separated int values'
return False, m1+m2
if not set(ret) <= set(valid_loglevels):
m2 = 'Valid loglevels: %s' % ','.join([str(i) for i in valid_loglevels])
return False, m1+m2
return ret,'OK'
def set_alrm_vals(s):
ret = []
m1 = 'Invalid alarm argument: %s\n' % s
for e in s.split(':'):
try: ret.append([Decimal(i) for i in e.split(',')])
except:
m2 = 'Alarms must be comma-separated decimal values'
return False, m1+m2
if len(ret[-1]) != 2:
m2 = 'Each element of alarm list must have 2 items (lo_alrm,hi_alrm)'
return False, m1+m2
if len(ret) != len(xchgs):
m2 = 'Alarm list must be %s colon-separated comma-separated lists' % len(xchgs)
return False, m1+m2
for g,a in zip(xchgs,ret):
lo,hi = a
if lo > hi:
m2 = 'Low alarm (%s) is greater than high alarm (%s)' % (lo,hi)
return False, m1+m2
setattr(g,'lo_alrm',lo)
setattr(g,'hi_alrm',hi)
return ret,'OK'
def set_poll_intervals(s,xchg=None):
ret = []
m1 = 'Invalid poll interval argument: %s\n' % s
m2 = 'Poll intervals must be comma-separated integer values'
m3 = 'Poll interval must be integer value'
for e in (s.split(','),[s])[bool(xchg)]:
try: ret.append(float(e))
except:
return False, m1+(m2,m3)[bool(xchg)]
if not xchg and len(ret) != len(xchgs):
m2 = 'Poll interval list have %s items' % len(xchgs)
return False, m1+m2
for g,p in zip((xchgs,[xchg])[bool(xchg)],ret):
setattr(g,'poll_secs',float(p))
return ret,'OK'
def set_xch_param(source,param,s,desc):
ret = []
m1 = 'Invalid %s argument: %s\n' % (desc,s)
for e in s.split(':'):
try: ret.append(float(e))
except:
m2 = '%s arg must be colon-separated float values' % desc.capitalize()
return False, m1+m2
if len(ret) != len(xchgs):
m2 = '%s list must have %s colon-separated items' % (
desc.capitalize(),len(xchgs))
return False, m1+m2
for g,p in zip(xchgs,ret):
a = getattr(g,source)
setattr(a,param,float(p))
return ret,'OK'
if 'log' in opts:
loglevels,errmsg = parse_loglevel_arg(opts['log'])
if not loglevels: die_pause(1,errmsg)
msg('Logging at level{} {}'.format(
('s','')[len(loglevels)==1],
' '.join([valid_loglevels[i].upper() for i in loglevels])))
else:
loglevels = []
if 'repeat_spec' in opts:
repeat_spec = opts['repeat_spec']
msg("Using program '{}' for alarm".format(repeat_spec))
if 'num_digits' in opts:
n = opts['num_digits']
if len(n) != 1 or n not in '56789':
die_pause(1,"'%s': invalid value for --num-digits option" % n)
num_digits = int(n)
if 'volume' in opts:
sound_vol = int(opts['volume'])
for k in sounds:
sounds[k][1] = float(sounds[k][1] * sound_vol / 100)
msg('Adjusting sound volume by {}%'.format(sound_vol))
if 'test_volume' in opts:
for k in sounds:
msg("Playing '{}' ({})".format(k,sounds[k][0]))
play_sound(
fn=sounds[k][0],
vol=sound_vol * sounds[k][1] / 100,
testing='testing' in opts
)
sys.exit()
if len(args) > 1: Opts.usage(opts_data)
if len(args) == 1:
ret,errmsg = set_alrm_vals(args[0])
if not ret: die_pause(1,'Error: ' + errmsg)
msg('Setting alarms to %s' % ', '.join(['{} {}'.format(i,j) for i,j in ret]))
if 'poll_intervals' in opts:
ret,errmsg = set_poll_intervals(opts['poll_intervals'])
if not ret: die_pause(1,errmsg)
msg('Polling every %s seconds' % ret)
tmux = 'TMUX' in os.environ
if tmux:
subprocess.check_output(['tmux','set','set-titles','on'])
subprocess.check_output(['tmux','set','status','off'])
# dcmd = "date -R%s | cut -d' ' -f2-5" % ('','u')['utc' in opts]
# subprocess.check_output(['tmux','set','status-right','#(%s)' % dcmd])
infoW = 15
bigDigitsW = big_digits['w']*num_digits + big_digits['pw']*1
lPaneW = bigDigitsW+infoW+1
topPaneH,rPaneW = 6,23
def CUR_UP(n): return '\033[%sA' % n
def CUR_DOWN(n): return '\033[%sB' % n
def CUR_RIGHT(n): return '\033[%sC' % n
def CUR_LEFT(n): return '\033[%sD' % n
def WIN_TITLE(s): return '\033]0;%s\033\\' % s
def WIN_RESIZE(w,h): return '\033[8;%s;%st' % (h,w)
def WIN_CORNER(): return '\033[3;200;0t'
CUR_UP1 = '\033[A'
CUR_DN1 = '\033[B'
CUR_SHOW = '\033[?25h'
CUR_HIDE = '\033[?25l'
BLINK = '\033[5m'
RESET = '\033[0m'
CUR_HOME = '\033[H'
ERASE_ALL = '\033[0J'
def draw_rectangle(s,l,h):
msg_r(s + '\n'.join([l] * h))
def blank_ticker():
draw_rectangle(CUR_HOME,' ' * lPaneW,topPaneH)
def blank_big_digits():
draw_rectangle(CUR_HOME,' ' * (bigDigitsW+1),topPaneH)
def park_cursor():
msg_r('\r' + CUR_RIGHT(lPaneW-1))
def colorize_bal(g,n):
d = g.tc
fs = '{:.%sf}' % g.fiat_precision
return (nocolor,green,red)[
(bool(d.save_bal[n]) and
(d.bal[n]!=d.save_bal[n]) + (d.bal[n]<d.save_bal[n]))
](fs.format(d.bal[n]))
def display_ac_info():
ac_fmt = green('--:--') if alrm_clock == 'off' else yelbg(alrm_clock)
info_lines = (
'{}'.format(ac_fmt),
'snd vol: {}%'.format(int(sound_vol)),
'{}'.format((yellow('unmuted'),'muted')[mute_alrms]),
'{}'.format(blue(audio_host[:infoW] or 'localhost')),
'',
"'?' for help"
)
r = CUR_RIGHT(bigDigitsW+1)
msg_r(CUR_HOME+r+('\n'+r).join(info_lines))
park_cursor()
ts = 'Alarm Clock: {}{}'.format(alrm_clock.upper(),('',' (m)')[mute_alrms])
if tmux:
subprocess.check_output(['tmux','set','set-titles-string',ts])
else:
msg_r(WIN_TITLE(ts))
def display_ticker(g,called_by_clock=False):
if not g: return
d = g.tc
log(3,'display_ticker(): ' + repr(d.__dict__))
if not hasattr(d,'timestamp'): return # DEBUG
avg = sum((d.bal[i] or d.save_bal[i]) for i in range(3)) / 3
alrm,lb,hb,rst = (
(None,'','',''),
('lo',BLINK,'',RESET),
('hi','',BLINK,RESET)
)[(avg<g.lo_alrm)+(avg>g.hi_alrm)*2]
start_alrm_maybe(alrm,['lo','hi'])
if (g is not ga) and not called_by_clock: return
lfmt = '{n:.{p}f}'.format(p=g.fiat_precision,n=d.bal[2])
msg_r(CUR_HOME+lb+hb+display_big_digits(lfmt,pre=' '+rst))
if (g is not ga) and called_by_clock:
park_cursor()
return
hms = get_hms(d.timestamp,utc='utc' in opts)
xcur = '{:.2f}'.format(d.bal[2]/usdcny if g.cur == 'CNY' else d.bal[2]*usdcny)
ac_fmt = green('--:--') if alrm_clock == 'off' else yelbg(alrm_clock)
info_lines = (
'{} {}'.format(hms,('%ss'%int(g.poll_secs))),
'{} bid/ask'.format(cyan(g.desc_short)),
'{} {}'.format(colorize_bal(g,0), colorize_bal(g,1)),
'{g.xcur_sign}{x} {h}'.format(g=g,x=xcur,
h=('',blue(audio_host[:infoW-len(xcur)-2]))[bool(audio_host)]),
'{} {} {}'.format(yellow(('♫','-')[mute_alrms]),
lb+yellow(str(g.lo_alrm))+rst,
hb+yellow(str(g.hi_alrm))+rst),
'{} vol {}%'.format(ac_fmt,int(sound_vol))
)
r = CUR_RIGHT(bigDigitsW+1)
msg_r(CUR_HOME+r+('\n'+r).join(info_lines))
park_cursor()
ccs = ('฿','Ł')[g.cc_unit=='LTC']
if len(xchgs) == 2:
x1,x2 = xchgs
ts = '{}{:.{}f} / {}{:.{}f}'.format(
x1.cur_sign,
x1.tc.bal[2],
x1.fiat_precision,
x2.cur_sign,
x2.tc.bal[2],
x2.fiat_precision,
)
else:
ts = '{}: {}{:.{}f} ({})'.format(
ccs,
g.cur_sign,
d.bal[2],
g.fiat_precision,
g.Desc
)
if tmux:
# subprocess.call(['tmux','rename-session',g.Desc], stderr=subprocess.PIPE)
subprocess.check_output(['tmux','set','set-titles-string',ts])
else:
msg_r(WIN_TITLE(ts))
def log(*args): # [lvl,g,src,data] OR [lvl,data]
if 'log' in opts and args[0] in loglevels + [0]:
if len(args) == 2:
s = '{}: {}\n'.format(get_day_hms(),args[1])
elif len(args) == 4:
s = '{}: {} {} - {}\n'.format(
get_day_hms(),
args[1].desc_short.upper(),
args[2].upper(),
args[3]
)
with llock:
fd = os.open(logfile,os.O_RDWR|os.O_APPEND|os.O_CREAT|os.O_SYNC)
os.write(fd,s)
os.close(fd)
def get_market_data(g,d,connfail_msg='full'):
# , post_data={}
tcd = 'TRADING_CONSOLE_DEBUG_CONNFAIL'
debug_connfail = tcd in os.environ and os.environ[tcd]
null = None # hack for eval'ing Huobi trades data
if debug:
fn = 'debug_market_data/%s_ticker.json' % g.desc
try:
with open(fn) as f: text = f.read()
log(3,'get_market_data(): {} {}'.format(g.desc,text))
return eval(text)
except:
die(2,'Unable to open datafile %s' % fn)
fail_count,conn_begin_time = 0,time.time()
if debug_connfail:
retry_interval,conn_alrm_timeout = 10,0
else:
retry_interval,conn_alrm_timeout = g.retry_interval,g.conn_alrm_timeout
while True:
try:
text = get_url(d.url,proxy=proxy,debug=debug_curl)
log(2,g,d.desc,text)
return eval(text)
except KeyboardInterrupt:
die(1,'\nUser interrupt (get_market_data)\n')
except EOFError:
die(1,'\nEnd of file\n')
except Exception as e:
fail_count += 1
if connfail_msg:
with dlock:
if g is ga:
m = {
'short':'Connect fail ({})'.format(fail_count),
'full': 'Connect fail. Retry in {} seconds ({})'.format(
retry_interval,fail_count)
}
dn = CUR_DOWN(topPaneH-1)
blank_big_digits()
msg_r(CUR_HOME+dn+m[connfail_msg]+' \b')
k = '%s_%s' % (g.desc,d.desc)
if time.time() - conn_begin_time > conn_alrm_timeout:
if fail_count == 1:
log(1,'Connect error (%s)' % k)
if d.desc == 'ticker':
with dlock: d.bal = d.bal[:2] + (0.0,) # can't assign to tuple, so this
start_alrm_maybe('conn_err')
# Sleep until user interrupt
if kill_flgs[k].wait(retry_interval):
kill_flgs[k].clear()
return False
def killwait(g,d):
log(3,g,d.desc,'Begin wait')
k = '%s_%s' % (g.desc,d.desc)
if kill_flgs[k].wait(g.poll_secs):
kill_flgs[k].clear()
if quit: return True
log(3,g,d.desc,'End wait')
return False
def log_loop():
while True:
for g in xchgs:
with dlock:
lstr = '{} last: {}{}'.format(g.desc_short.upper(),g.cur_sign,g.tc.bal[2])
log(1,lstr)
if kill_flgs['log'].wait(30):
kill_flgs['log'].clear()
if quit: return True
def ticker_loop(g):
d = g.tc
while True:
ret = get_market_data(g,d)
log(3,'get_market_data() returned: {}'.format(repr(ret)))
if not ret: # kill flag was set
if quit: break
continue
errmsg = 'ticker_loop: HTTP returned bad data'
with dlock:
millisec = False
if g.desc == 'bitfinex':
a = ret
bal = 'bid','ask','last_price'
ts = ret['timestamp']
elif g.desc == 'gemini':
a = ret
bal = 'bid','ask','last'
try:
ts = ret['volume']['timestamp']
except:
log(1,errmsg); continue
millisec = True
elif g.desc in ('okcoin_usd','okcoin_cny','huobi'):
try:
a = ret['ticker']
except:
log(1,errmsg); continue
bal = 'buy','sell','last'
ts = ret[('time','date')[g.desc[:6]=='okcoin']]
else:
die(1,"Can't handle symbol '{}'".format(g.desc))
# okcoin.cn CNY: {"date":"1477932232","ticker":{"buy":"4844.13","high":"4873.36","last":"4844.16","low":"4660.0","sell":"4844.14","vol":"2992115.73393084"}}
# gemini: {"bid":"1025.64","ask":"1026.93","volume":{"BTC":"1710.8752181914","USD":"1734356.065049020336","timestamp":1486377600000},"last":"1026.93"}
# okcoin.com USD {"date":"1486581152","ticker":{"buy":"1057.59","high":"1070.0","last":"1059.63","low":"1002.51","sell":"1058.34","vol":"4118.856"}}
try:
d.timestamp = int(float(ts))
except:
log(1,errmsg); continue
if millisec: d.timestamp /= 1000
d.save_bal = d.bal
try:
d.bal = tuple([float(a[k]) for k in bal])
except:
log(1,errmsg); continue
log(3,'{}: timestamp {}, bal {}'.format(g.desc,d.timestamp,d.bal))
if killwait(g,d): break
def clock_loop():
ac_active = False
ac_prefix=(' ','')['alarm_clock_only' in opts]
def do_ticker(x):
with dlock:
blank_big_digits()
display_ticker(x,called_by_clock=True)
if kill_flgs['clock'].wait(2): sys.exit()
while True:
if 'alarm_clock_only' in opts:
with dlock:
blank_ticker()
display_ac_info()
elif not ac_active:
if 'cur_ticker_only' in opts:
do_ticker(ga)
else:
for x in xchgs: do_ticker(x)
if alrm_clock == 'off' and ac_active:
ac_active = False
if alrm_clock != 'off' or 'alarm_clock_only' in opts:
now = get_hms(no_secs=True)
if now == alrm_clock:
ac_active = True
start_alrm_maybe('alrm_clock',['alrm_clock'])
if not any([(n in alrm_names and n != 'alrm_clock') for n in get_thrd_names()]):
bl,rs = (('',''),(BLINK,RESET))[ac_active]
with dlock:
blank_big_digits()
msg_r(CUR_HOME+bl+display_big_digits(now,pre=ac_prefix)+rs)
park_cursor()
if kill_flgs['clock'].wait(2): return
def input_loop():
global ga,sound_vol,alrm_clock,quit,mute_alrms
from mmgen.node_tools.Term import get_keypress
if 'alarm_clock_only' not in opts:
msg("Type '?' for help after the ticker starts")
ch = get_keypress("Hit 'q' to quit, any other key to continue: ")
if ch == 'q': die(0,'')
if 'resize_window' in opts:
if tmux:
msg('\nWARNING: Window resizing doesn\'t work in tmux')
else:
msg_r(WIN_RESIZE(lPaneW,topPaneH))
msg_r(WIN_CORNER())
time.sleep(0.1)
from mmgen.term import set_terminal_vars
set_terminal_vars()
from mmgen.term import get_terminal_size
twid = get_terminal_size()[0]
if twid < lPaneW:
die_pause(1,'\nTerminal window must be at least %s characters wide' % lPaneW)
msg_r(CUR_HIDE)
# raw_input('\nPress any key to exit'); sys.exit() # DEBUG
blank_ticker()
for k in xchgs:
th.Thread(target=globals()['ticker_loop'],args=[k],name=k.desc+'_ticker').start()
th.Thread(target=clock_loop,name='clock').start()
th.Thread(target=log_loop,name='log').start()
time.sleep(1) # Hack for get_keypress()
def redraw():
blank_ticker()
display_ticker(ga)
help_texts = {
'tc': '''
a - set alarm clock l - set low alarm
A - set remote audio host m - mute alarms
e - update USD/CNY rate M - write message to log
G - set log levels p - set poll interval
h - set high alarm t - reload ticker
k - kill alarm v - set sound volume
L - log current state x - cycle thru exchanges
c - toggle display of current ticker only
-/+ - adjust sound volume
''',
'ac': '''
a - set alarm clock m - mute alarms
A - set remote audio host k - kill alarm
M - write message to log v - set sound volume
-/+ - adjust sound volume
'''
}
help_text = ['{:^{w}}'.format('KEYSTROKE COMMANDS',w=lPaneW)]
help_text += help_texts[('tc','ac')['alarm_clock_only' in opts]].strip().split('\n')
help_text_prompt = 'ESC or ENTER to exit, ↑ and ↓ to scroll'
def do_help():
scrollpos = 0
def fmt_help_txt(pos):
return '\n'.join(help_text[pos:pos+5])+'\n'+help_text_prompt+' '
with dlock:
while True:
blank_ticker()
ch = get_keypress(CUR_HOME+fmt_help_txt(scrollpos),esc_sequences=True)
if ch == CUR_DN1 and scrollpos < len(help_text) - topPaneH + 1:
scrollpos += 1
elif ch == CUR_UP1 and scrollpos != 0:
scrollpos -= 1
elif ch in '\n\033': break
blank_ticker()
display_ticker(ga)
def ks_msg_nolock(s,delay):
blank_ticker()
msg_r(CUR_HOME+s)
time.sleep(delay)
blank_ticker()
display_ticker(ga)
def ks_msg(s,delay):
with dlock:
ks_msg_nolock(s,delay)
def set_value(v,prompt,vtype=int,poll=False,source=None,
all_srcs=False,global_var=False,allow_empty=False):
def type_chk(vtype,val):
try:
vtype(val); return True
except:
return False
errmsg = ''
with dlock:
while True:
blank_ticker()
em = errmsg+'\n' if errmsg else ''
termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_termattrs)
# time.sleep(0.1)
s = raw_input(CUR_HOME+CUR_SHOW+em+prompt)
msg_r(CUR_HIDE)
if not s:
if allow_empty:
ks_msg_nolock('Value unset',0.5)
else:
blank_ticker()
display_ticker(ga)
return False
if v == 'alrm_clock':
if s == 'off':
globals()[v] = s
if v in get_thrd_names():
kill_flgs['alrm'].set()
return s
a = s.split(':')
ok = True
if len(a) != 2: ok = False
if ok:
try: h,m = int(a[0]),int(a[1])
except: ok = False
if ok:
if not (24 >= h >= 0): ok = False
if not (60 >= m >= 0): ok = False
if ok:
globals()[v] = '{:02d}:{:02d}'.format(h,m)
break
else:
errmsg = 'Incorrect alarm clock value'
continue
elif v == 'loglevels':
ret,err = parse_loglevel_arg(s)
if ret:
globals()[v] = ret; break
else:
errmsg = err; continue
elif v == 'poll_secs':
ret,errmsg = set_poll_intervals(s,xchg=ga)
if ret: break
else: continue
elif all_srcs:
vals = s.split(',')
if len(vals) != len(sources):
errmsg = 'Input must be %s comma-separated %s values'%(
len(sources),vtype.__name__)
continue
for k,val in zip(sources,vals):
ret = type_chk(vtype,val)
if ret: setattr(getattr(ga,k),v,vtype(val))
else: break
if ret: break
elif source:
if type_chk(vtype,s):
setattr(getattr(ga,source),v,vtype(s)); break
else:
if type_chk(vtype,s):
if v == 'hi_alrm' and vtype(s) < ga.lo_alrm:
errmsg = 'High alarm must be >= low alarm'
continue
if v == 'lo_alrm' and vtype(s) > ga.hi_alrm:
errmsg = 'Low alarm must be <= high alarm'
continue
if global_var:
globals()[v] = vtype(s); break
else:
setattr(ga,v,vtype(s)); break
errmsg = 'Value%s must be of type %s' % (
('','s')[all_srcs],vtype.__name__)
redraw()
if poll:
kill_flgs[ga.desc+'_ticker'].set()
return True
while True:
ch = get_keypress()
if ch == 'q':
quit = True
for k in kill_flgs: kill_flgs[k].set()
for i in th.enumerate():
if i.name in main_thrd_names + alrm_names + ('clock',): i.join()
msg('')
break
elif ch == 'a':
m1 = 'Current alarm clock: %s' % alrm_clock
m2 = "Enter alarm clock time or 'off': "
if set_value('alrm_clock',m1+'\n'+m2):
ks_msg("Alarm clock set to %s" % alrm_clock,1.5)
elif ch == 'A':
set_value('audio_host','Enter remote audio host: ',
vtype=str,global_var=True,allow_empty=True)
elif ch in 'H?': do_help()
elif ch == 'k':
if 'alrm_clock' in get_thrd_names(): alrm_clock = 'off'
a = any([i in alrm_names for i in get_thrd_names()])
log(3,'\n alrm_names: {}\n get_thrd_names(): {}'.format(
alrm_names,get_thrd_names()))
if a: kill_flgs['alrm'].set()
ks_msg(('No active alarms','Killing alarm')[a],0.5)
elif ch == 'M':
with dlock:
blank_ticker()
termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_termattrs)
s = raw_input(CUR_HOME+'Enter log message: ')
log(0,'USR_MSG: '+s)
display_ticker(ga)
elif ch == 'm':
mute_alrms = not mute_alrms
if mute_alrms: kill_flgs['alrm'].set()
ks_msg(('Unm','M')[mute_alrms] + 'uting alarms',1)
with dlock:
blank_ticker()
display_ticker(ga)
elif ch == 'v': set_value('sound_vol','Enter sound volume: ',global_var=True)
elif ch in '-+':
with dlock:
sound_vol += (1,-1)[ch=='-']
blank_ticker(); display_ticker(ga)
elif not 'alarm_clock_only' in opts:
if ch == 'c':
toggle_option('cur_ticker_only')
ks_msg('Displaying %s' %
('all tickers','current ticker only')['cur_ticker_only' in opts],0.5)
elif ch == 'G': set_value('loglevels','Enter log levels: ',global_var=True)
elif ch == 'h': set_value('hi_alrm','Enter high alarm: ',Decimal)
elif ch == 'l': set_value('lo_alrm','Enter low alarm: ',Decimal)
elif ch == 'L':
ks_msg('Logging current ticker at %s' % get_hms(),1.5)
with dlock:
log(0,'{} CUR_STATE\n BID/ASK/LAST {}\n'.format(ga.desc.upper(),ga.tc.bal))
elif ch == 'p': set_value('poll_secs',
'Enter poll interval: ',float,poll=True)
elif ch == 't':
key = '%s_ticker' % (ga.desc)
if key in kill_flgs:
kill_flgs[key].set()
ks_msg('Reloading ticker data',1)
else:
ks_msg('%s: no such key' % key,1)
elif ch == 'x':
for x in range(len(xchgs)):
if ga is xchgs[x]: break
new_g = xchgs[x+1 if x+1 < len(xchgs) else 0]
ks_msg('Switching to exchange %s' % new_g.Desc,0.5)
with dlock:
ga = new_g
redraw()
elif ch == 'e':
for a,b in fx_pairs:
ks_msg('Updating {}/{} rate'.format(a.upper(),b.upper()),0.5)
ret = update_fx(a,b)
m,d = (('Unable to update',2),('Updated',0.7))[bool(ret)]
ks_msg('{} {}/{} rate'.format(m,a.upper(),b.upper()),d)
def launch(name,*args,**kwargs):
def at_exit():
# msg_r(CUR_HOME+ERASE_ALL) # DEBUG
msg_r(CUR_SHOW)
termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_termattrs)
for k in kill_flgs: kill_flgs[k].set()
for i in th.enumerate():
if i.name in main_thrd_names + alrm_names + ('clock',): i.join()
if 'log' in opts: log(0,'Exiting')
import atexit
atexit.register(at_exit)
log(0,'Starting log')
try:
globals()[name](*args,**kwargs)
except KeyboardInterrupt:
quit = True
for k in kill_flgs: kill_flgs[k].set()
sys.stderr.write('\nUser interrupt\n')
import termios
old_termattrs = termios.tcgetattr(sys.stdin.fileno())
launch('input_loop')
Reference in a new issue View git blame Copy permalink