diff --git a/server.py b/server.py index dfbc043..7048bed 100755 --- a/server.py +++ b/server.py @@ -694,7 +694,43 @@ def saveIpoAPIKey(): f.write(str(key)) return index() + +@app.route("/savePortfolioSettings", methods = ['PUT', 'POST']) +def savePortfolioSettings(): + + data= request.data.decode('utf-8') + settings = json.loads(data) + #THIS IS TO CREATE PORTFOLIO JSON FILE IF IT DOESN'T EXIST + initialize_json = '{"symbols":{}}' + if 'portfolio_settings.json' not in os.listdir('csv/'): + with open('csv/portfolio_settings.json', 'w') as f: + f.write(initialize_json) + + try: + f = open('csv/portfolio_settings.json') + portfolio = json.load(f) + f.close() + + shares1 = settings['shares'] + cost1 = settings['cost'] + symbol1 = settings['symbol'] + + days1 = settings['days'] + day_start = datetime.strptime(days1, "%Y-%m-%d") + day_today = datetime.strptime(datetime.now(pytz.utc).strftime("%Y-%m-%d"), "%Y-%m-%d") + days1 = str((day_today - day_start).days) + + portfolio['symbols'][symbol1] = {'shares':shares1, 'day':days1, 'cost':cost1} + + f = open("csv/portfolio_settings.json", 'w+') + json.dump(portfolio, f) + f.close() + except: + pass + + return index() + @app.route("/screensaver", methods = ['PUT', 'POST']) def screensaver(): global displaying_screensaver