Ej|ddlmZmZddlZddlmZmZmZddl m Z ddl m Z ddl mZe dZGd d eZdS) )ABCabstractmethodN)DictListOptional) _QUERY1_URL_)YfData)Tickerz /v1/financecxeZdZdZddefdZedefdZedefdZedefdZ ede fd Z ede fd Z edeejfd Zedee eeffd Zde fd ZddZde fdZdeejfdZeddZddZdS)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. Nkeyc||_||_t||_d|_d|_d|_d|_d|_dS)z Initializes the Domain object with a key, session. Args: key (str): Unique key identifying the domain entity. session (Optional[requests.Session]): Session object for HTTP requests. Defaults to None. )sessionN) _keyrr _data_name_symbol _overview_top_companies_research_reports)selfrrs R/var/www/html/bolsaweb/venv/lib/python3.11/site-packages/yfinance/domain/domain.py__init__zDomain.__init__sO  #G444 $( &* )-7;AEreturnc|jS)z Retrieves the key of the domain entity. Returns: str: The unique key of the domain entity. )rrs rrz Domain.key#s yrcD||j|jS)z| Retrieves the name of the domain entity. Returns: str: The name of the domain entity. )_ensure_fetchedrrs rnamez Domain.name-s" TZ(((zrcD||j|jS)z Retrieves the symbol of the domain entity. Returns: str: The symbol representing the domain entity. )r rrs rsymbolz Domain.symbol8s" T\***|rc^||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 rs rtickerz Domain.tickerCs* T\***dl###rcD||j|jS)z Retrieves the overview information of the domain entity. Returns: Dict: A dictionary containing an overview of the domain entity. )r rrs roverviewzDomain.overviewNs" T^,,,~rcD||j|jS)z Retrieves the top companies within the domain entity. Returns: pandas.DataFrame: A DataFrame containing the top companies in the domain. )r rrs r top_companieszDomain.top_companiesYs$ T0111""rcD||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 rrs rresearch_reportszDomain.research_reportsds$ T3444%%rcLddddd}|j||}|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-USUS) formatted withReturnslangregion)params)r get_raw_json)r query_url params_dictresults r_fetchz Domain._fetchos5%+67^bcc ((;(GG rcZ|d|_|d|_||di|_||di|_|d|_dS)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_common}sXXf%% xx)) --dhhz2.F.FGG"77QS8T8TUU!%*;! E E%\\."==AA%NN&ll?B??CCE4PP   rc|gd}d|D}|sdStj||dS)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 weightc g|]i}|d|d|d|diddfjS)r#r!rRrHrDNrP).0cs r z/Domain._parse_top_companies..s} b b bOP"#x !f !h !nR 8 8 < zDomain._parse_top_companiessn MLL b bTa b b b $ 4}1;OPPPZZ[cdddrc td)z Abstract method for fetching and parsing domain-specific data. Must be implemented by derived classes. z>_fetch_and_parse() needs to be implemented by children classes)NotImplementedErrorrs r_fetch_and_parsezDomain._fetch_and_parses ""bcccrc6||dSdS)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)N)rN)__name__ __module__ __qualname____doc__strrpropertyrr!r#r r%rr'rrXrYr)rr+r8r@r=r>rr_r rrr r s FFCFFFF$SXcXX$$$$X$$X#x 6###X#&$tCH~"6&&&X& 4     = = = = 4    (eXcm5Leeee*ddd^d$$$$$$rr )abcrrpandasrXtypingrrrconstr r?r r%r _QUERY_URL_r rhrrrns########'''''''''' *** x$x$x$x$x$Sx$x$x$x$x$r