Update app.py

app.py

@@ -1,30 +1,23 @@
-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
-###from urllib.request import urlopen # for getting data from FMP API
-import json # for parsing data from FMP API
-import urllib.request
+import os  # for extracting environment variable
 
+import json  # for parsing data from FMP API
+import urllib.request  # for getting 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)
     data = response.read().decode("utf-8")
     return json.loads(data)
-    
-    
-    ###response = urlopen(url)
-    ###data = response.read().decode("utf-8")
-    ###return json.loads(data)
 
 # Get FMP API stored as environment variable
-###apiKey = os.environ['FMP_API_KEY']
-apiKey = os.getenv['FMP_API_KEY']
+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()
@@ -32,16 +25,14 @@ if not apiKey:
 # 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 6 months after
+# This is the full API endpoint to get the earnings dates from today to 3 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
@@ -51,22 +42,18 @@ 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)
 
+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("달력은 미국시간 기준입니다.")
@@ -75,18 +62,6 @@ 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(',')
 
@@ -96,30 +71,27 @@ 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",
-        },
-        #"initialDate": today.strftime('%Y-%m-%d'),
-        "initialView": "dayGridMonth"
-    }
-    
-
-custom_css="""
+    "editable": "true",
+    "navLinks": "true",
+    "headerToolbar": {
+        "left": "today prev,next",
+        "center": "title",
+        "right": "dayGridDay,dayGridWeek,dayGridMonth,listMonth",
+    },
+    "initialView": "dayGridMonth"
+}
+
+custom_css = """
     .fc-event-past {
         opacity: 0.8;
     }
@@ -134,5 +106,4 @@ custom_css="""
     }
 """
 
-
 calendar = calendar(events=calendar_events, options=calendar_options, custom_css=custom_css)