U>qjtddlmZmZddlZddlmZmZmZddl m Z ddl m Z ddl mZe dZGd d eZy) )ABCabstractmethodN)DictListOptional) _QUERY1_URL_)YfData)Tickerz /v1/financecFeZdZdZddedefdZedefdZedefdZedefd Z ede fd Z ede fd Z edeej fd Zedee eeffd Zde fdZddZde fdZdeej fdZeddZddZy)Domainz Abstract base class representing a domain entity in financial data, with key attributes and methods for fetching and parsing data. Derived classes must implement the `_fetch_and_parse()` method. Nkeyregionc||_||_|jj|_t ||_d|_d|_d|_ d|_ d|_ y)a Initializes the Domain object with a key, session, and region. Args: key (str): Unique key identifying the domain entity. session (Optional[requests.Session]): Session object for HTTP requests. Defaults to None. region (str): Yahoo region (ISO 3166-1 alpha-2 country code, e.g. "US", "GB", "FR", "DE", "JP"). Determines the regional scope of returned data such as ``top_companies``. Defaults to "US". )sessionN) _keyrstripupper_regionr _data_name_symbol _overview_top_companies_research_reports)selfrrrs U/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/yfinance/domain/domain.py__init__zDomain.__init__sY  "LLN002 #G4 $( &* )-7;AEreturnc|jS)z Retrieves the key of the domain entity. Returns: str: The unique key of the domain entity. )rrs rrz Domain.key'syyrcP|j|j|jS)z| Retrieves the name of the domain entity. Returns: str: The name of the domain entity. )_ensure_fetchedrr"s rnamez Domain.name1s  TZZ(zzrcP|j|j|jS)z Retrieves the symbol of the domain entity. Returns: str: The symbol representing the domain entity. )r$rr"s rsymbolz Domain.symbol<s  T\\*||rcb|j|jt|jS)z Retrieves a Ticker object based on the domain entity's symbol. Returns: Ticker: A Ticker object associated with the domain entity. )r$rr r"s rtickerz Domain.tickerGs% T\\*dll##rcP|j|j|jS)z Retrieves the overview information of the domain entity. Returns: Dict: A dictionary containing an overview of the domain entity. )r$rr"s roverviewzDomain.overviewRs  T^^,~~rcP|j|j|jS)z Retrieves the top companies within the domain entity. Returns: pandas.DataFrame: A DataFrame containing the top companies in the domain. )r$rr"s r top_companieszDomain.top_companies]s$ T001"""rcP|j|j|jS)z Retrieves research reports related to the domain entity. Returns: List[Dict[str, str]]: A list of research reports, where each report is a dictionary with metadata. )r$rr"s rresearch_reportszDomain.research_reportshs$ T334%%%rcbddd|jd}|jj||}|S)z Fetches data from the given query URL. Args: query_url (str): The URL used for the data query. Returns: Dict: The JSON response data from the request. truezen-US) formatted withReturnslangr)params)rr get_raw_json)r query_url params_dictresults r_fetchz Domain._fetchss7%+67^b^j^jk ((;(G rc |jd|_|jd|_|j|jdi|_|j |jdi|_|jd|_y)z Parses and assigns common data fields such as name, symbol, overview, and top companies. Args: data (Dict): The raw data received from the API. r%r'r+ topCompaniesresearchReportsN)getrr_parse_overviewr_parse_top_companiesrr)rdatas r_parse_and_assign_commonzDomain._parse_and_assign_commonsrXXf% xx) --dhhz2.FG"77QS8TU!%*;!<rc V|jdd|jdijdd|jdd|jdd|jdd|jdijdd|jd ijddd S) z Parses the overview data for the domain entity. Args: overview (Dict): The raw overview data. Returns: Dict: A dictionary containing parsed overview information. companiesCountN marketCaprawmessageBoardId descriptionindustriesCount marketWeight employeeCount)companies_count market_capmessage_board_idrHindustries_count market_weightemployee_count)r>)rr+s rr?zDomain._parse_overviews (||, E%\\."=AA%N&ll?B?CCE4P  rc &gd}|Dcgc]U}|jd|jd|jd|jdijddfW}}|sytj||jdScc}w) a) Parses the top companies data and converts it into a pandas DataFrame. Args: top_companies (Dict): The raw top companies data. Returns: Optional[pandas.DataFrame]: A DataFrame containing top company data, or None if no data is available. )r'r%ratingz market weightr'r%rSrJrFN)columns)r>_pd DataFrame set_index)rr-top_companies_columnctop_companies_valuess rr@zDomain._parse_top_companiess MTa bOP"#x !f !h !nR 8 < _fetch_and_parse() needs to be implemented by children classes)NotImplementedErrorr"s r_fetch_and_parsezDomain._fetch_and_parses ""bccrc*||jyy)z Ensures that the given attribute is fetched by calling `_fetch_and_parse()` if the attribute is None. Args: attribute: The attribute to check and potentially fetch. N)r])r attributes rr$zDomain._ensure_fetcheds    ! ! # r)NUS)r N)__name__ __module__ __qualname____doc__strrpropertyrr%r'r r)rr+rrUrVr-rr/r:rBr?r@rr]r$rrr r s? FCFsF,Sc$$$$#x 6##&$tCH~"6&& 4  = 4 (eXcmm5Le*dd$rr )abcrrpandasrUtypingrrrconstr rAr r)r _QUERY_URL_r rgrrrms4#'' k* |$S|$r