o ޭh *@s6ddlmZddlZddlmZddlmZddlmZmZm Z m Z ddl m Z ddl mZmZmZddlmZmZdd lmZdd lmZmZmZmZerZdd lmZdd lmZee efZ dd d!Z#Gd"d#d#eZ$d?d'd(Z%d?d)d*Z&Gd+d,d,Z'Gd-d.d.Z(d@d4d5Z)dAd:d;Z*dS)B) annotationsN)Sequence)copy) TYPE_CHECKINGAnyCallablecast)Table) CriterionJSONAttributeCriterionTerm) FieldErrorOperationalError)Field)BackwardFKRelationForeignKeyFieldInstanceManyToManyFieldInstanceRelationalField)Model)QuerySettabler related_fieldrrelated_field_namestrreturnlist[TableCriterionTuple]c Cs&g}|jjj}t|tr5t|j}||||jjj ||j kf||||j ||jjj kf|St|t rb|j jpA|j j}||krR||d|}||||||jkf|S|j jpi|j j}|jjj|j}|jpw|j}||d|}||||||kf|S)N__) related_model_meta basetable isinstancerr throughappendmodel db_pk_column backward_key forward_keyrto_field_instance source_fieldmodel_field_nameas_get_table_namerelation_source_field fields_map) rrrrequired_joins related_table through_tableto_field_source_field from_fieldfrom_field_source_fieldr4Y/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/tortoise/query_utils.pyget_joins_for_related_field sV     %  r6 root_model type[Model]lookup_expressionSequence[Field]cCs|d}g}|}|ddD] }||jjvrt|dtt|jj|}|||j}qz||jj|dW|St yLt|dw)Nrz not resolvable) splitr fetch_fieldsr rrr-r"rKeyError)r7r9 field_namesfieldsr# field_namefieldr4r4r5expand_lookup_expressionWs    rCr#rB4tuple[Term, list[TableCriterionTuple], Field | None]c Cs<g}t||}|ddD]-}tt|}|t|||j|j}|jjj}t |t r8| | d|j}|}q |d}|j|jj vrxtt|}|jj} |t|||j| j}t |trr||krr| | d|j}|| j} n!|jr||j} n||j} |r||jjjjd} | r| || } | ||fS)z Resolves a nested field string like events__participants__name and returns the pypika term, required joins and the Field that can be used for converting the value. Nr;r function_cast)rCrrextendr6r)rrrr rr*r+r=rr$r(get_for_dialectdb capabilitiesdialect) r#rrBjoinsr@ iter_fieldrr/ last_fieldrelated_field_metatermfuncr4r4r5resolve_nested_fieldisD            rQc@s*eZdZd ddZd ddZd d d Zd S)EmptyCriterionotherr rcC|SNr4selfrSr4r4r5__or__zEmptyCriterion.__or__cCrTrUr4rVr4r4r5__and__rYzEmptyCriterion.__and__boolcCsdS)NFr4rWr4r4r5__bool__rYzEmptyCriterion.__bool__N)rSr rr )rr[)__name__ __module__ __qualname__rXrZr]r4r4r4r5rRs  rRleftr rightcCs|r|s|S||@SrUr4rarbr4r4r5_andrdcCs|r|s|S||BSrUr4rcr4r4r5_orrerfc@sJeZdZdZ   ddd d Zdd dZdddZdddZdddZdS) QueryModifierz: Internal structure used to generate SQL Queries. Nwhere_criterionCriterion | NonerK list[TableCriterionTuple] | Nonehaving_criterionrNonecCs&|pt|_|p g|_|pt|_dSrU)rRrhrKrk)rWrhrKrkr4r4r5__init__s  zQueryModifier.__init__rScCs,|jt|j|j|j|jt|j|jdS)N)rhrKrk) __class__rdrhrKrkrVr4r4r5rZs    zQueryModifier.__and__r cCst|j|jSrU)rdrhrkr\r4r4r5_and_criterionszQueryModifier._and_criterioncCsbd}}|js |jrt||}n|jr |jr |j|jBn|jp%|j}|||j|j|SrU)rkrfrorhrnrK)rWrSrhrkr4r4r5rXs  zQueryModifier.__or__cCs@d}}|jr|j|j@}n|jr|j}|||j|SrU)rkrhnegaternrK)rWrhrkr4r4r5 __invert__s  zQueryModifier.__invert__)NNN)rhrirKrjrkrirrl)rSrgrrg)rr )rrg) r^r_r`__doc__rmrZrorXrqr4r4r4r5rgs   rgc@s*eZdZdZdZddd d ZdddZdS)PrefetchaA Prefetcher container. One would directly use this when wanting to attach a custom QuerySet for specialised prefetching. :param relation: Related field name. :param queryset: Custom QuerySet to use for prefetching. :param to_attr: Sets the result of the prefetch operation to a custom attribute. )relationquerysetto_attrNrtrrurrv str | NonerrlcCs*||_||_||_t|jjjj|j_dSrU)rvrtrurr#r basequeryquery)rWrtrurvr4r4r5rmszPrefetch.__init__cCs|jd\}}}||jjjvrtd|d|jjjd|r<||jvr+t|j|<|j| t ||j |j ddS|j |g|j |j fdS)z Called internally to generate prefetching query. :param queryset: Custom QuerySet to use for prefetching. :raises OperationalError: If field does not exist in model. rz relation z for z not found)rvN)rt partitionr#rr=rdb_table _prefetch_mapsetaddrsrurv_prefetch_queries setdefaultr")rWrufirst_level_fieldrforwarded_prefetchr4r4r5resolve_for_querysets    zPrefetch.resolve_for_querysetrU)rtrrurrvrwrrl)rurrrl)r^r_r`rr __slots__rmrr4r4r4r5rss  rsvaluedict[str, Any]operator_keywords#dict[str, Callable[..., Criterion]]5tuple[list[str | int], Any, Callable[..., Criterion]]cCsP|\\}}dd|dD}|d|vr |t|dntj}|||fS)zZ Extracts the key parts, filter value and operator from a JSON filter dictionary. cSs$g|]}|r t|nt|qSr4)isdigitintr).0itemr4r4r5 s$z,get_json_filter_operator..rr;)itemsr<rpopoperatoreq)rrkey filter_value key_parts operator_r4r4r5get_json_filter_operators  r field_termr rlist[str | int]cCs t||S)z Resolves a JSON path from a list of key parts, e.g. converting (field, ['a', 'b', 'c']) to field->'a'->'b'->>'c'. Returns a pypika Term. )r )rrr4r4r5resolve_field_json_path!s r)rr rrrrrr)r7r8r9rrr:)r#r8rr rBrrrD)rar rbr rr )rrrrrr)rr rrrr )+ __future__rrcollections.abcrrtypingrrrrpypika_tortoiser pypika_tortoise.termsr r r tortoise.exceptionsr rtortoise.fields.basertortoise.fields.relationalrrrrtortoise.modelsrtortoise.querysetrtupleTableCriterionTupler6rCrQrRrdrfrgrsrrr4r4r4r5s0          7 7 . ,