U>qjrddlmZddlZddlmZmZddlmZddl m Z ddl m Z dd l mZmZGd d eZy) )print_functionN)DictOptional)utils)YfConfig)YfData)Domain _QUERY_URL_ceZdZdZddeffd ZdZedefdZedefdZ ede e jfdZ ede e jfd Zd ede e jfd Zd ede e jfd ZddZxZS)Industryz1 Represents an industry within a sector. regionct|tt||||td|j |_d|_d|_d|_ d|_ y)ab Args: key (str): The key identifier for the industry. session (optional): The session to use for requests. region (str): Yahoo region (ISO 3166-1 alpha-2 country code, e.g. "US", "GB", "FR", "DE", "JP"). Scopes top performing/growth company listings. Defaults to "US". )sessionz /industries/N) r superr__init__r _key _query_url _sector_key _sector_name_top_performing_companies_top_growth_companies)selfkeyrr __class__s W/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/yfinance/domain/industry.pyrzIndustry.__init__sV w h&sGV<(Mdii[A )-&%)"c"d|jdS)z Returns a string representation of the Industry instance. Returns: str: String representation of the Industry instance. zyfinance.Industry object <>)rrs r__repr__zIndustry.__repr__#s,DII;a88rreturncP|j|j|jS)zt Returns the sector key of the industry. Returns: str: The sector key. )_ensure_fetchedrr!s r sector_keyzIndustry.sector_key,s$ T--.rcP|j|j|jS)zv Returns the sector name of the industry. Returns: str: The sector name. )r%rr!s r sector_namezIndustry.sector_name7s$ T../   rcP|j|j|jS)z Returns the top performing companies in the industry. Returns: Optional[pd.DataFrame]: DataFrame containing top performing companies. )r%rr!s rtop_performing_companiesz!Industry.top_performing_companiesBs$ T;;<---rcP|j|j|jS)z Returns the top growth companies in the industry. Returns: Optional[pd.DataFrame]: DataFrame containing top growth companies. )r%rr!s rtop_growth_companieszIndustry.top_growth_companiesMs$ T778)))rr*c gd}|Dcgc]}|jdd|jdd|jdijdd|jdijdd|jdijddf}}|sytj|| jdScc}w) a- Parses the top performing companies data. Args: top_performing_companies (Dict): Dictionary containing top performing companies data. Returns: Optional[pd.DataFrame]: DataFrame containing parsed top performing companies data. )symbolname ytd returnz last pricez target pricer.Nr/ ytdReturnraw lastPrice targetPricecolumnsget_pd DataFrame set_index)rr*companies_columnccompanies_valuess r_parse_top_performing_companiesz(Industry._parse_top_performing_companiesXsV Rj kMN UU8T2UU640UU;r266udCUU;r266udCUU=488E Hkk  }}-9IJTTU]^^ksBCr,c Lgd}|Dcgc]h}|jdd|jdd|jdijdd|jdijddfj}}|sytj||jdScc}w) a Parses the top growth companies data. Args: top_growth_companies (Dict): Dictionary containing top growth companies data. Returns: Optional[pd.DataFrame]: DataFrame containing parsed top growth companies data. )r.r/r0zgrowth estimater.Nr/r1r2growthEstimater5r7)rr,r<r=r>s r_parse_top_growth_companiesz$Industry._parse_top_growth_companiesnsLUijPQUU8T2UU640UU;r266udCUU#3B7;;E4HKjj  }}-9IJTTU]^^jsA-B!cd} |j|j}|d}|j||jd|_|jd|_|j |jd|_|j|jd|_ |S#t$r}tjjstj}|j!d|j"d||jd |jd |jd ||jd Yd}~yd}~wwxYw) z7 Fetches and parses the industry data. Ndata sectorKey sectorNametopPerformingCompaniestopGrowthCompaniesz!Failed to get industry data for 'z ' reason: zGot response: z ------------- )_fetchr_parse_and_assign_commonr8rrr?rrBr Exceptionrdebughide_exceptionsr get_yf_loggererrorr)rresultrDeloggers r_fetch_and_parsezIndustry._fetch_and_parses# *[[1F&>D  ) )$ /#xx 4D  $ 6D -1-Q-QRVRZRZ[sRt-uD *)-)I)I$((SgJh)iD &M *>>11((*F LLrds(%!'P*vP*r