btc-ticker 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989
  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. "'?' for help"
  387. )
  388. r = CUR_RIGHT(bigDigitsW+1)
  389. msg_r(CUR_HOME+r+('\n'+r).join(info_lines))
  390. park_cursor()
  391. ts = 'Alarm Clock: {}{}'.format(alrm_clock.upper(),('',' (m)')[mute_alrms])
  392. if tmux:
  393. subprocess.check_output(['tmux','set','set-titles-string',ts])
  394. else:
  395. msg_r(WIN_TITLE(ts))
  396. def display_ticker(g,called_by_clock=False):
  397. if not g: return
  398. d = g.tc
  399. if not hasattr(d,'timestamp'): return
  400. avg = sum((d.bal[i] or d.save_bal[i]) for i in range(3)) / 3
  401. alrm,lb,hb,rst = (
  402. (None,'','',''),
  403. ('lo',BLINK,'',RESET),
  404. ('hi','',BLINK,RESET)
  405. )[(avg<g.lo_alrm)+(avg>g.hi_alrm)*2]
  406. start_alrm_maybe(alrm,['lo','hi'])
  407. if (g is not ga) and not called_by_clock: return
  408. lfmt = '{n:.{p}f}'.format(p=g.fiat_precision,n=d.bal[2])
  409. msg_r(CUR_HOME+lb+hb+display_big_digits(lfmt,pre=' '+rst))
  410. if (g is not ga) and called_by_clock:
  411. park_cursor()
  412. return
  413. hms = get_hms(d.timestamp,utc='utc' in opts)
  414. xcur = '{:.2f}'.format(d.bal[2]/usdcny if g.cur == 'CNY' else d.bal[2]*usdcny)
  415. ac_fmt = green('--:--') if alrm_clock == 'off' else yelbg(alrm_clock)
  416. info_lines = (
  417. '{} {}'.format(hms,('%ss'%int(g.poll_secs))),
  418. '{} bid/ask'.format(cyan(g.desc_short)),
  419. '{} {}'.format(colorize_bal(g,0), colorize_bal(g,1)),
  420. '{g.xcur_sign}{x} {h}'.format(g=g,x=xcur,
  421. h=('',blue(audio_host[:infoW-len(xcur)-2]))[bool(audio_host)]),
  422. '{} {} {}'.format(yellow(('♫','-')[mute_alrms]),
  423. lb+yellow(str(g.lo_alrm))+rst,
  424. hb+yellow(str(g.hi_alrm))+rst),
  425. '{} vol {}%'.format(ac_fmt,int(sound_vol))
  426. )
  427. r = CUR_RIGHT(bigDigitsW+1)
  428. msg_r(CUR_HOME+r+('\n'+r).join(info_lines))
  429. park_cursor()
  430. ccs = ('฿','Ł')[g.cc_unit=='LTC']
  431. if len(xchgs) == 2:
  432. x1,x2 = xchgs
  433. ts = '{}{:.{}f} / {}{:.{}f}'.format(
  434. x1.cur_sign,
  435. x1.tc.bal[2],
  436. x1.fiat_precision,
  437. x2.cur_sign,
  438. x2.tc.bal[2],
  439. x2.fiat_precision,
  440. )
  441. else:
  442. ts = '{}: {}{:.{}f} ({})'.format(
  443. ccs,
  444. g.cur_sign,
  445. d.bal[2],
  446. g.fiat_precision,
  447. g.Desc
  448. )
  449. if tmux:
  450. # subprocess.call(['tmux','rename-session',g.Desc], stderr=subprocess.PIPE)
  451. subprocess.check_output(['tmux','set','set-titles-string',ts])
  452. else:
  453. msg_r(WIN_TITLE(ts))
  454. def log(*args): # [lvl,g,src,data] OR [lvl,data]
  455. if 'log' in opts and args[0] in loglevels + [0]:
  456. if len(args) == 2:
  457. s = '{}: {}\n'.format(get_day_hms(),args[1])
  458. elif len(args) == 4:
  459. s = '{}: {} {} - {}\n'.format(
  460. get_day_hms(),
  461. args[1].desc_short.upper(),
  462. args[2].upper(),
  463. args[3]
  464. )
  465. with llock:
  466. fd = os.open(logfile,os.O_RDWR|os.O_APPEND|os.O_CREAT|os.O_SYNC)
  467. os.write(fd,s)
  468. os.close(fd)
  469. def get_market_data(g,d,connfail_msg='full'):
  470. # , post_data={}
  471. tcd = 'TRADING_CONSOLE_DEBUG_CONNFAIL'
  472. debug_connfail = tcd in os.environ and os.environ[tcd]
  473. null = None # hack for eval'ing Huobi trades data
  474. if debug:
  475. fn = 'debug_market_data/%s_ticker.json' % g.desc
  476. try:
  477. with open(fn) as f: text = f.read()
  478. return eval(text)
  479. except:
  480. die(2,'Unable to open datafile %s' % fn)
  481. fail_count,conn_begin_time = 0,time.time()
  482. if debug_connfail:
  483. retry_interval,conn_alrm_timeout = 10,0
  484. else:
  485. retry_interval,conn_alrm_timeout = g.retry_interval,g.conn_alrm_timeout
  486. while True:
  487. try:
  488. text = get_url(d.url,proxy=proxy,debug=debug_curl)
  489. log(2,g,d.desc,text)
  490. return eval(text)
  491. except KeyboardInterrupt:
  492. die(1,'\nUser interrupt (get_market_data)\n')
  493. except EOFError:
  494. die(1,'\nEnd of file\n')
  495. except Exception as e:
  496. fail_count += 1
  497. if connfail_msg:
  498. with dlock:
  499. if g is ga:
  500. m = {
  501. 'short':'Connect fail ({})'.format(fail_count),
  502. 'full': 'Connect fail. Retry in {} seconds ({})'.format(
  503. retry_interval,fail_count)
  504. }
  505. dn = CUR_DOWN(topPaneH-1)
  506. blank_big_digits()
  507. msg_r(CUR_HOME+dn+m[connfail_msg]+' \b')
  508. k = '%s_%s' % (g.desc,d.desc)
  509. if time.time() - conn_begin_time > conn_alrm_timeout:
  510. if fail_count == 1:
  511. log(1,'Connect error (%s)' % k)
  512. if d.desc == 'ticker':
  513. with dlock: d.bal = d.bal[:2] + (0.0,) # can't assign to tuple, so this
  514. start_alrm_maybe('conn_err')
  515. # Sleep until user interrupt
  516. if kill_flgs[k].wait(retry_interval):
  517. kill_flgs[k].clear()
  518. return False
  519. def killwait(g,d):
  520. log(3,g,d.desc,'Begin wait')
  521. k = '%s_%s' % (g.desc,d.desc)
  522. if kill_flgs[k].wait(g.poll_secs):
  523. kill_flgs[k].clear()
  524. if quit: return True
  525. log(3,g,d.desc,'End wait')
  526. return False
  527. def log_loop():
  528. while True:
  529. for g in xchgs:
  530. with dlock:
  531. lstr = '{} last: {}{}'.format(g.desc_short.upper(),g.cur_sign,g.tc.bal[2])
  532. log(1,lstr)
  533. if kill_flgs['log'].wait(30):
  534. kill_flgs['log'].clear()
  535. if quit: return True
  536. def ticker_loop(g):
  537. d = g.tc
  538. while True:
  539. ret = get_market_data(g,d)
  540. if not ret: # kill flag was set
  541. if quit: break
  542. continue
  543. errmsg = 'ticker_loop: HTTP returned bad data'
  544. with dlock:
  545. if g.desc == 'bitfinex':
  546. a = ret
  547. ts,bal = 'timestamp',('bid','ask','last_price')
  548. else:
  549. try:
  550. a = ret['ticker']
  551. except:
  552. log(1,errmsg); continue
  553. ts = ('time','date')[g.desc=='okcoin']
  554. bal = 'buy','sell','last'
  555. # okc: {"date":"1477932232","ticker":{"buy":"4844.13","high":"4873.36","last":"4844.16","low":"4660.0","sell":"4844.14","vol":"2992115.73393084"}}
  556. try:
  557. d.timestamp = int(float(ret[ts]))
  558. except:
  559. log(1,errmsg); continue
  560. d.save_bal = d.bal
  561. try:
  562. d.bal = tuple([float(a[k]) for k in bal])
  563. except:
  564. log(1,errmsg); continue
  565. if killwait(g,d): break
  566. def clock_loop():
  567. ac_active = False
  568. ac_prefix=(' ','')['alarm_clock_only' in opts]
  569. def do_ticker(x):
  570. with dlock:
  571. blank_big_digits()
  572. display_ticker(x,called_by_clock=True)
  573. if kill_flgs['clock'].wait(2): sys.exit()
  574. while True:
  575. if 'alarm_clock_only' in opts:
  576. with dlock:
  577. blank_ticker()
  578. display_ac_info()
  579. elif not ac_active:
  580. if 'cur_ticker_only' in opts:
  581. do_ticker(ga)
  582. else:
  583. for x in xchgs: do_ticker(x)
  584. if alrm_clock == 'off' and ac_active:
  585. ac_active = False
  586. if alrm_clock != 'off' or 'alarm_clock_only' in opts:
  587. now = get_hms(no_secs=True)
  588. if now == alrm_clock:
  589. ac_active = True
  590. start_alrm_maybe('alrm_clock',['alrm_clock'])
  591. if not any([(n in alrm_names and n != 'alrm_clock') for n in get_thrd_names()]):
  592. bl,rs = (('',''),(BLINK,RESET))[ac_active]
  593. with dlock:
  594. blank_big_digits()
  595. msg_r(CUR_HOME+bl+display_big_digits(now,pre=ac_prefix)+rs)
  596. park_cursor()
  597. if kill_flgs['clock'].wait(2): return
  598. def input_loop():
  599. global ga,sound_vol,alrm_clock,quit,mute_alrms
  600. from mmgen.node_tools.Term import get_keypress
  601. if 'alarm_clock_only' not in opts:
  602. msg("Type '?' for help after the ticker starts")
  603. ch = get_keypress("Hit 'q' to quit, any other key to continue: ")
  604. if ch == 'q': die(0,'')
  605. if 'resize_window' in opts:
  606. if tmux:
  607. msg('\nWARNING: Window resizing doesn\'t work in tmux')
  608. else:
  609. msg_r(WIN_RESIZE(lPaneW,topPaneH))
  610. msg_r(WIN_CORNER())
  611. time.sleep(0.1)
  612. from mmgen.term import set_terminal_vars
  613. set_terminal_vars()
  614. from mmgen.term import get_terminal_size
  615. twid = get_terminal_size()[0]
  616. if twid < lPaneW:
  617. die_pause(1,'\nTerminal window must be at least %s characters wide' % lPaneW)
  618. msg_r(CUR_HIDE)
  619. # raw_input('\nPress any key to exit'); sys.exit() # DEBUG
  620. blank_ticker()
  621. for k in xchgs:
  622. th.Thread(target=globals()['ticker_loop'],args=[k],name=k.desc+'_ticker').start()
  623. th.Thread(target=clock_loop,name='clock').start()
  624. th.Thread(target=log_loop,name='log').start()
  625. time.sleep(1) # Hack for get_keypress()
  626. def redraw():
  627. blank_ticker()
  628. display_ticker(ga)
  629. help_texts = {
  630. 'tc': '''
  631. a - set alarm clock l - set low alarm
  632. A - set remote audio host m - mute alarms
  633. e - update USD/CNY rate M - write message to log
  634. G - set log levels p - set poll interval
  635. h - set high alarm t - reload ticker
  636. k - kill alarm v - set sound volume
  637. L - log current state x - cycle thru exchanges
  638. c - toggle display of current ticker only
  639. -/+ - adjust sound volume
  640. ''',
  641. 'ac': '''
  642. a - set alarm clock m - mute alarms
  643. A - set remote audio host k - kill alarm
  644. M - write message to log v - set sound volume
  645. -/+ - adjust sound volume
  646. '''
  647. }
  648. help_text = ['{:^{w}}'.format('KEYSTROKE COMMANDS',w=lPaneW)]
  649. help_text += help_texts[('tc','ac')['alarm_clock_only' in opts]].strip().split('\n')
  650. help_text_prompt = 'ESC or ENTER to exit, ↑ and ↓ to scroll'
  651. def do_help():
  652. scrollpos = 0
  653. def fmt_help_txt(pos):
  654. return '\n'.join(help_text[pos:pos+5])+'\n'+help_text_prompt+' '
  655. with dlock:
  656. while True:
  657. blank_ticker()
  658. ch = get_keypress(CUR_HOME+fmt_help_txt(scrollpos),esc_sequences=True)
  659. if ch == CUR_DN1 and scrollpos < len(help_text) - topPaneH + 1:
  660. scrollpos += 1
  661. elif ch == CUR_UP1 and scrollpos != 0:
  662. scrollpos -= 1
  663. elif ch in '\n\033': break
  664. blank_ticker()
  665. display_ticker(ga)
  666. def ks_msg_nolock(s,delay):
  667. blank_ticker()
  668. msg_r(CUR_HOME+s)
  669. time.sleep(delay)
  670. blank_ticker()
  671. display_ticker(ga)
  672. def ks_msg(s,delay):
  673. with dlock:
  674. ks_msg_nolock(s,delay)
  675. def set_value(v,prompt,vtype=int,poll=False,source=None,
  676. all_srcs=False,global_var=False,allow_empty=False):
  677. def type_chk(vtype,val):
  678. try:
  679. vtype(val); return True
  680. except:
  681. return False
  682. errmsg = ''
  683. with dlock:
  684. while True:
  685. blank_ticker()
  686. em = errmsg+'\n' if errmsg else ''
  687. termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_termattrs)
  688. # time.sleep(0.1)
  689. s = raw_input(CUR_HOME+CUR_SHOW+em+prompt)
  690. msg_r(CUR_HIDE)
  691. if not s:
  692. if allow_empty:
  693. ks_msg_nolock('Value unset',0.5)
  694. else:
  695. blank_ticker()
  696. display_ticker(ga)
  697. return False
  698. if v == 'alrm_clock':
  699. if s == 'off':
  700. globals()[v] = s
  701. if v in get_thrd_names():
  702. kill_flgs['alrm'].set()
  703. return s
  704. a = s.split(':')
  705. ok = True
  706. if len(a) != 2: ok = False
  707. if ok:
  708. try: h,m = int(a[0]),int(a[1])
  709. except: ok = False
  710. if ok:
  711. if not (24 >= h >= 0): ok = False
  712. if not (60 >= m >= 0): ok = False
  713. if ok:
  714. globals()[v] = '{:02d}:{:02d}'.format(h,m)
  715. break
  716. else:
  717. errmsg = 'Incorrect alarm clock value'
  718. continue
  719. elif v == 'loglevels':
  720. ret,err = parse_loglevel_arg(s)
  721. if ret:
  722. globals()[v] = ret; break
  723. else:
  724. errmsg = err; continue
  725. elif v == 'poll_secs':
  726. ret,errmsg = set_poll_intervals(s,xchg=ga)
  727. if ret: break
  728. else: continue
  729. elif all_srcs:
  730. vals = s.split(',')
  731. if len(vals) != len(sources):
  732. errmsg = 'Input must be %s comma-separated %s values'%(
  733. len(sources),vtype.__name__)
  734. continue
  735. for k,val in zip(sources,vals):
  736. ret = type_chk(vtype,val)
  737. if ret: setattr(getattr(ga,k),v,vtype(val))
  738. else: break
  739. if ret: break
  740. elif source:
  741. if type_chk(vtype,s):
  742. setattr(getattr(ga,source),v,vtype(s)); break
  743. else:
  744. if type_chk(vtype,s):
  745. if v == 'hi_alrm' and vtype(s) < ga.lo_alrm:
  746. errmsg = 'High alarm must be >= low alarm'
  747. continue
  748. if v == 'lo_alrm' and vtype(s) > ga.hi_alrm:
  749. errmsg = 'Low alarm must be <= high alarm'
  750. continue
  751. if global_var:
  752. globals()[v] = vtype(s); break
  753. else:
  754. setattr(ga,v,vtype(s)); break
  755. errmsg = 'Value%s must be of type %s' % (
  756. ('','s')[all_srcs],vtype.__name__)
  757. redraw()
  758. if poll:
  759. kill_flgs[ga.desc+'_ticker'].set()
  760. return True
  761. while True:
  762. ch = get_keypress()
  763. if ch == 'q':
  764. quit = True
  765. for k in kill_flgs: kill_flgs[k].set()
  766. for i in th.enumerate():
  767. if i.name in main_thrd_names + alrm_names + ('clock',): i.join()
  768. msg('')
  769. break
  770. elif ch == 'a':
  771. m1 = 'Current alarm clock: %s' % alrm_clock
  772. m2 = "Enter alarm clock time or 'off': "
  773. if set_value('alrm_clock',m1+'\n'+m2):
  774. ks_msg("Alarm clock set to %s" % alrm_clock,1.5)
  775. elif ch == 'A':
  776. set_value('audio_host','Enter remote audio host: ',
  777. vtype=str,global_var=True,allow_empty=True)
  778. elif ch in 'H?': do_help()
  779. elif ch == 'k':
  780. if 'alrm_clock' in get_thrd_names(): alrm_clock = 'off'
  781. a = any([i in alrm_names for i in get_thrd_names()])
  782. log(3,'\n alrm_names: {}\n get_thrd_names(): {}'.format(
  783. alrm_names,get_thrd_names()))
  784. if a: kill_flgs['alrm'].set()
  785. ks_msg(('No active alarms','Killing alarm')[a],0.5)
  786. elif ch == 'M':
  787. with dlock:
  788. blank_ticker()
  789. termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_termattrs)
  790. s = raw_input(CUR_HOME+'Enter log message: ')
  791. log(0,'USR_MSG: '+s)
  792. display_ticker(ga)
  793. elif ch == 'm':
  794. mute_alrms = not mute_alrms
  795. if mute_alrms: kill_flgs['alrm'].set()
  796. ks_msg(('Unm','M')[mute_alrms] + 'uting alarms',1)
  797. with dlock:
  798. blank_ticker()
  799. display_ticker(ga)
  800. elif ch == 'v': set_value('sound_vol','Enter sound volume: ',global_var=True)
  801. elif ch in '-+':
  802. with dlock:
  803. sound_vol += (1,-1)[ch=='-']
  804. blank_ticker(); display_ticker(ga)
  805. elif not 'alarm_clock_only' in opts:
  806. if ch == 'c':
  807. toggle_option('cur_ticker_only')
  808. ks_msg('Displaying %s' %
  809. ('all tickers','current ticker only')['cur_ticker_only' in opts],0.5)
  810. elif ch == 'G': set_value('loglevels','Enter log levels: ',global_var=True)
  811. elif ch == 'h': set_value('hi_alrm','Enter high alarm: ',Decimal)
  812. elif ch == 'l': set_value('lo_alrm','Enter low alarm: ',Decimal)
  813. elif ch == 'L':
  814. ks_msg('Logging current ticker at %s' % get_hms(),1.5)
  815. with dlock:
  816. log(0,'{} CUR_STATE\n BID/ASK/LAST {}\n'.format(ga.desc.upper(),ga.tc.bal))
  817. elif ch == 'p': set_value('poll_secs',
  818. 'Enter poll interval: ',float,poll=True)
  819. elif ch == 't':
  820. key = '%s_ticker' % (ga.desc)
  821. if key in kill_flgs:
  822. kill_flgs[key].set()
  823. ks_msg('Reloading ticker data',1)
  824. else:
  825. ks_msg('%s: no such key' % key,1)
  826. elif ch == 'x':
  827. for x in range(len(xchgs)):
  828. if ga is xchgs[x]: break
  829. new_g = xchgs[x+1 if x+1 < len(xchgs) else 0]
  830. ks_msg('Switching to exchange %s' % new_g.Desc,0.5)
  831. with dlock:
  832. ga = new_g
  833. redraw()
  834. elif ch == 'e':
  835. for a,b in fx_pairs:
  836. ks_msg('Updating {}/{} rate'.format(a.upper(),b.upper()),0.5)
  837. ret = update_fx(a,b)
  838. m,d = (('Unable to update',2),('Updated',0.7))[bool(ret)]
  839. ks_msg('{} {}/{} rate'.format(m,a.upper(),b.upper()),d)
  840. def launch(name,*args,**kwargs):
  841. def at_exit():
  842. # msg_r(CUR_HOME+ERASE_ALL) # DEBUG
  843. msg_r(CUR_SHOW)
  844. termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_termattrs)
  845. for k in kill_flgs: kill_flgs[k].set()
  846. for i in th.enumerate():
  847. if i.name in main_thrd_names + alrm_names + ('clock',): i.join()
  848. if 'log' in opts: log(0,'Exiting')
  849. import atexit
  850. atexit.register(at_exit)
  851. log(0,'Starting log')
  852. try:
  853. globals()[name](*args,**kwargs)
  854. except KeyboardInterrupt:
  855. quit = True
  856. for k in kill_flgs: kill_flgs[k].set()
  857. sys.stderr.write('\nUser interrupt\n')
  858. import termios
  859. old_termattrs = termios.tcgetattr(sys.stdin.fileno())
  860. launch('input_loop')