diff --git a/mongo.py b/mongo.py index 8904bd9..b453ef6 100644 --- a/mongo.py +++ b/mongo.py @@ -43,7 +43,7 @@ def get_stocks(user): stocks = Portfolio.objects.get(user=user).stocks decoded_stocks = {} for stock, amount in stocks.items(): - decoded_stocks[decodeKey(stock)] = amount.to_decimal() + decoded_stocks[decode_key(stock)] = amount.to_decimal() return decoded_stocks @@ -55,7 +55,7 @@ def add_stock(user, stock, amount): if not (yfi.stock_exists(stock)): # Stock ticker does not exist on yahoo finance return False - stock = encodeKey(stock) + stock = encode_key(stock) if stock in portfolio.stocks.keys(): portfolio.stocks[stock] = Decimal128( amount + portfolio.stocks[stock].to_decimal() @@ -68,7 +68,7 @@ def add_stock(user, stock, amount): def delete_stock(user, stock): user_check(user) - stock = encodeKey(stock) + stock = encode_key(stock) portfolio = Portfolio.objects.get(user=user) if stock in portfolio.stocks.keys(): del portfolio.stocks[stock] @@ -84,7 +84,7 @@ def get_watchlist(user): wlist = Watchlist.objects.get(user=user).stocks decoded_stocks = {} for stock, val in wlist.items(): - decoded_stocks[decodeKey(stock)] = val.to_decimal() + decoded_stocks[decode_key(stock)] = val.to_decimal() return decoded_stocks @@ -96,7 +96,7 @@ def watch(user, stock, est_price="0"): if not (yfi.stock_exists(stock)): # Stock ticker does not exist on yahoo finance return False - stock = encodeKey(stock) + stock = encode_key(stock) watchlist.stocks[stock] = Decimal128(Decimal(est_price)) watchlist.save() return True