o l>j-/@s dZddlmZddlZddlZddlZddlmZmZm Z m Z m Z m Z ddl mZddlmZmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZdd l m!Z!e"e#Z$d Z%Gd d d e&Z'Gd dde'Z(Gddde'Z)Gddde'Z*GdddZ+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@c@eZdZdZdS) NotesErrorzBase Notes transport error.N__name__ __module__ __qualname____doc__r"r"S/home/thesage/.local/lib/python3.10/site-packages/pyicloud/services/notes/client.pyr/rc@r)NotesAuthErrorz!Auth/PCS/cookie issues (401/403).Nrr"r"r"r#r%3r$r%c$eZdZdZd d fdd ZZS) NotesRateLimitedz429 Too Many Requests.Nmessagestr retry_afterOptional[float]ct|||_dSN)super__init__r*)selfr(r* __class__r"r#r/:  zNotesRateLimited.__init__r-)r(r)r*r+rrr r!r/ __classcell__r"r"r1r#r'7r'cr&) NotesApiErrorzCatch-all API error.Nr(r)payloadOptional[object]cr,r-)r.r/r8)r0r(r8r1r"r#r/Br3zNotesApiError.__init__r-)r(r)r8r9r4r"r"r1r#r7?r6r7c@seZdZdZdeddId d ZedJddZdKddZdddddLd$d%Z dd&dMd*d+Z dNd/d0Z d1d2dOd7d8Z dPd:d;Z dQd=d>ZedRdBdCZedSdGdHZdS)TCloudKitNotesClientz 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 Cs0t|||||dd|jd|_||_tddS)NlowerT)r;r<bool_param_style redact_urls debug_hookz CloudKitNotesClient initialized.)r_dump_http_debug_client_validation_extraLOGGERinfo)r0r=sessionr>r;r<r"r"r#r/Ts  zCloudKitNotesClient.__init__exc ExceptionreturnrcCs`|jp|}t|trtt||t|tr tt||jd|t|tr/t t||j d|)N)r*)r8) __cause__ isinstancerr%r)rr'r*rr7r8)rLcauser"r"r#_raise_notes_errorjs    z&CloudKitNotesClient._raise_notes_erroropboolcCs4t|trt|jtr|||jpi|jdSdS)NTF)rPrrOr _log_validationr8)r0rSrLr"r"r#_log_cloudkit_validationus z,CloudKitNotesClient._log_cloudkit_validation) desired_keys results_limit continuationqueryr zone_idrrWOptional[List[str]]rX Optional[int]rY Optional[str]rc Cstd|jz|jj|||||d}tdt|j|WSttt fyD}z| d|r4t d| |WYd}~dSd}~ww)Nz"Executing query for recordType: %s)rZr[rWrXrYzQuery returned %d records.z records.queryz!Query response validation failed.) rIrJ recordTyperGrZlenrecordsrrrrVerrorrR)r0rZr[rWrXrYresprLr"r"r#rZs"   zCloudKitNotesClient.query)rW record_names Iterable[str]r c Cst|}tdt|z|jj|t|d}tdt|j|WStt t fyG}z| d|r7t d| |WYd}~dSd}~ww)Nz Executing lookup for %d records.)r[rWzLookup returned %d records.zrecords.lookupz"Lookup response validation failed.)listrIrJr`rGlookuprrarrrrVrbrR)r0rdrWrecord_names_listrcrLr"r"r#rgs   zCloudKitNotesClient.lookupzone_reqrIterator[CKZoneChangesZone]c cstd|jjd}z,|jj|dD]!}td|t|j|V|jr+t dntd|d7}qWdSt t t fy]}z| d|rMtd||WYd}~dSd}~ww) Nz#Start fetching changes for zone: %srriz$Changes page %d returned %d records.z+More changes to come, advancing sync token.zAll changes fetched.z changes.zonez#Changes response validation failed.)rIrJzoneIDzoneNamerG iter_changesr`ra moreComingdebugrrrrVrbrR)r0ripage_numzonerLr"r"r#changess,     zCloudKitNotesClient.changesi chunk_sizeurlruintIterator[bytes]c csVz|jj||dEdHWdStttfy*}z ||WYd}~dSd}~ww)Nrt)rGdownload_asset_streamrrrrR)r0rvrurLr"r"r#rysz)CloudKitNotesClient.download_asset_stream directoryc Csddl}ddl}tdt|||j|ddd|j}|j ||}t |d}| |D]}| |q2Wdn1sDwYtd||S)Nrz)Downloading asset from %s to directory %sTexist_okz icloud-asset-wbz Finished downloading asset to %s) osuuidrIrJrmakedirsuuid4hexpathjoinopenrywrite) r0rvrzr~rfnamerfchunkr"r"r#download_asset_tos"   z%CloudKitNotesClient.download_asset_to zone_namec CsLtd|tdtdtddtdddd gd }z|jj|t|d d }|r1td |WSWnt yJ}z t d|WYd}~nd}~wwtdt t |ddggdd}z |jj |d}Wntttfy}z ||WYd}~nd}~ww|jr|jdnd}|rt|ddrtdt|jStd|td)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)r_filterBy)rm)rZr[z+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)rmzoneTypeF)rldesiredRecordTypes desiredKeysreverserkr syncTokenz2Successfully obtained sync token via changes call.z(Failed to obtain sync token for zone: %szUnable to obtain sync token)rIrJrpr r r rGquery_sync_tokenrrMwarningrrrsrrrrRzonesgetattrr)rrbr7) r0rqtokenerienvrLzr"r"r#current_sync_tokens\           z&CloudKitNotesClient.current_sync_tokenr8rNonec CstdsdStdd}|dd}tjdd}z tj|dd Wn ty.YdSwtj||d |d }z%t |d d d}t j ||d|dddWdn1s[wYWn tyjYnwtj||d |d} zt |dd} t |di} d} z|j } Wn tyd} Ynwt | d d dN}|d| d|dt| d| rttdd} t| | kr|| d| dn|| WdWdSWdWdSWdWdS1swYWdStyYdSw)NPYICLOUD_NOTES_DEBUGtime %Y%m%d-%H%M%S/. workspace notes_debugTr{_z_http_request.jsonwutf-8encoding)rvr8F ensure_asciiindentz_http_response.txt status_codeheaderszstatus=z url=z headers=z PYICLOUD_DEBUG_MAX_BYTES524288z [truncated] )r~getenv __import__strftimereplacerrrrMrjsondumprtextrdictrwr`)rSrvr8rctssafe_opout_dirreq_pathrres_pathstatusr body_text max_bytesr"r"r#rF(s\        "  &z$CloudKitNotesClient._dump_http_debugdataerrr cCstdsdStdd}tjdd}z tj|ddWn ty(YdSwtj||d|d }z+t|d d d }t j || |d |dddWdWdS1sZwYWdStykYdSw)NrrrrrTr{rz_validation.jsonrrr)rSerrorsrFrr) r~rrrrrrrMrrrr)rSrrrrrrr"r"r#rUOs,  & z#CloudKitNotesClient._log_validation)r=r)r>r?r;r@r<rA)rLrMrNr)rSr)rLrMrNrT) rZr r[rrWr\rXr]rYr^rNr)rdrerWr\rNr )rirrNrj)rvr)rurwrNrx)rvr)rzr)rNr))rr)rNr))rSr)rvr)r8rrNr)rSr)rrrr rNr)rrr r!DEFAULT_TIMEOUTr/ staticmethodrRrVrZrgrsryrrrFrUr"r"r"r#r:Js.     : &r:),r! __future__rrloggingr~typingrrrrrrpydanticr pyicloud.common.cloudkitr r r r rrrrrrpyicloud.common.cloudkit.clientrrrrr _constantsr getLoggerrrIrrMrr%r'r7r:r"r"r"r#s"   0