Dj-/`dZddlmZddlZddlZddlZddlmZmZm Z m Z m Z m Z ddl mZddlmZmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZdd l m!Z!ej"e#Z$d Z%Gd d e&Z'Gd de'Z(Gdde'Z)Gdde'Z*GddZ+dS)a Low-level CloudKit client for the Notes container. This "escape hatch" is also used internally by NotesService to implement developer-friendly methods. It returns typed Pydantic models from pyicloud.services.notes_models.cloudkit and hides HTTP details. ) annotationsN)DictIterableIteratorListNoReturnOptional)ValidationError) CKFVStringCKLookupResponseCKQueryFilterBy CKQueryObjectCKQueryResponseCKZoneChangesZoneCKZoneChangesZoneReqCKZoneID CKZoneIDReqCloudKitExtraMode)CloudKitApiErrorCloudKitAuthErrorCloudKitContainerClientCloudKitRateLimitedredact_cloudkit_url)NOTES_ZONE_REQ)g$@gN@ceZdZdZdS) NotesErrorzBase Notes transport error.N__name__ __module__ __qualname____doc__Y/home/thesage/.hermes/venv/lib/python3.11/site-packages/pyicloud/services/notes/client.pyrr/s%%%%r$rceZdZdZdS)NotesAuthErrorz!Auth/PCS/cookie issues (401/403).Nrr#r$r%r'r'3s++++r$r'c&eZdZdZdd fd ZxZS) NotesRateLimitedz429 Too Many Requests.Nmessagestr retry_afterOptional[float]cXt|||_dSN)super__init__r,)selfr*r, __class__s r%r1zNotesRateLimited.__init__:s) !!!&r$r/)r*r+r,r-rr r!r"r1 __classcell__r3s@r%r)r)7sH  '''''''''''r$r)c&eZdZdZdd fd ZxZS) NotesApiErrorzCatch-all API error.Nr*r+payloadOptional[object]cXt|||_dSr/)r0r1r9)r2r*r9r3s r%r1zNotesApiError.__init__Bs& !!! r$r/)r*r+r9r:r4r6s@r%r8r8?sHr$r8ceZdZdZdedd>d Zed?dZd@dZdddddAd!Z dd"dBd&Z dCd*Z d+d,dDd1Z dEd3Z dFd5ZedGd9ZedHd=ZdS)ICloudKitNotesClientz Raw CloudKit service for the Notes container. Methods map 1:1 to CloudKit endpoints: - /records/query - /records/lookup - /changes/zone N)validation_extratimeoutbase_urlr+ base_paramsDict[str, object]r>CloudKitExtraMode | Noner?tuple[float, float]c t|||||dd|j|_||_tddS)NlowerT)r>r?bool_param_style redact_urls debug_hookz CloudKitNotesClient initialized.)r_dump_http_debug_client_validation_extraLOGGERinfo)r2r@sessionrAr>r?s r%r1zCloudKitNotesClient.__init__TsX/   -$,     "2 677777r$exc Exceptionreturnrc^|jp|}t|trtt ||t|t r$t t ||j|t|tr$tt ||j |)N)r,)r9) __cause__ isinstancerr'r+rr)r,rr8r9)rPcauses r%_raise_notes_errorz&CloudKitNotesClient._raise_notes_errorjs $ c, - - 6 S** 5 c. / / U"3s88IIIu T c+ , , JC#+>>>E I r$opboolct|tr?t|jtr%|||jpi|jdSdS)NTF)rUrrTr _log_validationr9)r2rXrPs r%_log_cloudkit_validationz,CloudKitNotesClient._log_cloudkit_validationusZ c+ , ,  M?2 2    S[%6B F F F4ur$) desired_keys results_limit continuationqueryrzone_idrr]Optional[List[str]]r^ Optional[int]r_ Optional[str]rctd|j |j|||||}tdt |j|S#tttf$rP}| d|rt d| |Yd}~dSd}~wwxYw)Nz"Executing query for recordType: %s)r`rar]r^r_zQuery returned %d records.z records.queryz!Query response validation failed.) rMrN recordTyperKr`lenrecordsrrrr\errorrW)r2r`rar]r^r_resprPs r%r`zCloudKitNotesClient.querys  8%:JKKK )<%%)+) &D KK4c$,6G6G H H HK "35HI ) ) ),,_cBB B @AAA  # #C ( ( ( ( ( ( ( ( ( )sA A00CACC)r] record_names Iterable[str]r ct|}tdt| |j|t |}tdt|j|S#tttf$rP}| d|rt d| |Yd}~dSd}~wwxYw)Nz Executing lookup for %d records.)rar]zLookup returned %d records.zrecords.lookupz"Lookup response validation failed.)listrMrNrgrKlookuprrhrrrr\rirW)r2rkr]record_names_listrjrPs r%rozCloudKitNotesClient.lookups !.. 6>D%AD  D%i chunk_sizeurlr~intIterator[bytes]c#K |j||Ed{VdS#tttf$r }||Yd}~dSd}~wwxYw)Nr})rKdownload_asset_streamrrrrW)r2rr~rPs r%rz)CloudKitNotesClient.download_asset_streams  )|99%:         !"35HI ) ) )  # #C ( ( ( ( ( ( ( ( ( )s"(AAA directorycddl}ddl}tdt |||j|dd|j}|j ||}t|d5}| |D]}| | dddn #1swxYwYtd||S)Nrz)Downloading asset from %s to directory %sTexist_okz icloud-asset-wbz Finished downloading asset to %s) osuuidrMrNrmakedirsuuid4hexpathjoinopenrwrite) r2rrrrfnamerfchunks r%download_asset_toz%CloudKitNotesClient.download_asset_tos2   7  $ $     I----2 022w||Iu-- $   33C88                    6=== s .CC  C  zone_namec td|tdtdt ddt dd g } |j|t|  }|rtd |Sn2#t$r%}t d|Yd}~nd}~wwxYwtdtt|dggd} |j |}n9#ttt f$r}||Yd}~nd}~wwxYw|jr |jdnd}|r?t'|ddr.tdt)|jStd|t/d)aZ Fetch a current sync token cheaply by issuing a zero-limit query that requests getCurrentSyncToken=true (already in params) and reading the top-level token. Some deployments place the token in CKQueryResponse.syncToken; if absent, we fall back to a one-shot changes call (no records) to harvest a token. z(Fetching current sync token for zone: %sz-Attempting to get sync token via cheap query. SearchIndexesEQUALS indexNameSTRINGrecents)typevalue) comparator fieldName fieldValue)rffilterBy)rv)r`raz+Successfully obtained sync token via query.z;Failed to get sync token via query, falling back. Error: %sNz0Falling back to get sync token via changes call.REGULAR_CUSTOM_ZONE)rvzoneTypeF)rudesiredRecordTypes desiredKeysreversertr syncTokenz2Successfully obtained sync token via changes call.z(Failed to obtain sync token for zone: %szUnable to obtain sync token)rMrNryrr r rKquery_sync_tokenrrQwarningrrr|rrrrWzonesgetattrr+rrir8) r2rqtokenerqenvrPzs r%current_sync_tokenz&CloudKitNotesClient.current_sync_tokens7  > JJJ DEEE &'))xyIII  L11#Y7772E  IJJJ      NNMq    DDDD    GHHH'Y9NOOO!      ),&&&99CC "35HI ) ) )  # #C ( ( ( ( ( ( ( ( )I /CIaLL4  $K.. $ KKL M M Mq{## # ?KKK9:::s1(AB11 C ;CC  D<<E2E--E2r9rNonec tjdsdStdd}|dd}tjdd} tj|d n#t$rYdSwxYwtj||d |d } t|d d 5}tj ||d|dddddn #1swxYwYn#t$rYnwxYwtj||d |d} t|dd} t|di} d} |j } n#t$rd} YnwxYwt| d d 5}|d| d|dt| d| rkt!tjdd} t#| | kr!|| d| dzn|| ddddS#1swxYwYdS#t$rYdSwxYw)NPYICLOUD_NOTES_DEBUGtime %Y%m%d-%H%M%S/. workspace notes_debugTr_z_http_request.jsonwutf-8encoding)rr9F ensure_asciiindentz_http_response.txt status_codeheaderszstatus=z url=z headers=z PYICLOUD_DEBUG_MAX_BYTES524288z [truncated] )rgetenv __import__strftimereplacerrrrQrjsondumprtextrdictrrg)rXrr9rjtssafe_opout_dirreq_pathrres_pathstatusr body_text max_bytess r%rJz$CloudKitNotesClient._dump_http_debug(s7y/00  F    ( ( 9 9**S#&&',,{M::  K$ / / / / /    FF 7<<B)L)L)L)L)LMM hg666 ! G44aeTU                   D 7<<B)L)L)L)L)LMM T=$77FdIr22GI ! I  ! ! !  !hg666 +!R&RRRRtG}}RRRSSS+ #BI.H($S$S T TI9~~ 11 *9* 58I IJJJJ *** + + + + + + + + + + + + + + + + + +    DD s0B BB?DC9- D9C==DC=D DD<$I!E)(I) E85I7E88I BH4' I4H88I;H8<I IIdataerrr c*tjdsdStdd}tjdd} tj|dn#t$rYdSwxYwtj||d|d } t|d d 5}tj || |d |ddddddS#1swxYwYdS#t$rYdSwxYw)NrrrrrTrrz_validation.jsonrrr)rXerrorsrFrr) rrrrrrrrQrrrr)rXrrrrrrs r%r[z#CloudKitNotesClient._log_validationOsy/00  F    ( ( 9 9',,{M::  K$ / / / / /    FF w||G%A%AR%A%A%ABB dC'222 a tDD!&                       DD sHA11 A?>A?)D;/C7* D7C;;D>C;?D DD)r@r+rArBr>rCr?rD)rPrQrRr)rXr+rPrQrRrY) r`rrarr]rbr^rcr_rdrRr)rkrlr]rbrRr )rqrrRrr)rr+r~rrRr)rr+rr+rRr+)rr+rRr+)rXr+rr+r9rrRr)rXr+rrrr rRr)rr r!r"DEFAULT_TIMEOUTr1 staticmethodrWr\r`ror|rrrrJr[r#r$r%r=r=Jse6:'6888888,\-1'+&*))))))>-1 )))))).))))> ) ) ) ) ) )(6;6;6;6;t$$$\$L\r$r=),r" __future__rrloggingrtypingrrrrrr pydanticr pyicloud.common.cloudkitr r r rrrrrrrpyicloud.common.cloudkit.clientrrrrr _constantsr getLoggerrrMrrQrr'r)r8r=r#r$r%rsM#"""""  EEEEEEEEEEEEEEEE$$$$$$                        '&&&&&  8 $ $ &&&&&&&&,,,,,Z,,,'''''z'''JYYYYYYYYYYr$