U>qj8*ddlmZmZddlZddlmZmZmZmZm Z ddl m Z m Z ddl m Z mZddl mZmZddlmZdd lmZmZed eeej.f Ze d ZGd deZGddeZGddeZGddeZy))ABCabstractmethodN)ListUnionDictTypeVarLiteral)EQUITY_SCREENER_EQ_MAPEQUITY_SCREENER_FIELDS)FUND_SCREENER_EQ_MAPFUND_SCREENER_FIELDS)ETF_SCREENER_EQ_MAPETF_SCREENER_FIELDSYFNotImplementedError)dynamic_docstring'generate_list_table_from_dict_universalT)bound) eqzis-inbtwngtltgtelteandorc|eZdZdedeedeeeejffdZ e e de fdZ e e de fdZdedddfdZdeeeejfddfd Zdeeeejfddfd Zdeeeejfddfd Zdedddfd Zde fd ZddefdZdefdZy) QueryBaseoperatoroperandc|j}t|ts tdt |dkr t d|dk(r|j |ne|dvr|j|nO|dk(r|j|n8|dk(r|j|n!|dvr|j|n t d ||_ ||_ y) NzInvalid operand typerzInvalid field for EquityQueryIS-IN>ORANDEQBTWN>GTLTGTELTEzInvalid Operator Value) upper isinstancelist TypeErrorlen ValueError_validate_isin_operand_validate_or_and_operand_validate_eq_operand_validate_btwn_operand_validate_gt_ltr!operands)selfr!r"s V/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/yfinance/screener/query.py__init__zQueryBase.__init__s>>#'4(23 3 w<1 <= = w   ' ' 0  %  ) )' 2    % %g .    ' ' 0 0 0   )56 6   returnctd)Nz/valid_fields() needs to be implemented by childrr9s r: valid_fieldszQueryBase.valid_fields($$UVVr<ctd)Nz/valid_values() needs to be implemented by childrr?s r: valid_valueszQueryBase.valid_values-rAr<Nct|dkr tdtd|Ddurtdt |dy)Nz$Operand must be length longer than 1c3<K|]}t|tywNr.r ).0es r: z5QueryBase._validate_or_and_operand..5s9Az!Y'9FzOperand must be type z for OR/AND)r1r2allr0typer9r"s r:r4z"QueryBase._validate_or_and_operand2sL w<1 CD D 99 9U B3DJ<{KL L Cr<ctdk7r tdtfd|jj Dstdt |dddd|j vro|j d}t|tr3tj|j Dcgc]}|c}}d|vrtd ddyycc}w) NrzOperand must be length 2 for EQc3,K|] }d|v ywrNrIfields_by_typer"s r:rKz1QueryBase._validate_eq_operand..<b^GAJ.0bInvalid field for  "r"rEInvalid EQ value ") r1r2anyr@valuesrNrCr.dictsetunion)r9r"vvrJs ` r:r5zQueryBase._validate_eq_operand8s w<1 >? ?btGXGXG_G_Gabb1$t*R |1MN N 1:** *""71:.B"d#!SU[[biik":1":;qz# #5gaj\!CDD$ + #;s> C&cftdk7r tdtfd|jj Dstdt |t dtjdur tdt dtjdur tdy) Nz!Operand must be length 3 for BTWNc3,K|] }d|v ywrRrSrTs r:rKz3QueryBase._validate_btwn_operand..JrVrWrXrEFz Invalid comparison type for BTWNr r1r2r\r@r]rNr.numbersRealr0rOs `r:r6z QueryBase._validate_btwn_operandGs w<1 @A AbtGXGXG_G_Gabb1$t*>? ? gaj',, /5 8>? ? gaj',, /5 8>? ? 9r<c tdk7r tdtfd|jj Dstdt |dddt dtjd ur td y) Nrz"Operand must be length 2 for GT/LTc3,K|] }d|v ywrRrSrTs r:rKz,QueryBase._validate_gt_lt..TrVrWrXrYrrZrEFz!Invalid comparison type for GT/LTrerOs `r:r7zQueryBase._validate_gt_ltQs w<1 AB BbtGXGXG_G_Gabb1$t*R |1MN N gaj',, /5 8?@ @ 9r<c tdkr tdtfd|jj Dstdt |dddd|j vr|j d}t|tr3tj|j Dcgc]}|c}}tdtD]}||vs td |dyycc}w) Nrz#Operand must be length 2+ for IS-INc3,K|] }d|v ywrRrSrTs r:rKz3QueryBase._validate_isin_operand..]rVrWrXrYrrZrEr[) r1r2r\r@r]rNrCr.r^r_r`range)r9r"rarJis ` r:r3z QueryBase._validate_isin_operandYs w DcD|j}|j}|jdk(r=d}|jddDcgc]#}t|d|jd|g%}}||Dcgc]$}t|tr|j n|&c}dScc}wcc}w)Nr$r%rEr'r)r!r8)r!r8rNr.r to_dict)r9opopsvos r:rozQueryBase.to_dictis ]]mm ==G #BDHMMRSRTDUVq:4:dT]]1%5q$9:VCVQTUA 1i(@aGU  WVs (B*)Bcdz|jj}t|jtrt d|jDr=dj fd|jD}|d|jd|ddS|d|jd t|jd S|d|jd t|jd S) N c3<K|]}t|tywrGrH)rIrps r:rKz%QueryBase.__repr__..{sE:b),ErLz, c3K|]8}dt|tr|jdzn t|:yw)rurEN)r.r __repr__repr)rIrpindent indent_strs r:rKz%QueryBase.__repr__..}sD*"l" 2y@YR[[!%<_cdf_g$hi*s>A(z, [  z])z, )) __class____name__r.r8r/r\joinr!ry)r9rz class_name operands_strr{s ` @r:rxzQueryBase.__repr__usF] ^^,, dmmT *Et}}EE$zz*"mm*  %Qt}}oU<.:,VXYY%Qt}}oRT]]8K7LANN!\4==/D4G3HJ Jr<c"|jSrG)rxr?s r:__str__zQueryBase.__str__s}}r<)r)r __module__ __qualname__Operatorrrstrrfrgr;propertyrrr@rCr4r5r6r7r3rorxrrSr<r:r r s\  E4 ;LdSViY]^e^j^jYk;k4m 0WdWWWdWWM[0AMdM EDsGLL7H1I,J Et E@d5gll9J3K.L@QU@AtE#w||2C,D'EA$AId;.?IDI    KCK(r<r ceZdZdZedeeiedefdZ edee ddgiedefd Z y ) EquityQuerya The `EquityQuery` class constructs filters for stocks based on specific criteria such as region, sector, exchange, and peer group. Start with value operations: `EQ` (equals), `IS-IN` (is in), `BTWN` (between), `GT` (greater than), `LT` (less than), `GTE` (greater or equal), `LTE` (less or equal). Combine them with logical operations: `AND`, `OR`. Example: Predefined Yahoo query `aggressive_small_caps`: .. code-block:: python from yfinance import EquityQuery EquityQuery('and', [ EquityQuery('is-in', ['exchange', 'NMS', 'NYQ']), EquityQuery('lt', ["epsgrowth.lasttwelvemonths", 15]) ]) valid_operand_fields_tabler=ctSz[ Valid operands, grouped by category. {valid_operand_fields_table} )r r?s r:r@zEquityQuery.valid_fields &%r<valid_values_tableexchangeindustry) concat_keysctSz Most operands take number values, but some have a restricted set of valid values. {valid_values_table} )r r?s r:rCzEquityQuery.valid_valuesrr<N) rrr__doc__rrr rrr@r rCrSr<r:rrs(46]^t6uvw &d&x&,.UVl|FHR{S/TUV &d&V&r<rceZdZdZedeeiedefdZ edee iedefdZ y) FundQuerya The `FundQuery` class constructs filters for mutual funds based on specific criteria such as region, sector, exchange, and peer group. Start with value operations: `EQ` (equals), `IS-IN` (is in), `BTWN` (between), `GT` (greater than), `LT` (less than), `GTE` (greater or equal), `LTE` (less or equal). Combine them with logical operations: `AND`, `OR`. Example: Predefined Yahoo query `solid_large_growth_funds`: .. code-block:: python from yfinance import FundQuery FundQuery('and', [ FundQuery('eq', ['categoryname', 'Large Growth']), FundQuery('is-in', ['performanceratingoverall', 4, 5]), FundQuery('lt', ['initialinvestment', 100001]), FundQuery('lt', ['annualreturnnavy1categoryrank', 50]), FundQuery('eq', ['exchange', 'NAS']) ]) rr=ctSr)r r?s r:r@zFundQuery.valid_fields $#r<rctSr)r r?s r:rCzFundQuery.valid_valuesrr<N) rrrrrrr rrr@r rCrSr<r:rrss,46]^r6stu $d$v$,.UVj.klm $d$n$r<rceZdZdZedeeiedefdZ edee iedefdZ y)ETFQuerya The `ETFQuery` class constructs filters for ETFs based on specific criteria such as category, fund family, exchange, and performance ratings. Start with value operations: `EQ` (equals), `IS-IN` (is in), `BTWN` (between), `GT` (greater than), `LT` (less than), `GTE` (greater or equal), `LTE` (less or equal). Combine them with logical operations: `AND`, `OR`. Example: Predefined Yahoo query `top_etfs_us`: .. code-block:: python from yfinance import ETFQuery ETFQuery('and', [ ETFQuery('gt', ['intradayprice', 10]), ETFQuery('is-in', ['performanceratingoverall', 4, 5]), ETFQuery('eq', ['region', 'us']) ]) rr=ctSr)rr?s r:r@zETFQuery.valid_fields #"r<rctSr)rr?s r:rCzETFQuery.valid_valuesrr<N) rrrrrrrrrr@rrCrSr<r:rrss(46]^q6rst #d#u#,.UVi.jkl #d#m#r<r)abcrrrftypingrrrrr yfinance.constr r r r rryfinance.exceptionsrutilsrrrrgrrr rrrrSr<r:rsz#66IEC5N CuS',,./0 O P{{|%&)%&P'$ '$R%#y%#r<