j{0dZddlmZGddZdS)a QueryChain is a wrapper for sequence of queries. Features: * Easy iteration for sequence of queries * Limit, offset and count which are applied to all queries in the chain * Smart __getitem__ support Initialization ^^^^^^^^^^^^^^ QueryChain takes iterable of queries as first argument. Additionally limit and offset parameters can be given :: chain = QueryChain([session.query(User), session.query(Article)]) chain = QueryChain( [session.query(User), session.query(Article)], limit=4 ) Simple iteration ^^^^^^^^^^^^^^^^ :: chain = QueryChain([session.query(User), session.query(Article)]) for obj in chain: print obj Limit and offset ^^^^^^^^^^^^^^^^ Lets say you have 5 blog posts, 5 articles and 5 news items in your database. :: chain = QueryChain( [ session.query(BlogPost), session.query(Article), session.query(NewsItem) ], limit=5 ) list(chain) # all blog posts but not articles and news items chain = chain.offset(4) list(chain) # last blog post, and first four articles Just like with original query object the limit and offset can be chained to return a new QueryChain. :: chain = chain.limit(5).offset(7) Chain slicing ^^^^^^^^^^^^^ :: chain = QueryChain( [ session.query(BlogPost), session.query(Article), session.query(NewsItem) ] ) chain[3:6] # New QueryChain with offset=3 and limit=6 Count ^^^^^ Let's assume that there are five blog posts, five articles and five news items in the database, and you have the following query chain:: chain = QueryChain( [ session.query(BlogPost), session.query(Article), session.query(NewsItem) ] ) You can then get the total number rows returned by the query chain with :meth:`~QueryChain.count`:: >>> chain.count() 15 )copyc>eZdZdZd dZdZdZdZdZdZ d Z dS) QueryChaina QueryChain can be used as a wrapper for sequence of queries. :param queries: A sequence of SQLAlchemy Query objects :param limit: Similar to normal query limit this parameter can be used for limiting the number of results for the whole query chain. :param offset: Similar to normal query offset this parameter can be used for offsetting the query chain as a whole. .. versionadded: 0.26.0 Nc0||_||_||_dSN)queries_limit_offset)selfrlimitoffsets f/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/sqlalchemy_utils/query_chain.py__init__zQueryChain.__init__{s   c#<Kd}d}|jD]}t|}|jr||j|z }|jr||j|z }d}|D]}|dz }|dz }|V|s||z }||z }dS)Nr)rrr r r r count)r consumedskippedquery query_copy obj_countobjs r__iter__zQueryChain.__iter__s\ % %EeJ{ < DK($:;;| = T\G%;<<I  A Q   %:++---9$! % %rc|d|Srr values rr zQueryChain.limitsFUF|rc||dSrrrs rr zQueryChain.offsetsEFF|rc>td|jDS)zY Return the total number of rows this QueryChain's queries would return. c3>K|]}|VdSr)r).0qs r z#QueryChain.count..s*3317799333333r)sumrr s rrzQueryChain.counts#33dl333333rct|trC||j|j|jn|j|j|jn|jS||dD]}|cSdS)N)rr r r) isinstanceslice __class__rstopr startr )r keyrs r __getitem__zQueryChain.__getitem__s c5 ! ! >> "%("6chhDK$'I$9syyt|"  CE{     rc&dt|zS)Nz)idr&s r__repr__zQueryChain.__repr__s%400r)NN) __name__ __module__ __qualname____doc__rrr r rr.r1rrrrros   %%%*444   11111rrN)r5rrrrrr6sXjjV>1>1>1>1>1>1>1>1>1>1r