cryptobook.py

#! /usr/bin/env python3.6
import sys
import argparse
import os
import time
import json
from terminaltables import SingleTable
from colr import color
from blessed import Terminal
from function.function import *
from classes.order import Order
from api.bittrex import API_Bittrex


def price(args):
	
	price = get_last_price(args)
	if not price:
		return
	if args.quiet:
		print(price)
	else:
		print(get_exchange(args), get_market(args), price)

def buy(args):
	
	path = args.config['book-path']
	# Check if the file book.json exist and if not create it
	# and iniate it with '[]'
	if not os.path.isfile(path):
		data = "[]"
		file = open(path, 'a+')
		file.write(data)
		file.close()
	order = Order(args)
	data = json.loads(open(path).read())
	data.append(order.toDict())
	with open(path, 'w') as outfile:
		json.dump(data, outfile, indent=4)

	# Display the order with a nice table
	table_data = [
		['id', 'Exchange', 'Market', 'Price'],
		[order.id , order.exchange, order.market, order.last]
	]
	table = SingleTable(table_data)
	table.title = 'Buy Order'
	print (table.table)

def close(args):
	
	path = args.config['book-path']
	if not os.path.isfile(path):
			print('Order book empty, use buy command to fill it')
			return
	data = json.loads(open(path).read())
	i = 0
	while i < len(data):
		if data[i]['id'] == args.id:
			table_data = [['id', 'Exchange', 'Market', 'Price', 'Current', 'Profit']]
			current = get_last_price_tmp(data[i]['market'])
			profit = get_profit(data[i]['last'], current)
			table_data.append([data[i]['id'], data[i]['exchange'], 
								data[i]['market'], data[i]['last'], current , profit])
			table = SingleTable(table_data)
			table.title = 'Close Order'
			print (table.table)
			data.pop(i)
			break
		i += 1
	with open('./book.json', 'w') as outfile:
		json.dump(data, outfile, indent=4)

def position(args):

	path = args.config['book-path']
	if not os.path.isfile(path):
		print('Order book empty, use buy command to fill it')
		return
	data = json.loads(open(path).read())
	table_data = [['id', 'Exchange', 'Market', 'Price', 'Current', 'Profit']]
	i = 0
	while i < len(data):
		current = get_last_price_tmp(data[i]['market'])
		profit = get_profit(data[i]['last'], current)
		table_data.append([data[i]['id'], data[i]['exchange'], 
						data[i]['market'], data[i]['last'], current , profit])
		i += 1
	table = SingleTable(table_data)
	
	if args.live:
		return(table.table)
	else:
		print(table.table)

def refresh(args):
	"""Ugly way to make a auto refresh tab"""
	term = Terminal()
	key = ""
	with term.fullscreen(), term.cbreak():
		while key != 'q':
			tab = position(args)
			print(term.move_y(0) + ('Press Q to exit the live mode').rstrip() +
			'\n' + term.center(tab).rstrip() + term.clear_eos)
			loop = term.inkey(timeout=0.5)

def main(args):
	
	config_path = "./config.json"
	try:
		args.config = json.loads(open(config_path).read())
	except:
		print('Error loading config.json')
		sys.exit(0)
	if hasattr(args, 'function'):
		if hasattr(args, 'live'):
			refresh(args)
		else:
			args.function(args)
	else:
		print('Use -h to see the usage')

if __name__ == '__main__':

# Create the top-level parser
	parser = argparse.ArgumentParser(prog='Cryptobook')

	parser.add_argument('--version',
						action='version',
						version='%(prog)s 0.0.1')

# Creat subparser for the Price command
	subparsers = parser.add_subparsers(help='Use -h with the subcommande to see the help section')

# Create the parser for the "price" command
	parser_price = subparsers.add_parser('price',
						help='Show the current price of the currency specified in currency')
	parser_price.add_argument('currency', 
						type=str,
						help='Currency you want to see the price default exchange in BTC')

	parser_price.add_argument('-m', '--market',
						action='store',
						type=str,
						dest='market',
						nargs='?',
						choices=['BTC', 'ETH', 'USDT'],
						help='Choose the market BTC ETH USDT| default is BTC')

	parser_price.add_argument('-e', '--exchange',
						action='store',
						type=str,
						dest='exchange',
						nargs='?',
						help='Choose the market default is Bittrex')

	parser_price.add_argument('-q', '--quiet',
						action='store_true',
						help='Only price is Ouput')

	parser_price.set_defaults(function=price)

# Create the parser for the "buy" command
	parser_buy = subparsers.add_parser('buy',
					help='Show the current price of the currency specified in currency')
	
	parser_buy.add_argument('currency',
					type=str,
					help='Currency you want to see the price default exchange in BTC')

	parser_buy.add_argument('-m', '--market',
					action='store',
					type=str,
					dest='market',
					nargs='?',
					choices=['BTC', 'ETH', 'USDT'],
					help='Choose the market BTC ETH | default is BTC')

	parser_buy.add_argument('-e', '--exchange',
					action='store',
					type=str,
					dest='exchange',
					nargs='?',
					help='Choose the exchange default is Bittrex')
	parser_buy.add_argument('-p', '--price',
					type=float,
					help='If you want to be more accurate you can specify the price, ignoring the one from the API')

	parser_buy.set_defaults(function=buy)

# Create the parser for the "position" command
	parser_pos = subparsers.add_parser('position',
					help='Show your current position')

	parser_pos.add_argument('-l', '--live',
					action='store_true',
					help='Choose the market default is Bittrex')

	parser_pos.set_defaults(function=position)

# Create the parser for the "close" command
	parser_close = subparsers.add_parser('close',
					help='Show your current position')

	parser_close.add_argument('id',
					type=int,
					help='Close the id from the order book')

	parser_close.set_defaults(function=close)

# Parse argument lists
	args = parser.parse_args()
	main(args)