daniil-berg/mwfin
2
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

company financials returned as separate dicts; test fix

Browse Source
This commit is contained in:
Maximilian Fajnberg
2021-11-28 15:20:28 +01:00
parent 326b956be4
commit 2d9119a4be
2 changed files with 16 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/mwfin/functions.py
View File

@ -1,5 +1,5 @@
import logging
from typing import Union, List
from typing import Union, List, Dict
from aiohttp.client import ClientSession
from bs4 import BeautifulSoup
@ -63,7 +63,7 @@ def extract_row_data(tr: Tag) -> tuple[str, tuple[int]]:
item_name = str(tr.td.div.string).strip()
data_div = tr.find_all('td')[-1].div.div
values_str: str = data_div.attrs['data-chart-data']
values = tuple(int(float(s)) for s in values_str.split(','))
values = tuple(int(float(s if s != '' else 0)) for s in values_str.split(','))
return item_name, values
@ -115,11 +115,12 @@ async def get_cash_flow_statement(ticker_symbol: str, quarterly: bool = False,
async def get_company_financials(ticker_symbol: str, quarterly: bool = False,
session: ClientSession = None) -> ResultDict:
session: ClientSession = None) -> Dict[str, ResultDict]:
"""
Returns all fundamentals (balance sheet, income statement and cash flow statement) of the specified company.
"""
financials = await get_balance_sheet(ticker_symbol, quarterly, session)
financials.update(await get_income_statement(ticker_symbol, quarterly, session))
financials.update(await get_cash_flow_statement(ticker_symbol, quarterly, session))
return financials
return {
constants.BS: await get_balance_sheet(ticker_symbol, quarterly, session),
constants.IS: await get_income_statement(ticker_symbol, quarterly, session),
constants.CF: await get_cash_flow_statement(ticker_symbol, quarterly, session)
}