crypto_data.py

from requests.exceptions import HTTPError
import krakenex
import matplotlib.pyplot as plt


import pprint
import datetime
import time

kraken = krakenex.API()

def checkValidity(id): #false means invalid, true means valid
    try:
        response = kraken.query_public('Depth', {'pair': id, 'count': '500'})
        bids = response['result'][list(response["result"].keys())[0]]['bids']
        return True
    except KeyError:
        return False


def findMarketDepth(pair):
    try:
        response = kraken.query_public('Depth', {'pair': pair, 'count': '500'})
        bids = response['result'][list(response["result"].keys())[0]]['bids']
        asks = response['result'][list(response["result"].keys())[0]]['asks']
    except HTTPError as e:
        print(str(e))

    def findSubtotals(results):
        orders = {}
        for i in results:
            orders[float(i[0])] = float(i[1])

        sum = 0
        for value in orders.keys():
            sum += value
        marketPrice = sum/500

        total = 0
        for value in orders.keys():
            weighted = 1/((abs((value-marketPrice)/marketPrice))+1)*orders[value]
            total += weighted
        return total
    
    return(findSubtotals(asks) + findSubtotals(bids))


def findAveragePrice(pair):
    try:
        response = kraken.query_public('Depth', {'pair': pair, 'count': '500'})
        asks = response['result'][list(response["result"].keys())[0]]['asks']
    except HTTPError as e:
        print(str(e))

    sum = 0
    for i in asks:
        sum += float(i[0])
    return sum / 500

def graph(pair):
    try:
        response = kraken.query_public('Depth', {'pair': pair, 'count': '500'})
        bids = response['result'][list(response["result"].keys())[0]]['bids']
        asks = response['result'][list(response["result"].keys())[0]]['asks']
    except HTTPError as e:
        print(str(e))
    
    plt.title("Market Depth of " + pair)

    orders = {}
    for i in bids:
        orders[float(i[0])] = float(i[1])
    maxBid = max(list(orders.keys()))
    plt.plot(list(orders.keys()), list(orders.values()), color = "green")
    orders = {}
    for i in asks:
        orders[float(i[0])] = float(i[1])
    minBid = min(list(orders.keys()))
    plt.plot(list(orders.keys()), list(orders.values()), color = "red")

    plt.axvline(x = (minBid + maxBid)/2.0, ymin = 0, ymax = 80, color = "black")
    plt.xlabel(pair[-3:])
    plt.ylabel("Volume")
    plt.savefig("graph.png")
    return "graph.png"
    
def findOhlc(pair):
    now = datetime.datetime.now()
    unixNow = int(time.mktime(now.timetuple()))
    try:
        response = kraken.query_public('OHLC', {'pair': pair, 'interval': 1440, 'since': unixNow-3666})
        open = response['result'][list(response["result"].keys())[0]][0][1]
        high = response['result'][list(response["result"].keys())[0]][0][2]
        low = response['result'][list(response["result"].keys())[0]][0][3]
        close = response['result'][list(response["result"].keys())[0]][0][4]
    except HTTPError as e:
        print(str(e))
        
    return("Open: " + open + ", High: " + high + ", Low: " + low + ", Close: " + close)