assignment2/server.py at master · delagroove/assignment2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
from app import create_app
from flask_socketio import send  # , emit
from pymongo import MongoClient
import requests
import json
from bson import json_util
from collections import namedtuple

socketio, app = create_app()
db = None

def _json_object_hook(d): return namedtuple('X', d.keys())(*d.values())
def json2obj(data): return json.loads(data, object_hook=_json_object_hook)

@socketio.on('connect')
def connect():
    client = MongoClient(port=27017)
    db = client.crypto
    send('connected')
    return 'connected'


@socketio.on('coin-list')
def coinList():
    res = requests.get('https://min-api.cryptocompare.com/data/all/coinlist')
    res_json = [x for x in res.json()['Data'].values() if int(x['SortOrder']) < 100]
    newlist = sorted(res_json, key=lambda k: k['SortOrder'])
    return newlist


@socketio.on('select-coin')
def selectCoin(thecoin):
    res = requests.get("https://min-api.cryptocompare.com/data/histoday?fsym="+thecoin+"&tsym=USD&limit=100")
    return res.text


@socketio.on('query-bank')
def queryBank():
    client = MongoClient(port=27017)
    db = client.crypto
    result = db.settings.find_one()
    return json.dumps(result, sort_keys=True, indent=4, default=json_util.default)


@socketio.on('bank-add')
def bankAdd(amount):
    client = MongoClient(port=27017)
    db = client.crypto
    result = db.settings.find_one()
    res = json.dumps(result, sort_keys=True, indent=4, default=json_util.default)
    result_json = json.loads(res)
    result_json['cash'] = result_json['cash'] + amount
    print(result_json['cash'])
    result = db.settings.update_one({}, {"$set": {"cash": result_json['cash']}})
    return result_json['cash']


@socketio.on('bank-remove')
def bankRemove(amount):
    client = MongoClient(port=27017)
    db = client.crypto
    result = db.settings.find_one()
    res = json.dumps(result, sort_keys=True, indent=4, default=json_util.default)
    result_json = json.loads(res)
    result_json['cash'] = result_json['cash'] - amount
    print(result_json['cash'])
    result = db.settings.update_one({}, {"$set": {"cash": result_json['cash']}})
    return result_json['cash']


@socketio.on('set-buy')
def setBuy(values):
    client = MongoClient(port=27017)
    db = client.crypto
    result = db.orders.insert(values)
    return json.dumps(result, sort_keys=True, indent=4, default=json_util.default)


@socketio.on('set-sell')
def setSell(values):
    client = MongoClient(port=27017)
    db = client.crypto
    result = db.orders.insert(values)
    return json.dumps(result, sort_keys=True, indent=4, default=json_util.default)


@socketio.on('get-blotter')
def getBlotter():
    client = MongoClient(port=27017)
    db = client.crypto
    result = db.orders.find()
    return json.dumps(list(result), sort_keys=True, indent=4, default=json_util.default)


@socketio.on('get-pl')
def getPl(coin):
    client = MongoClient(port=27017)
    db = client.crypto
    result = db.orders.find()
    return json.dumps(list(result), sort_keys=True, indent=4, default=json_util.default)


@socketio.on('query-portfolio')
def queryPortfolio(coin):
    client = MongoClient(port=27017)
    db = client.crypto
    result = db.orders.find({"symbol": coin})
    return json.dumps(list(result), sort_keys=True, indent=4, default=json_util.default)


if __name__ == '__main__':
    socketio.run(app)


def num_shares(trades, symbol):
    shares = 0
    for trade in trades:

        if trade.side == 'BUY' and trade.symbol == symbol:
            shares += int(trade.quantity)
        if trade.side == 'SELL' and trade.symbol == symbol:
            shares -= int(trade.quantity)
    return shares


def sold_shares(trades, symbol):
    shares = 0
    for trade in trades:
        if trade.side == 'SELL' and trade.symbol == symbol:
            shares += int(trade.quantity)
    return shares


def pl(market, the_wap, shares):
    return float((market-the_wap)*int(shares))


def wap(trades, symbol):
    the_wap = 0
    acum = 0
    for trade in trades:
        if trade.side == 'BUY' and trade.symbol == symbol:
            the_wap += int(trade.quantity)*trade.price
            acum += int(trade.quantity)
    if acum > 0:
        return float(the_wap/acum)
    else:
        return float(0.00)