35 lines
911 B
Text
35 lines
911 B
Text
import json
|
|
import asyncio
|
|
from config import config
|
|
db = config['database']
|
|
|
|
def write_file(data):
|
|
with open(db, 'w') as f:
|
|
f.write(data)
|
|
|
|
def get_stocks(user):
|
|
with open(db) as f:
|
|
data = json.loads(f.read())
|
|
|
|
if user in data.keys():
|
|
user_data = data[user]
|
|
return user_data['portfolio']
|
|
else:
|
|
data[user] = {"portfolio": {}}
|
|
write_file(json.dumps(data))
|
|
return get_stocks(user)
|
|
|
|
def add_stock(user, stock, amount):
|
|
with open(db) as f:
|
|
data = json.loads(f.read())
|
|
if user in data.keys():
|
|
print(data.keys())
|
|
if stock in data[user]['portfolio'].keys():
|
|
data[user]['portfolio'][stock] += amount
|
|
else:
|
|
data[user]['portfolio'][stock] = amount
|
|
write_file(json.dumps(data))
|
|
return True
|
|
else:
|
|
get_stocks(user)
|
|
return add_stock(user, stock, amount)
|