j dZddlmZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZede Zed e Zd d gZGd d e eefZGdd e eefZy)z"Pagination wrapper for Honcho SDK.) annotations) AsyncIterator AwaitableCallableIterator)AnyGeneric) BaseModel)TypeVarT)boundU)defaultSyncPage AsyncPageceZdZdZ d ddZddZddZddZeddZ eddZ edd Z edd Z edd Z dd Zdd Zy)rz Paginated result wrapper that transforms objects from type T to type U. Provides iteration and transformation capabilities for paginated API responses. Nc||_||_||_||_|j dg}|Dcgc]}|j |c}|_ycc}w)a Initialize the page. Args: data: Raw paginated response data with items, page, size, total, pages item_type: Type to parse items as transform_func: Optional function to transform objects from type T to type U. If None, objects are passed through unchanged. fetch_next: Optional callback to fetch the next page. Takes page number. itemsN_data _item_type_transform_func _fetch_nextgetmodel_validate _raw_itemsselfdata item_typetransform_func fetch_next raw_itemsitems /home/cube/projects/richard/traning coach/.omo/evidence/nutricoach-v150-combined/st_01a0560c-r63-installed-wheel-first-claim-qa/venv/lib/python3.12/site-packages/honcho/pagination.py__init__zSyncPage.__init__s_"&* #,8FEOHHWb) 7@$ 7@tI $ $T *y$ $ Ac#K|}|I|jD]&}|j|j|#|(|j}|Hyyw)aA Iterate over all transformed items across all pages. Warning: This iterator automatically fetches ALL subsequent pages as you iterate. For large datasets, this may result in many API calls. If you only need the current page, use the `items` property instead. Nrr get_next_pagerpager$s r%__iter__zSyncPage.__iter__4s['+''3..t44J ( %%'D s A AAc^|j|}|j|j|S|Sz4Get a transformed item by index on the current page.rrrindexr$s r% __getitem__zSyncPage.__getitem__F2u%    +''- - c,t|jSz,Get the number of items on the current page.lenrrs r%__len__zSyncPage.__len__M4??##r5c|j)|jDcgc]}|j|c}St|jScc}wz.Get all transformed items on the current page.rrlistrr$s r%rzSyncPage.itemsQI    +;???K?4D((.?K KDOO$$LA c8|jjdSz/Get the total number of items across all pages.totalrrr:s r%rFzSyncPage.totalXzz~~g&&r5c8|jjdSzGet the current page number.r,rGr:s r%r,z SyncPage.page]zz~~f%%r5c8|jjdSzGet the page size.sizerGr:s r%rNz SyncPage.sizebrKr5c8|jjdSzGet the total number of pages.pagesrGr:s r%rQzSyncPage.pagesgrHr5cF|j}|j}||y||kSzCheck if there's a next page.Fr,rQr current_page total_pagess r% has_next_pagezSyncPage.has_next_pagel-yy jj  ;#6k))r5c|jsy|jy|j}|y|j|dzSzx Fetch the next page of results. Returns None if there are no more pages or no fetch callback. NrXrr,rrVs r%r*zSyncPage.get_next_pagetsJ !!#    #yy   q 011r5NN) rdict[str, Any]r type[T]r!Callable[[T], U] | Noner"z(Callable[[int], 'SyncPage[T, U]'] | NonereturnNone)rczIterator[U] | Iterator[T]r2intrczU | Trcrfrczlist[U] | list[T]rcz int | Nonercbool)rcz'SyncPage[T, U] | None')__name__ __module__ __qualname____doc__r&r-r3r;propertyrrFr,rNrQrXr*r5r%rrs37?C    0  =    8($$%% ''&&&&''*2r5ceZdZdZ d ddZddZddZddZeddZ eddZ edd Z edd Z edd Z dd Zdd Zy)rz Async paginated result wrapper that transforms objects from type T to type U. Provides async iteration and transformation capabilities for paginated API responses. Nc||_||_||_||_|j dg}|Dcgc]}|j |c}|_ycc}w)a Initialize the async page. Args: data: Raw paginated response data with items, page, size, total, pages item_type: Type to parse items as transform_func: Optional function to transform objects from type T to type U. If None, objects are passed through unchanged. fetch_next: Optional async callback to fetch the next page. Takes page number. rNrrs r%r&zAsyncPage.__init__sb"&* #,8F   HHWb) 7@$ 7@tI $ $T *y$ $ r'cK|}|S|jD](}|j|j|$|*|jd{}|Ryy7 w)aG Async iterate over all transformed items across all pages. Warning: This iterator automatically fetches ALL subsequent pages as you iterate. For large datasets, this may result in many API calls. If you only need the current page, use the `items` property instead. Nr)r+s r% __aiter__zAsyncPage.__aiter__sc(,''3..t44J ( ++--D  .sAAAAAc^|j|}|j|j|S|Sr/r0r1s r%r3zAsyncPage.__getitem__r4r5c,t|jSr7r8r:s r%r;zAsyncPage.__len__r<r5c|j)|jDcgc]}|j|c}St|jScc}wr>r?rAs r%rzAsyncPage.itemsrBrCc8|jjdSrErGr:s r%rFzAsyncPage.totalrHr5c8|jjdSrJrGr:s r%r,zAsyncPage.pagerKr5c8|jjdSrMrGr:s r%rNzAsyncPage.sizerKr5c8|jjdSrPrGr:s r%rQzAsyncPage.pagesrHr5cF|j}|j}||y||kSrSrTrUs r%rXzAsyncPage.has_next_pagerYr5cK|jsy|jy|j}|y|j|dzd{S7wr[r]r^s r%r*zAsyncPage.get_next_pagesT !!#    #yy  %%lQ&67777sAAA Ar_) rr`r rar!rbr"z4Callable[[int], Awaitable['AsyncPage[T, U]']] | Nonercrd)rcz#AsyncIterator[U] | AsyncIterator[T]rergrhrirj)rcz'AsyncPage[T, U] | None')rlrmrnror&rur3r;rprrFr,rNrQrXr*rqr5r%rrs37KO    0  I    <.$$%% ''&&&&''*8r5N)ro __future__rcollections.abcrrrrtypingrr pydanticr typing_extensionsr r r__all__rrrqr5r%rsi("HH% Cy! C { #r2wq!t}r2jt81 t8r5