feat: add yahoo to the available quote sources

glorat · Nov 13, 2022 · ca41105 · ca41105
1 parent 15c8e3e
commit ca41105
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/client/src/lib/assetdb/enums.ts b/client/src/lib/assetdb/enums.ts
@@ -44,6 +44,7 @@ export const quoteSourceTypes = [
   {value: 'av', label: 'Alpha Vantage'},
   {value: 'investpy', label: 'investpy'},
   {value: 'ft', label: 'ft.com symbol'},
+  {value: 'yahoo', label: 'Yahoo finance symbol'},
 ].map(cleanUp);
 
 export const investmentAssetTypes = [

diff --git a/python/requirements.txt b/python/requirements.txt
@@ -1,2 +1,3 @@
 investpy==1.0.8
 investiny==0.7.2
+yfinance==0.1.85
diff --git a/python/yahoo.py b/python/yahoo.py
@@ -18,12 +18,14 @@ def doit(args):
     # print(ticker, file=sys.stderr)
 
     t = yf.Ticker(ticker)
+    currency = t.info['currency']
 
     # df = t.history(period="1mo")
     df = t.history(period="max")
     df['timestamp'] = df.index.strftime('%Y-%m-%d')
     df['close'] = df['Close'].round(4)
-    ret = df.to_csv(columns=['timestamp', 'close'], index=False)
+    df['currency'] = currency
+    ret = df.to_csv(columns=['timestamp', 'close', 'currency'], index=False)
     return ret