jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-11-10 11:21:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

improved market cap value extraction (#5169)

Browse Source 
* improved market cap value extraction

* pylint

---------

Co-authored-by: James Maslek <jmaslek11@gmail.com>
This commit is contained in:
Bim Phomthong
2023-11-08 23:50:55 +01:00
committed by GitHub
parent cfc40e7cc4
commit acf8b06689
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
openbb_terminal/portfolio/portfolio_optimization/optimizer_model.py
View File

@@ -343,18 +343,19 @@ def get_market_cap(symbol) -> float:
latest_year = market_cap.index[-1]
value = market_cap.loc[latest_year]["Market capitalization"]
values_str = str(value)
if "M" in value:
updated_value = float(value.split(" M")[0]) * 1000000
elif "B" in value:
updated_value = float(value.split(" B")[0]) * 1000000000
elif "T" in value:
updated_value = float(value.split(" T")[0]) * 1000000000000
# use values_str in string operations
if values_str.endswith("M"):
updated_value = float(values_str.split(" M", maxsplit=1)[0]) * 1000000
elif values_str.endswith("B"):
updated_value = float(values_str.split(" B", maxsplit=1)[0]) * 1000000000
elif values_str.endswith("T"):
updated_value = float(values_str.split(" T", maxsplit=1)[0]) * 1000000000000
else:
updated_value = float(value)
updated_value = float(values_str)
else:
updated_value = 0
return updated_value