WgtddlmZmZddlmZddlmZddlmZddl m Z m Z m Z ddl ZedZGd d eZy) )ABCabstractmethod)Ticker) _QUERY1_URL_)YfData)DictListOptionalNz /v1/financecBeZdZdZddefdZedefdZedefdZedefdZ ede fd Z ede fd Z edeej fd Zedee eeffd Zde fd ZddZde 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. Nkeyc||_||_||_t||_d|_d|_d|_d|_d|_ y)aK Initializes the Domain object with a key, session, and proxy. Args: key (str): Unique key identifying the domain entity. session (Optional[requests.Session]): Session object for HTTP requests. Defaults to None. proxy (Optional[Dict]): Proxy settings. Defaults to None. )sessionN) _keyproxyrr_data_name_symbol _overview_top_companies_research_reports)selfrrrs I/var/www/html/venv/lib/python3.12/site-packages/yfinance/domain/domain.py__init__zDomain.__init__sJ   #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_fetchedrrs rnamez Domain.name.s  TZZ(zzrcP|j|j|jS)z Retrieves the symbol of the domain entity. Returns: str: The symbol representing the domain entity. )r!rrs rsymbolz Domain.symbol9s  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!rrrs rtickerz Domain.tickerDs% 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!rrs roverviewzDomain.overviewOs  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!rrs r top_companieszDomain.top_companiesZs$ 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!rrs rresearch_reportszDomain.research_reportses$ T334%%%rczddddd}|jj||jj||}|S)a  Fetches data from the given query URL. Args: query_url (str): The URL used for the data query. proxy (Dict): Proxy settings for the request. Returns: Dict: The JSON response data from the request. truezen-USUS) formatted withReturnslangregion)user_agent_headersparamsr)r get_raw_jsonr4)r query_urlr params_dictresults r_fetchz Domain._fetchpsH%+67^bc ((tzzGdGdmxAF(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)NotImplementedErrorrs 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)NN)rN)__name__ __module__ __qualname____doc__strrpropertyrr"r$rr&r r(r rUrVr*r r,r:rBr?r@rr]r!rrr r s7 FCF(Sc$$$$#x 6##&$tCH~"6&& $  = 4 (eXcmm5Le*dd$rr )abcrrr&rconstrrArtypingr r r pandasrU _QUERY_URL_r rfrrrls4# ''k* {$S{$r