Logger.py

import sys
import time
import datetime
import atexit
import json
import io
import ConsoleUtils
from RingBuffer import RingBuffer

class ConsoleOutput(object):
	def __init__(self):
		self._status = ''
		atexit.register(self._exit)

	def _exit(self):
		self._status += '  ' # In case the shell added a ^C
		self.status('')

	def status(self, msg, time=''):
		status = str(msg)
		cols = ConsoleUtils.get_terminal_size()[0]
		if msg != '' and len(status) > cols:
			#truncate status, try preventing console bloating
			status = str(msg)[:cols-4] + '...'
		update = '\r'
		update += status
		update += ' ' * (len(self._status) - len(status))
		update += '\b' * (len(self._status) - len(status))
		sys.stderr.write(update)
		self._status = status

	def printline(self, line):
		update = '\r'
		update += line + ' ' * (len(self._status) - len(line)) + '\n'
		update += self._status
		sys.stderr.write(update)


class JsonOutput(object):
	def __init__(self, file, logLimit):
		self.jsonOutputFile = file
		self.jsonOutput = {}
		self.jsonOutputCoins = {}
		self.jsonOutput["raw_data"] = self.jsonOutputCoins
		self.jsonOutputLog = RingBuffer(logLimit);

	def status(self, status, time):
		self.jsonOutput["last_update"] = time
		self.jsonOutput["last_status"] = status

	def printline(self, line):
		self.jsonOutputLog.append(line)

	def writeJsonFile(self):
		with io.open(self.jsonOutputFile, 'w', encoding='utf-8') as f:
			self.jsonOutput["log"] = self.jsonOutputLog.get()
			f.write(unicode(json.dumps(self.jsonOutput, ensure_ascii=False, sort_keys=True)))
			f.close()
	
	def statusValue(self, coin, key, value):
		if(coin not in self.jsonOutputCoins):
			self.jsonOutputCoins[coin] = {}
		self.jsonOutputCoins[coin][key] = str(value)
	
	def clearStatusValues(self):
		self.jsonOutputCoins = {}

class Logger(object):
	def __init__(self, jsonFile = '', jsonLogSize = -1):
		self._lended = ''
		if jsonFile != '' and jsonLogSize != -1:
			self.console = JsonOutput(jsonFile, jsonLogSize)
		else:
			self.console = ConsoleOutput()
		self.refreshStatus()

	def timestamp(self):
		ts = time.time()
		return datetime.datetime.fromtimestamp(ts).strftime('%Y-%m-%d %H:%M:%S')

	def log(self, msg):
		self.console.printline(self.timestamp() + ' ' + msg)
		self.refreshStatus()

	def offer(self, amt, cur, rate, days, msg):
		line = self.timestamp() + ' Placing ' + str(amt) + ' ' + str(cur) + ' at ' + str(float(rate)*100) + '% for ' + days + ' days... ' + self.digestApiMsg(msg)
		self.console.printline(line)
		self.refreshStatus()

	def cancelOrders(self, cur, msg):
		line = self.timestamp() + ' Canceling all ' + str(cur) + ' orders... ' + self.digestApiMsg(msg)
		self.console.printline(line)
		self.refreshStatus()

	def refreshStatus(self, lended=''):
		if lended != '':
			self._lended = lended;
		self.console.status(self._lended, self.timestamp())

	def updateStatusValue(self, coin, key, value):
		if(hasattr(self.console , 'statusValue')):
			self.console.statusValue(coin, key, value)
	
	def persistStatus(self):
		if(hasattr(self.console , 'writeJsonFile')):
			self.console.writeJsonFile()
		if(hasattr(self.console , 'clearStatusValues')):
			self.console.clearStatusValues()

	def digestApiMsg(self, msg):
		try:
			m = (msg['message'])
		except KeyError:
			pass
		try:
			m = (msg['error'])
		except KeyError:
			pass
		return m