add symbol to portfolio settings

This commit is contained in:
Justin 2023-03-16 19:00:25 +08:00 committed by GitHub
parent 9b9ca1c153
commit d0fbaa74a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -695,6 +695,42 @@ def saveIpoAPIKey():
return index() 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']) @app.route("/screensaver", methods = ['PUT', 'POST'])
def screensaver(): def screensaver():
global displaying_screensaver global displaying_screensaver