app.py

from flask import Flask, render_template
import trader
from datetime import datetime, timedelta
import threading
import math
from time import sleep

app = Flask(__name__)
trader.setup("BTCUSD")
Time = datetime.now()

curr_price = trader.quote()
THRESHOLD = 0.6/100
bought = True
buys = [[math.floor(datetime.now().timestamp() * 1000), curr_price]]
data = [[math.floor(datetime.now().timestamp() * 1000), curr_price]]
Buy_price = curr_price
sell_price = curr_price
sells = []
balance = 0
print(f"current price: {curr_price}")
investment = curr_price

def buy():
    global bought, curr_price, Buy_price, balance, sell_price
    if not bought:
        Buy_price = curr_price
        balance = (balance * 100 - Buy_price * 100) / 100
        bought = True
        print("buy", curr_price, balance)
        buys.append([math.floor(datetime.now().timestamp() * 1000), curr_price])
        

def sell():
    global bought, curr_price, Buy_price, balance, sell_price
    if bought:
        sell_price = curr_price
        balance = balance + sell_price
        bought = False
        print("sell", curr_price, balance)
        sells.append([math.floor(datetime.now().timestamp() * 1000), curr_price])

def stock_log():
    global bought, curr_price, Buy_price, balance, sell_price
    print("value: "+str(curr_price), "Stock held: " + str(bought),
                    "balance: "+str(balance))
    
def getNAV():
    global bought, curr_price, Buy_price, balance, sell_price
    if bought:
        return balance + curr_price
    else:
        return balance

def main():
    global bought, curr_price, Buy_price, balance, sell_price
    while True:
        if trader.quote() < curr_price:
            curr_price = trader.quote()
            data.append([math.floor(datetime.now().timestamp() * 1000), curr_price])
            if bought and curr_price > Buy_price and (curr_price * 100 - Buy_price * 100) / 100 >= THRESHOLD * curr_price:
                sell()
            else:
                stock_log()
        elif trader.quote() > curr_price:
            curr_price = trader.quote()
            data.append([math.floor(datetime.now().timestamp() * 1000), curr_price])
            if not bought and curr_price < sell_price and (sell_price * 100 - curr_price * 100) / 100 >= (THRESHOLD/2) * curr_price:
                buy()
            else:
                stock_log()
        


@app.route("/")
def home():
    return render_template("zing.jinja")

@app.route("/investment")
def initial():
    global investment
    return { "initial":  investment }

@app.route("/info", methods=["POST"])
def info():
    return {
        "data": [
            {
                "type": "line",
                "values": data
            },
            {
                "type": "scatter",
                "values": buys,
                "marker": {
                    "size": 8,
                    "background-color": "#00ff00",
                }
            },
            {
                "type": "scatter",
                "values": sells,
                "marker": {
                    "size": 8,
                    "background-color": "#ff0000",
                }
            }
        ],
        "bought": bought,
        "balance": balance,
        "NAV": getNAV()
        }

@app.route("/buy", methods = ["POST"])
def buy_req():
    buy()
    return { "bought": bought }

@app.route("/sell", methods = ["POST"])
def sell_req():
    sell()
    return { "bought": bought }

# def decimate():
#     global data
#     start = data[0][0] % 5
#     while True:
#         sleep(60)
#         print(len(data))
#         for i in data:
#             if (datetime.utcfromtimestamp(i[0]/1000).minute % 5 != start) and not i in buys:
#                 print(datetime.utcfromtimestamp(i[0]/1000).minute)
#                 data.remove(i)
#         print(len(data))


threading.Thread(target=main).start()
# simple = threading.Thread(target=decimate)
# simple.start()

if __name__ == "__main__":
    app.run()