From d0fbaa74a43de5a4ded73d446d89d174d6ae6223 Mon Sep 17 00:00:00 2001
From: Justin <justinodunn@hotmail.com>
Date: Thu, 16 Mar 2023 19:00:25 +0800
Subject: [PATCH] add symbol to portfolio settings

---
 server.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

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