btc-ticker 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985
  1. #!/usr/bin/python
  2. # -*- coding: UTF-8 -*-
  3. #
  4. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  5. # Copyright (C)2013-2016 Philemon <mmgen-py@yandex.com>
  6. #
  7. # This program is free software: you can redistribute it and/or modify
  8. # it under the terms of the GNU General Public License as published by
  9. # the Free Software Foundation, either version 3 of the License, or
  10. # (at your option) any later version.
  11. #
  12. # This program is distributed in the hope that it will be useful,
  13. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. # GNU General Public License for more details.
  16. #
  17. # You should have received a copy of the GNU General Public License
  18. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  19. #
  20. """
  21. btc-ticker: ticker and alarm clock for mmgen-node-tools
  22. """
  23. import sys,os,time,subprocess
  24. import threading as th
  25. from collections import OrderedDict
  26. from decimal import Decimal
  27. from mmgen.share import Opts
  28. from mmgen.util import msg,msg_r,die,die_pause
  29. from mmgen.node_tools.Global import *
  30. from mmgen.node_tools.Util import *
  31. from mmgen.node_tools.Sound import *
  32. from mmgen.node_tools.Term import *
  33. from mmgen.color import *
  34. init_color()
  35. quit = False
  36. sound_vol = float(100)
  37. audio_host,repeat_spec = '','3s:5m,1m:30m,5m:1h,30m:1d'
  38. # repeat_spec = '1:5m' # DEBUG
  39. valid_loglevels = OrderedDict([(1,'info'),(2,'data'),(3,'debug')])
  40. logfile = 'ticker_log.txt'
  41. alrm_clock = 'off'
  42. num_digits = 6
  43. proxy = None
  44. fx_host= 'https://finance.yahoo.com'
  45. sounds_dir = nt.system_data_dir+'/audio'
  46. sounds = {
  47. 'lo': [sounds_dir+'/ringtone.wav', 95],
  48. 'hi': [sounds_dir+'/Positive.wav', 102],
  49. 'conn_err': [sounds_dir+'/Rhodes.wav', 105],
  50. 'alrm_clock': [sounds_dir+'/Counterpoint.wav', 105],
  51. }
  52. alrm_names = tuple(sounds.keys())
  53. dlock,llock,alock = th.Lock(),th.Lock(),th.Lock()
  54. data_dir = nt.data_dir+'/lib/ticker/'
  55. try: os.makedirs(data_dir)
  56. except: pass
  57. def update_fx(a,b,debug=False):
  58. if debug:
  59. with open('/tmp/ticker.out') as f: text = f.read()
  60. else:
  61. text = get_url('{}/q?s={}{}'.format(fx_host,a,b),gzip_ok=True,proxy=proxy,debug=debug_curl)
  62. if not text: return False
  63. import re
  64. ret = re.split('yfs_l10_{}{}=.+?>'.format(a,b),text,maxsplit=1)[1].split('<',1)[0][:10]
  65. try:
  66. globals()[a+b] = float(ret)
  67. with open(data_dir+'/{}{}.txt'.format(a,b),'wb') as f: f.write(ret+'\n')
  68. return True
  69. except:
  70. return False
  71. def get_fx(a,b):
  72. try:
  73. with open(data_dir+'/{}{}.txt'.format(a,b)) as f:
  74. d = f.read().rstrip()
  75. globals()[a+b] = float(d)
  76. return True
  77. except:
  78. return update_fx(a,b)
  79. with open('/etc/timezone') as f:
  80. os.environ['TZ'] = f.read().rstrip()
  81. class Xch(object): pass
  82. class Src(object): pass
  83. num_xchgs = 2
  84. xchgs = tuple([Xch() for i in range(num_xchgs)])
  85. sources = OrderedDict([('tc','ticker')])
  86. for g in xchgs:
  87. for k in sources:
  88. setattr(g,k,Src())
  89. setattr(getattr(g,k),'desc',sources[k])
  90. for g in [xchgs[0]]:
  91. g.cur = 'USD'
  92. g.Desc = 'BitFinex'
  93. g.desc = 'bitfinex'
  94. g.desc_short = 'bfx'
  95. g.tc.url = 'https://api.bitfinex.com/v1/pubticker/btcusd'
  96. g.poll_secs = 60
  97. g.cur_sign = '$'
  98. g.xcur_sign = '¥'
  99. g.fiat_precision = 2
  100. g.hi_alrm = 999999
  101. g.lo_alrm = 1
  102. g.cc_unit = 'BTC'
  103. for g in [xchgs[1]]:
  104. g.cur = 'CNY'
  105. g.Desc = 'OKCoin'
  106. g.desc = 'okcoin'
  107. g.desc_short = 'okc'
  108. g.tc.url = 'https://www.okcoin.cn/api/v1/ticker.do?symbol=btc_cny'
  109. g.poll_secs = 60
  110. g.cur_sign = '¥'
  111. g.xcur_sign = '$'
  112. g.fiat_precision = 1
  113. g.hi_alrm = 999999
  114. g.lo_alrm = 1
  115. g.cc_unit = 'BTC'
  116. opts_data = {
  117. 'prog_name': sys.argv[0].split('/')[-1],
  118. 'desc': 'Price alarm for Bitcoin exchange',
  119. 'usage': '[opts] [<low price alarm> <high price alarm>]',
  120. #-b, --big-number Print big numbers
  121. 'options': """
  122. -h, --help Print this help message
  123. -a, --alarm-clock-only Disable ticker, use as alarm clock only
  124. -c, --cur-ticker-only Display only current exchange's ticker price
  125. -d, --debug Debug mode. Use saved HTTP data from files
  126. -D, --debug-curl Debug curl
  127. -l, --log= l Log all data to file '{lf}' at levels 'l' (comma-separated: {lls})
  128. -n, --num-digits= n Display 'n' number of big digits
  129. -p, --poll-intervals=i1[,i2] Poll servers every 'i' seconds (default: '{pi}')
  130. -P, --proxy= x Connect via proxy 'x' (see PROXY EXAMPLES below for format)
  131. -r, --repeat-spec Sleep interval/duration program for the alarm
  132. (default: '{r}')
  133. (see REPEAT SPEC FORMAT below)
  134. -R, --resize-window Resize window to optimum dimensions
  135. -t, --testing Testing mode. Don't execute shell commands
  136. -u, --utc Show times in UTC rather than local time
  137. -v, --volume= n Adjust sound volume by percentage 'n' (default: {v})
  138. -V, --test-volume Test the alarm volume and exit
  139. -x, --xchgs= x,y[,…] Display only exchanges 'x,y[,…]' (comma-separated
  140. list of integers, see CONFIGURED EXCHANGES below)
  141. """.format(
  142. v=int(sound_vol),
  143. pi = ','.join([str(g.poll_secs) for g in xchgs]),
  144. r=repeat_spec,
  145. lf=logfile,
  146. lls=', '.join(['%s=%s'%(i,j) for i,j in valid_loglevels.items()])
  147. ),
  148. 'notes': '''
  149. REPEAT SPEC FORMAT:
  150. <interval>:<duration>[,<interval>:<duration>[,…]]
  151. For example, '3s:5m,1m:2h,30m:1d' means:
  152. ring alarm every 3 seconds for 5 minutes,
  153. then every minute for 2 hours,
  154. then every 30 minutes for 1 day
  155. PROXY EXAMPLES:
  156. socks5h://localhost:9050 (for Tor running on localhost)
  157. http://192.168.0.4:8118 (for Privoxy running on host 192.168.0.4)
  158. CONFIGURED EXCHANGES:
  159. {x}
  160. '''.format(
  161. x='\n '.join(['%s - %s'%(n,x.Desc) for n,x in enumerate(xchgs,1)])
  162. )
  163. }
  164. opts,args = Opts.parse_opts(sys.argv,opts_data)[:2]
  165. debug_curl = 'debug_curl' in opts
  166. proxy = opts['proxy'] if 'proxy' in opts else None
  167. # Global var: the currently displayed exchange
  168. if 'alarm_clock_only' in opts:
  169. num_digits = 4
  170. mute_alrms = False
  171. ga = None
  172. xchgs = []
  173. else:
  174. mute_alrms = True
  175. ga = xchgs[0]
  176. fx_pairs = [('usd','cny')]
  177. for a,b in fx_pairs:
  178. msg_r('Getting {}/{} exchange rate...'.format(a.upper(),b.upper()))
  179. if get_fx(a,b):
  180. msg('OK')
  181. else:
  182. die_pause(1,'Unable to get {}/{} exchange rate'.format(a.upper(),b.upper()))
  183. if 'xchgs' in opts:
  184. usr_xchgs = [int(i)-1 for i in opts['xchgs'].split(',')]
  185. xchgs = [xchgs[i] for i in usr_xchgs]
  186. # Initialize some variables for active exchanges
  187. for g in xchgs:
  188. g.retry_interval = 30
  189. g.conn_alrm_timeout = 300
  190. g.tc.bal = 0.0,0.0,0.0
  191. if not hasattr(g,'desc_short'): g.desc_short = g.desc
  192. for k in sources:
  193. s = getattr(g,k)
  194. main_thrd_names = tuple([x.desc+'_ticker' for x in xchgs])
  195. kill_flgs = dict([(k,th.Event()) for k in main_thrd_names + ('alrm','clock','log')])
  196. debug = 'debug' in opts
  197. if debug: msg('Debugging mode. Using saved data from files')
  198. def do_errmsg(s,k='ticker'):
  199. if k == 'ticker':
  200. with dlock:
  201. blank_ticker()
  202. msg_r(CUR_HOME)
  203. msgred_r(s)
  204. time.sleep(5)
  205. def toggle_option(k):
  206. with dlock:
  207. if k in opts: del opts[k]
  208. else: opts[k] = True
  209. def get_thrd_names(): return [i.name for i in th.enumerate()]
  210. def start_alrm(name):
  211. kill_flgs['alrm'].clear()
  212. t = th.Thread(
  213. target = play_sound,
  214. name = name,
  215. kwargs = {
  216. 'fn': sounds[name][0], # (fn,vol)
  217. 'repeat_spec': repeat_spec,
  218. 'remote_host': audio_host,
  219. 'vol': sound_vol * sounds[name][1] / 100,
  220. 'kill_flg': kill_flgs['alrm'],
  221. 'testing': False
  222. }
  223. )
  224. t.daemon = True
  225. t.start()
  226. def start_alrm_maybe(my_thrd_name,kill_list=None):
  227. if mute_alrms: return
  228. # Allow for an empty kill list
  229. klist = kill_list if kill_list != None else alrm_names
  230. with alock: # kill alarm thrds except mine
  231. if any([n in klist and n != my_thrd_name for n in get_thrd_names()]):
  232. kill_flgs['alrm'].set()
  233. if not my_thrd_name: return # if thread name 'None', kill first, then return
  234. if not any([n in alrm_names for n in get_thrd_names()]):
  235. start_alrm(my_thrd_name)
  236. def parse_loglevel_arg(s):
  237. m1 = 'Invalid loglevel argument: %s\n' % s
  238. try: ret = [int(i) for i in s.split(',')]
  239. except:
  240. m2 = 'Loglevels must be comma-separated int values'
  241. return False, m1+m2
  242. if not set(ret) <= set(valid_loglevels):
  243. m2 = 'Valid loglevels: %s' % ','.join([str(i) for i in valid_loglevels])
  244. return False, m1+m2
  245. return ret,'OK'
  246. def set_alrm_vals(s):
  247. ret = []
  248. m1 = 'Invalid alarm argument: %s\n' % s
  249. for e in s.split(':'):
  250. try: ret.append([Decimal(i) for i in e.split(',')])
  251. except:
  252. m2 = 'Alarms must be comma-separated decimal values'
  253. return False, m1+m2
  254. if len(ret[-1]) != 2:
  255. m2 = 'Each element of alarm list must have 2 items (lo_alrm,hi_alrm)'
  256. return False, m1+m2
  257. if len(ret) != len(xchgs):
  258. m2 = 'Alarm list must be %s colon-separated comma-separated lists' % len(xchgs)
  259. return False, m1+m2
  260. for g,a in zip(xchgs,ret):
  261. lo,hi = a
  262. if lo > hi:
  263. m2 = 'Low alarm (%s) is greater than high alarm (%s)' % (lo,hi)
  264. return False, m1+m2
  265. setattr(g,'lo_alrm',lo)
  266. setattr(g,'hi_alrm',hi)
  267. return ret,'OK'
  268. def set_poll_intervals(s,xchg=None):
  269. ret = []
  270. m1 = 'Invalid poll interval argument: %s\n' % s
  271. m2 = 'Poll intervals must be comma-separated integer values'
  272. m3 = 'Poll interval must be integer value'
  273. for e in (s.split(','),[s])[bool(xchg)]:
  274. try: ret.append(float(e))
  275. except:
  276. return False, m1+(m2,m3)[bool(xchg)]
  277. if not xchg and len(ret) != len(xchgs):
  278. m2 = 'Poll interval list have %s items' % len(xchgs)
  279. return False, m1+m2
  280. for g,p in zip((xchgs,[xchg])[bool(xchg)],ret):
  281. setattr(g,'poll_secs',float(p))
  282. return ret,'OK'
  283. def set_xch_param(source,param,s,desc):
  284. ret = []
  285. m1 = 'Invalid %s argument: %s\n' % (desc,s)
  286. for e in s.split(':'):
  287. try: ret.append(float(e))
  288. except:
  289. m2 = '%s arg must be colon-separated float values' % desc.capitalize()
  290. return False, m1+m2
  291. if len(ret) != len(xchgs):
  292. m2 = '%s list must have %s colon-separated items' % (
  293. desc.capitalize(),len(xchgs))
  294. return False, m1+m2
  295. for g,p in zip(xchgs,ret):
  296. a = getattr(g,source)
  297. setattr(a,param,float(p))
  298. return ret,'OK'
  299. if 'log' in opts:
  300. loglevels,errmsg = parse_loglevel_arg(opts['log'])
  301. if not loglevels: die_pause(1,errmsg)
  302. msg('Logging at level{} {}'.format(
  303. ('s','')[len(loglevels)==1],
  304. ' '.join([valid_loglevels[i].upper() for i in loglevels])))
  305. else:
  306. loglevels = []
  307. if 'repeat_spec' in opts:
  308. repeat_spec = opts['repeat_spec']
  309. msg("Using program '{}' for alarm".format(repeat_spec))
  310. if 'num_digits' in opts:
  311. n = opts['num_digits']
  312. if len(n) != 1 or n not in '56789':
  313. die_pause(1,"'%s': invalid value for --num-digits option" % n)
  314. num_digits = int(n)
  315. if 'volume' in opts:
  316. sound_vol = int(opts['volume'])
  317. for k in sounds:
  318. sounds[k][1] = float(sounds[k][1] * sound_vol / 100)
  319. msg('Adjusting sound volume by {}%'.format(sound_vol))
  320. if 'test_volume' in opts:
  321. for k in sounds:
  322. msg("Playing '{}' ({})".format(k,sounds[k][0]))
  323. play_sound(
  324. fn=sounds[k][0],
  325. vol=sound_vol * sounds[k][1] / 100,
  326. testing='testing' in opts
  327. )
  328. sys.exit()
  329. if len(args) > 1: Opts.usage(opts_data)
  330. if len(args) == 1:
  331. ret,errmsg = set_alrm_vals(args[0])
  332. if not ret: die_pause(1,'Error: ' + errmsg)
  333. msg('Setting alarms to %s' % ', '.join(['{} {}'.format(i,j) for i,j in ret]))
  334. if 'poll_intervals' in opts:
  335. ret,errmsg = set_poll_intervals(opts['poll_intervals'])
  336. if not ret: die_pause(1,errmsg)
  337. msg('Polling every %s seconds' % ret)
  338. tmux = 'TMUX' in os.environ
  339. if tmux:
  340. subprocess.check_output(['tmux','set','set-titles','on'])
  341. subprocess.check_output(['tmux','set','status','off'])
  342. # dcmd = "date -R%s | cut -d' ' -f2-5" % ('','u')['utc' in opts]
  343. # subprocess.check_output(['tmux','set','status-right','#(%s)' % dcmd])
  344. infoW = 15
  345. bigDigitsW = big_digits['w']*num_digits + big_digits['pw']*1
  346. lPaneW = bigDigitsW+infoW+1
  347. topPaneH,rPaneW = 6,23
  348. def CUR_UP(n): return '\033[%sA' % n
  349. def CUR_DOWN(n): return '\033[%sB' % n
  350. def CUR_RIGHT(n): return '\033[%sC' % n
  351. def CUR_LEFT(n): return '\033[%sD' % n
  352. def WIN_TITLE(s): return '\033]0;%s\033\\' % s
  353. def WIN_RESIZE(w,h): return '\033[8;%s;%st' % (h,w)
  354. def WIN_CORNER(): return '\033[3;200;0t'
  355. CUR_UP1 = '\033[A'
  356. CUR_DN1 = '\033[B'
  357. CUR_SHOW = '\033[?25h'
  358. CUR_HIDE = '\033[?25l'
  359. BLINK = '\033[5m'
  360. RESET = '\033[0m'
  361. CUR_HOME = '\033[H'
  362. ERASE_ALL = '\033[0J'
  363. def draw_rectangle(s,l,h):
  364. msg_r(s + '\n'.join([l] * h))
  365. def blank_ticker():
  366. draw_rectangle(CUR_HOME,' ' * lPaneW,topPaneH)
  367. def blank_big_digits():
  368. draw_rectangle(CUR_HOME,' ' * (bigDigitsW+1),topPaneH)
  369. def park_cursor():
  370. msg_r('\r' + CUR_RIGHT(lPaneW-1))
  371. def colorize_bal(g,n):
  372. d = g.tc
  373. fs = '{:.%sf}' % g.fiat_precision
  374. return (nocolor,green,red)[
  375. (bool(d.save_bal[n]) and
  376. (d.bal[n]!=d.save_bal[n]) + (d.bal[n]<d.save_bal[n]))
  377. ](fs.format(d.bal[n]))
  378. def display_ac_info():
  379. ac_fmt = green('--:--') if alrm_clock == 'off' else yelbg(alrm_clock)
  380. info_lines = (
  381. '{}'.format(ac_fmt),
  382. 'snd vol: {}%'.format(int(sound_vol)),
  383. '{}'.format((yellow('unmuted'),'muted')[mute_alrms]),
  384. '{}'.format(blue(audio_host[:infoW] or 'localhost'))
  385. )
  386. r = CUR_RIGHT(bigDigitsW+1)
  387. msg_r(CUR_HOME+r+('\n'+r).join(info_lines))
  388. park_cursor()
  389. ts = 'Alarm Clock: {}{}'.format(alrm_clock.upper(),('',' (m)')[mute_alrms])
  390. if tmux:
  391. subprocess.check_output(['tmux','set','set-titles-string',ts])
  392. else:
  393. msg_r(WIN_TITLE(ts))
  394. def display_ticker(g,called_by_clock=False):
  395. if not g: return
  396. d = g.tc
  397. if not hasattr(d,'timestamp'): return
  398. avg = sum((d.bal[i] or d.save_bal[i]) for i in range(3)) / 3
  399. alrm,lb,hb,rst = (
  400. (None,'','',''),
  401. ('lo',BLINK,'',RESET),
  402. ('hi','',BLINK,RESET)
  403. )[(avg<g.lo_alrm)+(avg>g.hi_alrm)*2]
  404. start_alrm_maybe(alrm,['lo','hi'])
  405. if (g is not ga) and not called_by_clock: return
  406. lfmt = '{n:.{p}f}'.format(p=g.fiat_precision,n=d.bal[2])
  407. msg_r(CUR_HOME+lb+hb+display_big_digits(lfmt,pre=' '+rst))
  408. if (g is not ga) and called_by_clock:
  409. park_cursor()
  410. return
  411. hms = get_hms(d.timestamp,utc='utc' in opts)
  412. xcur = '{:.2f}'.format(d.bal[2]/usdcny if g.cur == 'CNY' else d.bal[2]*usdcny)
  413. ac_fmt = green('--:--') if alrm_clock == 'off' else yelbg(alrm_clock)
  414. info_lines = (
  415. '{} {}'.format(hms,('%ss'%int(g.poll_secs))),
  416. '{} bid/ask'.format(cyan(g.desc_short)),
  417. '{} {}'.format(colorize_bal(g,0), colorize_bal(g,1)),
  418. '{g.xcur_sign}{x} {h}'.format(g=g,x=xcur,
  419. h=('',blue(audio_host[:infoW-len(xcur)-2]))[bool(audio_host)]),
  420. '{} {} {}'.format(yellow(('♫','-')[mute_alrms]),
  421. lb+yellow(str(g.lo_alrm))+rst,
  422. hb+yellow(str(g.hi_alrm))+rst),
  423. '{} vol {}%'.format(ac_fmt,int(sound_vol))
  424. )
  425. r = CUR_RIGHT(bigDigitsW+1)
  426. msg_r(CUR_HOME+r+('\n'+r).join(info_lines))
  427. park_cursor()
  428. ccs = ('฿','Ł')[g.cc_unit=='LTC']
  429. if len(xchgs) == 2:
  430. x1,x2 = xchgs
  431. ts = '{}{:.{}f} / {}{:.{}f}'.format(
  432. x1.cur_sign,
  433. x1.tc.bal[2],
  434. x1.fiat_precision,
  435. x2.cur_sign,
  436. x2.tc.bal[2],
  437. x2.fiat_precision,
  438. )
  439. else:
  440. ts = '{}: {}{:.{}f} ({})'.format(
  441. ccs,
  442. g.cur_sign,
  443. d.bal[2],
  444. g.fiat_precision,
  445. g.Desc
  446. )
  447. if tmux:
  448. # subprocess.call(['tmux','rename-session',g.Desc], stderr=subprocess.PIPE)
  449. subprocess.check_output(['tmux','set','set-titles-string',ts])
  450. else:
  451. msg_r(WIN_TITLE(ts))
  452. def log(*args): # [lvl,g,src,data] OR [lvl,data]
  453. if 'log' in opts and args[0] in loglevels + [0]:
  454. if len(args) == 2:
  455. s = '{}: {}\n'.format(get_day_hms(),args[1])
  456. elif len(args) == 4:
  457. s = '{}: {} {} - {}\n'.format(
  458. get_day_hms(),
  459. args[1].desc_short.upper(),
  460. args[2].upper(),
  461. args[3]
  462. )
  463. with llock:
  464. fd = os.open(logfile,os.O_RDWR|os.O_APPEND|os.O_CREAT|os.O_SYNC)
  465. os.write(fd,s)
  466. os.close(fd)
  467. def get_market_data(g,d,connfail_msg='full'):
  468. # , post_data={}
  469. tcd = 'TRADING_CONSOLE_DEBUG_CONNFAIL'
  470. debug_connfail = tcd in os.environ and os.environ[tcd]
  471. null = None # hack for eval'ing Huobi trades data
  472. if debug:
  473. fn = 'debug_market_data/%s_ticker.json' % g.desc
  474. try:
  475. with open(fn) as f: text = f.read()
  476. return eval(text)
  477. except:
  478. die(2,'Unable to open datafile %s' % fn)
  479. fail_count,conn_begin_time = 0,time.time()
  480. if debug_connfail:
  481. retry_interval,conn_alrm_timeout = 10,0
  482. else:
  483. retry_interval,conn_alrm_timeout = g.retry_interval,g.conn_alrm_timeout
  484. while True:
  485. try:
  486. text = get_url(d.url,proxy=proxy,debug=debug_curl)
  487. log(2,g,d.desc,text)
  488. return eval(text)
  489. except KeyboardInterrupt:
  490. die(1,'\nUser interrupt (get_market_data)\n')
  491. except EOFError:
  492. die(1,'\nEnd of file\n')
  493. except Exception as e:
  494. fail_count += 1
  495. if connfail_msg:
  496. with dlock:
  497. if g is ga:
  498. m = {
  499. 'short':'Connect fail ({})'.format(fail_count),
  500. 'full': 'Connect fail. Retry in {} seconds ({})'.format(
  501. retry_interval,fail_count)
  502. }
  503. dn = CUR_DOWN(topPaneH-1)
  504. blank_big_digits()
  505. msg_r(CUR_HOME+dn+m[connfail_msg]+' \b')
  506. k = '%s_%s' % (g.desc,d.desc)
  507. if time.time() - conn_begin_time > conn_alrm_timeout:
  508. if fail_count == 1:
  509. log(1,'Connect error (%s)' % k)
  510. if d.desc == 'ticker':
  511. with dlock: d.bal = d.bal[:2] + (0.0,) # can't assign to tuple, so this
  512. start_alrm_maybe('conn_err')
  513. # Sleep until user interrupt
  514. if kill_flgs[k].wait(retry_interval):
  515. kill_flgs[k].clear()
  516. return False
  517. def killwait(g,d):
  518. log(3,g,d.desc,'Begin wait')
  519. k = '%s_%s' % (g.desc,d.desc)
  520. if kill_flgs[k].wait(g.poll_secs):
  521. kill_flgs[k].clear()
  522. if quit: return True
  523. log(3,g,d.desc,'End wait')
  524. return False
  525. def log_loop():
  526. while True:
  527. for g in xchgs:
  528. with dlock:
  529. lstr = '{} last: {}{}'.format(g.desc_short.upper(),g.cur_sign,g.tc.bal[2])
  530. log(1,lstr)
  531. if kill_flgs['log'].wait(30):
  532. kill_flgs['log'].clear()
  533. if quit: return True
  534. def ticker_loop(g):
  535. d = g.tc
  536. while True:
  537. ret = get_market_data(g,d)
  538. if not ret: # kill flag was set
  539. if quit: break
  540. continue
  541. errmsg = 'ticker_loop: HTTP returned bad data'
  542. with dlock:
  543. if g.desc == 'bitfinex':
  544. a = ret
  545. ts,bal = 'timestamp',('bid','ask','last_price')
  546. else:
  547. try:
  548. a = ret['ticker']
  549. except:
  550. log(1,errmsg); continue
  551. ts = ('time','date')[g.desc=='okcoin']
  552. bal = 'buy','sell','last'
  553. # okc: {"date":"1477932232","ticker":{"buy":"4844.13","high":"4873.36","last":"4844.16","low":"4660.0","sell":"4844.14","vol":"2992115.73393084"}}
  554. try:
  555. d.timestamp = int(float(ret[ts]))
  556. except:
  557. log(1,errmsg); continue
  558. d.save_bal = d.bal
  559. try:
  560. d.bal = tuple([float(a[k]) for k in bal])
  561. except:
  562. log(1,errmsg); continue
  563. if killwait(g,d): break
  564. def clock_loop():
  565. ac_active = False
  566. ac_prefix=(' ','')['alarm_clock_only' in opts]
  567. def do_ticker(x):
  568. with dlock:
  569. blank_big_digits()
  570. display_ticker(x,called_by_clock=True)
  571. if kill_flgs['clock'].wait(2): sys.exit()
  572. while True:
  573. if 'alarm_clock_only' in opts:
  574. with dlock:
  575. blank_ticker()
  576. display_ac_info()
  577. elif not ac_active:
  578. if 'cur_ticker_only' in opts:
  579. do_ticker(ga)
  580. else:
  581. for x in xchgs: do_ticker(x)
  582. if alrm_clock == 'off' and ac_active:
  583. ac_active = False
  584. if alrm_clock != 'off' or 'alarm_clock_only' in opts:
  585. now = get_hms(no_secs=True)
  586. if now == alrm_clock:
  587. ac_active = True
  588. start_alrm_maybe('alrm_clock',['alrm_clock'])
  589. if not any([(n in alrm_names and n != 'alrm_clock') for n in get_thrd_names()]):
  590. bl,rs = (('',''),(BLINK,RESET))[ac_active]
  591. with dlock:
  592. blank_big_digits()
  593. msg_r(CUR_HOME+bl+display_big_digits(now,pre=ac_prefix)+rs)
  594. park_cursor()
  595. if kill_flgs['clock'].wait(2): return
  596. def input_loop():
  597. global ga,sound_vol,alrm_clock,quit,mute_alrms
  598. from mmgen.node_tools.Term import get_keypress
  599. ch = get_keypress("Hit 'q' to quit, any other key to continue: ")
  600. if ch == 'q': die(0,'')
  601. if 'resize_window' in opts:
  602. if tmux:
  603. msg('\nWARNING: Window resizing doesn\'t work in tmux')
  604. else:
  605. msg_r(WIN_RESIZE(lPaneW,topPaneH))
  606. msg_r(WIN_CORNER())
  607. time.sleep(0.1)
  608. from mmgen.term import set_terminal_vars
  609. set_terminal_vars()
  610. from mmgen.term import get_terminal_size
  611. twid = get_terminal_size()[0]
  612. if twid < lPaneW:
  613. die_pause(1,'\nTerminal window must be at least %s characters wide' % lPaneW)
  614. msg_r(CUR_HIDE)
  615. # raw_input('\nPress any key to exit'); sys.exit() # DEBUG
  616. blank_ticker()
  617. for k in xchgs:
  618. th.Thread(target=globals()['ticker_loop'],args=[k],name=k.desc+'_ticker').start()
  619. th.Thread(target=clock_loop,name='clock').start()
  620. th.Thread(target=log_loop,name='log').start()
  621. time.sleep(1) # Hack for get_keypress()
  622. def redraw():
  623. blank_ticker()
  624. display_ticker(ga)
  625. help_texts = {
  626. 'tc': '''
  627. a - set alarm clock l - set low alarm
  628. A - set remote audio host m - mute alarms
  629. e - update USD/CNY rate M - write message to log
  630. G - set log levels p - set poll interval
  631. h - set high alarm t - reload ticker
  632. k - kill alarm v - set sound volume
  633. L - log current state x - cycle thru exchanges
  634. c - toggle display of current ticker only
  635. -/+ - adjust sound volume
  636. ''',
  637. 'ac': '''
  638. a - set alarm clock m - mute alarms
  639. A - set remote audio host k - kill alarm
  640. M - write message to log v - set sound volume
  641. -/+ - adjust sound volume
  642. '''
  643. }
  644. help_text = ['{:^{w}}'.format('KEYSTROKE COMMANDS',w=lPaneW)]
  645. help_text += help_texts[('tc','ac')['alarm_clock_only' in opts]].strip().split('\n')
  646. help_text_prompt = 'ESC or ENTER to exit, ↑ and ↓ to scroll'
  647. def do_help():
  648. scrollpos = 0
  649. def fmt_help_txt(pos):
  650. return '\n'.join(help_text[pos:pos+5])+'\n'+help_text_prompt+' '
  651. with dlock:
  652. while True:
  653. blank_ticker()
  654. ch = get_keypress(CUR_HOME+fmt_help_txt(scrollpos),esc_sequences=True)
  655. if ch == CUR_DN1 and scrollpos < len(help_text) - topPaneH + 1:
  656. scrollpos += 1
  657. elif ch == CUR_UP1 and scrollpos != 0:
  658. scrollpos -= 1
  659. elif ch in '\n\033': break
  660. blank_ticker()
  661. display_ticker(ga)
  662. def ks_msg_nolock(s,delay):
  663. blank_ticker()
  664. msg_r(CUR_HOME+s)
  665. time.sleep(delay)
  666. blank_ticker()
  667. display_ticker(ga)
  668. def ks_msg(s,delay):
  669. with dlock:
  670. ks_msg_nolock(s,delay)
  671. def set_value(v,prompt,vtype=int,poll=False,source=None,
  672. all_srcs=False,global_var=False,allow_empty=False):
  673. def type_chk(vtype,val):
  674. try:
  675. vtype(val); return True
  676. except:
  677. return False
  678. errmsg = ''
  679. with dlock:
  680. while True:
  681. blank_ticker()
  682. em = errmsg+'\n' if errmsg else ''
  683. termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_termattrs)
  684. # time.sleep(0.1)
  685. s = raw_input(CUR_HOME+CUR_SHOW+em+prompt)
  686. msg_r(CUR_HIDE)
  687. if not s:
  688. if allow_empty:
  689. ks_msg_nolock('Value unset',0.5)
  690. else:
  691. blank_ticker()
  692. display_ticker(ga)
  693. return False
  694. if v == 'alrm_clock':
  695. if s == 'off':
  696. globals()[v] = s
  697. if v in get_thrd_names():
  698. kill_flgs['alrm'].set()
  699. return s
  700. a = s.split(':')
  701. ok = True
  702. if len(a) != 2: ok = False
  703. if ok:
  704. try: h,m = int(a[0]),int(a[1])
  705. except: ok = False
  706. if ok:
  707. if not (24 >= h >= 0): ok = False
  708. if not (60 >= m >= 0): ok = False
  709. if ok:
  710. globals()[v] = '{:02d}:{:02d}'.format(h,m)
  711. break
  712. else:
  713. errmsg = 'Incorrect alarm clock value'
  714. continue
  715. elif v == 'loglevels':
  716. ret,err = parse_loglevel_arg(s)
  717. if ret:
  718. globals()[v] = ret; break
  719. else:
  720. errmsg = err; continue
  721. elif v == 'poll_secs':
  722. ret,errmsg = set_poll_intervals(s,xchg=ga)
  723. if ret: break
  724. else: continue
  725. elif all_srcs:
  726. vals = s.split(',')
  727. if len(vals) != len(sources):
  728. errmsg = 'Input must be %s comma-separated %s values'%(
  729. len(sources),vtype.__name__)
  730. continue
  731. for k,val in zip(sources,vals):
  732. ret = type_chk(vtype,val)
  733. if ret: setattr(getattr(ga,k),v,vtype(val))
  734. else: break
  735. if ret: break
  736. elif source:
  737. if type_chk(vtype,s):
  738. setattr(getattr(ga,source),v,vtype(s)); break
  739. else:
  740. if type_chk(vtype,s):
  741. if v == 'hi_alrm' and vtype(s) < ga.lo_alrm:
  742. errmsg = 'High alarm must be >= low alarm'
  743. continue
  744. if v == 'lo_alrm' and vtype(s) > ga.hi_alrm:
  745. errmsg = 'Low alarm must be <= high alarm'
  746. continue
  747. if global_var:
  748. globals()[v] = vtype(s); break
  749. else:
  750. setattr(ga,v,vtype(s)); break
  751. errmsg = 'Value%s must be of type %s' % (
  752. ('','s')[all_srcs],vtype.__name__)
  753. redraw()
  754. if poll:
  755. kill_flgs[ga.desc+'_ticker'].set()
  756. return True
  757. while True:
  758. ch = get_keypress()
  759. if ch == 'q':
  760. quit = True
  761. for k in kill_flgs: kill_flgs[k].set()
  762. for i in th.enumerate():
  763. if i.name in main_thrd_names + alrm_names + ('clock',): i.join()
  764. msg('')
  765. break
  766. elif ch == 'a':
  767. m1 = 'Current alarm clock: %s' % alrm_clock
  768. m2 = "Enter alarm clock time or 'off': "
  769. if set_value('alrm_clock',m1+'\n'+m2):
  770. ks_msg("Alarm clock set to %s" % alrm_clock,1.5)
  771. elif ch == 'A':
  772. set_value('audio_host','Enter remote audio host: ',
  773. vtype=str,global_var=True,allow_empty=True)
  774. elif ch in 'H?': do_help()
  775. elif ch == 'k':
  776. if 'alrm_clock' in get_thrd_names(): alrm_clock = 'off'
  777. a = any([i in alrm_names for i in get_thrd_names()])
  778. log(3,'\n alrm_names: {}\n get_thrd_names(): {}'.format(
  779. alrm_names,get_thrd_names()))
  780. if a: kill_flgs['alrm'].set()
  781. ks_msg(('No active alarms','Killing alarm')[a],0.5)
  782. elif ch == 'M':
  783. with dlock:
  784. blank_ticker()
  785. termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_termattrs)
  786. s = raw_input(CUR_HOME+'Enter log message: ')
  787. log(0,'USR_MSG: '+s)
  788. display_ticker(ga)
  789. elif ch == 'm':
  790. mute_alrms = not mute_alrms
  791. if mute_alrms: kill_flgs['alrm'].set()
  792. ks_msg(('Unm','M')[mute_alrms] + 'uting alarms',1)
  793. with dlock:
  794. blank_ticker()
  795. display_ticker(ga)
  796. elif ch == 'v': set_value('sound_vol','Enter sound volume: ',global_var=True)
  797. elif ch in '-+':
  798. with dlock:
  799. sound_vol += (1,-1)[ch=='-']
  800. blank_ticker(); display_ticker(ga)
  801. elif not 'alarm_clock_only' in opts:
  802. if ch == 'c':
  803. toggle_option('cur_ticker_only')
  804. ks_msg('Displaying %s' %
  805. ('all tickers','current ticker only')['cur_ticker_only' in opts],0.5)
  806. elif ch == 'G': set_value('loglevels','Enter log levels: ',global_var=True)
  807. elif ch == 'h': set_value('hi_alrm','Enter high alarm: ',Decimal)
  808. elif ch == 'l': set_value('lo_alrm','Enter low alarm: ',Decimal)
  809. elif ch == 'L':
  810. ks_msg('Logging current ticker at %s' % get_hms(),1.5)
  811. with dlock:
  812. log(0,'{} CUR_STATE\n BID/ASK/LAST {}\n'.format(ga.desc.upper(),ga.tc.bal))
  813. elif ch == 'p': set_value('poll_secs',
  814. 'Enter poll interval: ',float,poll=True)
  815. elif ch == 't':
  816. key = '%s_ticker' % (ga.desc)
  817. if key in kill_flgs:
  818. kill_flgs[key].set()
  819. ks_msg('Reloading ticker data',1)
  820. else:
  821. ks_msg('%s: no such key' % key,1)
  822. elif ch == 'x':
  823. for x in range(len(xchgs)):
  824. if ga is xchgs[x]: break
  825. new_g = xchgs[x+1 if x+1 < len(xchgs) else 0]
  826. ks_msg('Switching to exchange %s' % new_g.Desc,0.5)
  827. with dlock:
  828. ga = new_g
  829. redraw()
  830. elif ch == 'e':
  831. for a,b in fx_pairs:
  832. ks_msg('Updating {}/{} rate'.format(a.upper(),b.upper()),0.5)
  833. ret = update_fx(a,b)
  834. m,d = (('Unable to update',2),('Updated',0.7))[bool(ret)]
  835. ks_msg('{} {}/{} rate'.format(m,a.upper(),b.upper()),d)
  836. def launch(name,*args,**kwargs):
  837. def at_exit():
  838. # msg_r(CUR_HOME+ERASE_ALL) # DEBUG
  839. msg_r(CUR_SHOW)
  840. termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_termattrs)
  841. for k in kill_flgs: kill_flgs[k].set()
  842. for i in th.enumerate():
  843. if i.name in main_thrd_names + alrm_names + ('clock',): i.join()
  844. if 'log' in opts: log(0,'Exiting')
  845. import atexit
  846. atexit.register(at_exit)
  847. log(0,'Starting log')
  848. try:
  849. globals()[name](*args,**kwargs)
  850. except KeyboardInterrupt:
  851. quit = True
  852. for k in kill_flgs: kill_flgs[k].set()
  853. sys.stderr.write('\nUser interrupt\n')
  854. import termios
  855. old_termattrs = termios.tcgetattr(sys.stdin.fileno())
  856. launch('input_loop')