Update app.py

app.py

@@ -1,38 +1,33 @@
-
-import streamlit as st  # for overall GUI
-from streamlit_calendar import calendar  # to show calendar
-from dateutil.relativedelta import relativedelta  # for addition to dates
+import streamlit as st # for overall GUI
+from streamlit_calendar import calendar # to show calendar
+from dateutil.relativedelta import relativedelta # for addition to dates
 import datetime
-import os  # for extracting environment variable
-
-import json  # for parsing data from FMP API
-import urllib.request  # for getting data from FMP API
+import os # for extracting environment variable
+from urllib.request import urlopen # for getting data from FMP API
+import json # for parsing data from FMP API
 
 # For parsing data from API from JSON to a Python Dictionary
 def get_jsonparsed_data(url):
-    headers = {'User-Agent': 'Mozilla/5.0'}
-    request = urllib.request.Request(url, headers=headers)
-    response = urllib.request.urlopen(request)
+    response = urlopen(url)
     data = response.read().decode("utf-8")
     return json.loads(data)
 
 # Get FMP API stored as environment variable
-apiKey = os.getenv('FMP_API_KEY')
-if not apiKey:
-    st.error("API Key not found. Please set the FMP_API_KEY environment variable.")
-    st.stop()
+apiKey = os.environ['FMP_API_KEY']
 
 # Financialmodelingprep (FMP) api base url
 base_url = "https://financialmodelingprep.com/api/v3/"
 
+
 # Get today's date and add 3 months to it
 # Convert both today's date and the 3 months later date to strings (for input into API endpoint URL later)
 # This is the date range within which we want to get our earnings dates 
+
 today = datetime.datetime.today()
 today_string = today.strftime('%Y-%m-%d')
 future_string = (today + relativedelta(months=3)).strftime('%Y-%m-%d')
 
-# This is the full API endpoint to get the earnings dates from today to 3 months after
+# This is the full API endpoint to get the earnings dates from today to 6 months after
 url = f"{base_url}earning_calendar?from={today_string}&to={future_string}&apikey={apiKey}"
 
 # This decorator ensures that the call to the FMP API will only run once at the start of this app
@@ -42,18 +37,22 @@ url = f"{base_url}earning_calendar?from={today_string}&to={future_string}&apikey
 def get_earnings_dates(url):
     events = get_jsonparsed_data(url)
     return events
-
+    
 events = get_earnings_dates(url)
 
+
 with st.sidebar:
     st.title("미국 주식 어닝 달력")
     st.header("알고싶은 미국주식 티커를 입력해주세요~")
+    #tickers = ['GOOG', 'META', 'TSLA', 'NET', 'V', 'MA', 'BA', 'C']
+    
+    # For users to enter tickers of interest
     tickers_string = st.text_area('달력에 포함될 티커를 ,로 구분해서 입력해주세요~ \
 								예 "NVDA, META, AMZN," 입력후 컨트롤 + 엔터 또는 "적용" 버튼 클릭 ', 
-                                value='TSM,NVDA,AMD,INTC,MU,AAPL,GOOGL,FSLR,ARM,QCOM,MSFT,AMAT,LRCX,SNPS,CDNS,TSLA,ETN,FLNC,ANET,NXT,GEV,CRM,').upper()
+                                value = 'TSM,NVDA,AMD,INTC,MU,AAPL,GOOGL,FSLR,ARM,QCOM,MSFT,AMAT,LRCX,SNPS,CDNS,TSLA,ETN,FLNC,ANET,NXT,GEV,CRM,').upper()
     st.button("적용")
     st.write("기본세팅: ")
-    st.write("TSMC, Nvidia, AMD, Intel, Micron, Apple, 알파벳, FirstSolar, ARM, Qualcomm, MS, Applied mat, Lam research, Synopsys, Cadence design, Tesla, Eaton, Fluence energy, Arista net, Nextracker, GE 베르노바, SalesForce")
+    st.write("TSMC, Nvidia, AMD, Intel, Micron, Apple, 알파벳,FirstSolar, ARM, Qualcomm, MS, Applied mat, Lam research, Synopsys, Cadence design, Tesla, Eaton,Fluence energy, Arista net, Nextracker, GE베르노바, SalesForce") 
     st.write("참고: Earnings 날짜는 오늘부터 3개월 이내로 제한됨")
     st.write('')
     st.write("달력은 미국시간 기준입니다.")
@@ -62,6 +61,18 @@ with st.sidebar:
     st.write("[장후: 한국시간 익일 아침]")
     st.write('')
 
+
+    
+    #st.markdown("## [Explanatory Article](https://medium.datadriveninvestor.com/build-a-stock-earnings-calendar-of-your-favorite-stocks-in-python-36bba1950a61)")
+    st.write('')
+
+    # Where the data came from
+    #st.markdown("## [Financial Modeling Prep API](https://intelligence.financialmodelingprep.com/pricing-plans?couponCode=damianboh&utm_campaign=damianboh&utm_medium=blog&utm_source=medium)\
+    #\n\nEarnings Dates for all tickers are obtained from the FinancialModelingPrep API, feel free to sign up\
+    #[here](https://intelligence.financialmodelingprep.com/pricing-plans?couponCode=damianboh&utm_campaign=damianboh&utm_medium=blog&utm_source=medium)\
+    #if you wish.")
+   
+# Parse user input into a list
 tickers_string = tickers_string.replace(' ', '')
 tickers = tickers_string.split(',')
 
@@ -71,27 +82,30 @@ for event in events:
     if event['symbol'] in tickers:
         calendar_event = {}
         calendar_event['title'] = event['symbol']
-        if event['time'] == 'bmo':  # before market opens, add sunrise symbol
+        if event['time'] == 'bmo': # before market opens, add sunrise symbol
             calendar_event['title'] = '장전' + calendar_event['title']
-        elif event['time'] == 'amc':  # after market closes, add sunset symbol
-            calendar_event['title'] = '장후' + calendar_event['title']
+        elif event['time'] == 'amc': # after market closes, add sunset symbol
+            calendar_event['title'] = '장후'   + calendar_event['title']     
         calendar_event['start'] = event['date']
         calendar_events.append(calendar_event)
 
 st.header("미국 주식 어닝 달력")
 
+
 calendar_options = {
-    "editable": "true",
-    "navLinks": "true",
-    "headerToolbar": {
-        "left": "today prev,next",
-        "center": "title",
-        "right": "dayGridDay,dayGridWeek,dayGridMonth,listMonth",
-    },
-    "initialView": "dayGridMonth"
-}
-
-custom_css = """
+        "editable": "true",
+        "navLinks": "true",
+        "headerToolbar": {
+            "left": "today prev,next",
+            "center": "title",
+            "right": "dayGridDay,dayGridWeek,dayGridMonth,listMonth",
+        },
+        #"initialDate": today.strftime('%Y-%m-%d'),
+        "initialView": "dayGridMonth"
+    }
+    
+
+custom_css="""
     .fc-event-past {
         opacity: 0.8;
     }
@@ -106,4 +120,5 @@ custom_css = """
     }
 """
 
+
 calendar = calendar(events=calendar_events, options=calendar_options, custom_css=custom_css)