Upload data.py

core/data.py

@@ -0,0 +1,16 @@
+import pandas as pd
+yf_available = True
+try:
+    import yfinance as yf
+except:
+    yf_available = False
+
+def get_data(src, ticker, file_obj, start, end):
+    if src == "Yahoo Finance" and yf_available:
+        df = yf.download(ticker, start=start, end=end)
+        df = df[['Close']].dropna().reset_index()
+    else:
+        df = pd.read_csv(file_obj.name, parse_dates=['Date'])
+        df = df[['Date', 'Close']]
+    df.columns = ['Date', 'Value']
+    return df