o l>j(@sJdZddlmZddlZddlmZmZmZmZm Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZmZddlmZmZddlmZddlmZd d lmZm Z m!Z!d d l"m#Z#m$Z$m%Z%m&Z&m'Z'd d l(m)Z)m*Z*d d l+m,Z,m-Z-m.Z.d dl/m0Z0m1Z1d dl2m3Z3m4Z4e5e6Z7dZ8Gddde&Z9Gddde&Z:GdddeZ;dS)a High-level Apple Notes service built on top of a preconfigured CloudKit client. Public API: - NotesService.recents(limit=50) -> Iterable[NoteSummary] - NotesService.recents_in_folder(folder_id, limit=20) -> Iterable[NoteSummary] - NotesService.iter_all(since=None) -> Iterable[NoteSummary] - NotesService.folders() -> Iterable[NoteFolder] - NotesService.in_folder(folder_id, limit=None) -> Iterable[NoteSummary] - NotesService.get(note_id, with_attachments=False) -> Note - NotesService.sync_cursor() -> str - NotesService.export_note(note_id, output_dir, **config_kwargs) -> str - NotesService.render_note(note_id, **config_kwargs) -> str - NotesService.iter_changes(since=None) -> Iterable[ChangeEvent] - NotesService.raw -> CloudKitNotesClient Prefer the typed service methods for normal use. ``raw`` is an escape hatch for advanced or unsupported CloudKit workflows. ) annotationsN)DictIterableIteratorListOptional) CKErrorItem CKFVStringCKLookupResponseCKQueryFilterBy CKQueryObjectCKQueryResponse CKQuerySortByCKRecord CKReferenceCKTombstoneRecordCKZoneChangesZoneReqCloudKitExtraMode)CKReferenceFieldCKReferenceListField) BaseService) BodyDecoder) NOTES_ZONENOTES_ZONE_NAMENOTES_ZONE_REQ)CloudKitNotesClient NotesApiErrorNotesAuthError NotesErrorNotesRateLimited) AttachmentIdNoteBody) AttachmentNote NoteSummary)NotesDesiredKeyNotesRecordType) ChangeEvent NoteFolder HasSubfolderc@eZdZdZdS) NoteNotFoundz#Raised when a note cannot be found.N__name__ __module__ __qualname____doc__r2r2T/home/thesage/.local/lib/python3.10/site-packages/pyicloud/services/notes/service.pyr,=r,c@r+)NoteLockedErrorz@Raised when a note is password protected and cannot be accessed.Nr-r2r2r2r3r5Cr4r5cseZdZdZdZdZdZdddxfd dZdddyddZdddzddZ ddd{ddZ d|d!d"Z ddd}d$d%Z d&d'd~d,d-Z dd.d/Zdd1d2Zdd3d4Zdddd6d7Zedd9d:Zdd;d<Zdd=ddAdBZeddEdFZeddIdJZddLdMZddNdOZddQdRZddTdUZddVddZd[Zedd^d_ZeddadbZddddeZ eddfdgZ!eddhdiZ"eddkdlZ#ddpdqZ$drdsddvdwZ%Z&S) NotesServicea# Typed, developer-friendly Notes API. The service exposes lightweight listing helpers, full-note retrieval, attachment download helpers, and HTML render/export utilities while hiding most CloudKit details behind ``NoteSummary``, ``Note``, and ``Attachment`` models. zcom.apple.notes productionprivateN)cloudkit_validation_extra service_rootstrparamsDict[str, str]r9CloudKitExtraMode | Nonecshtj|||d|jd|jd|jd|j}ddd|p!i}t||||d|_i|_i|_ dS)N)r:sessionr<z /database/1//T) remapEnumsgetCurrentSyncToken)base_urlr? base_paramsvalidation_extra) super__init__r: _CONTAINER_ENV_SCOPEr_raw_folder_name_cache_attachment_meta_cache)selfr:r?r<r9endpointrD __class__r2r3rGZs" zNotesService.__init__2limitrTintreturnIterable[NoteSummary]c cs|dkrdStjtjtjtjtjtjtjtjtj g }t dt ddt dddd gt d d d gd }d}d}td| ||}|dkrFdS|jj|t||td||d}|jD] }t|trz||} |d7}| V||krztd|dSqZt|dd}|stddSq<)a  Yield the newest note summaries, ordered most-recent first. Args: limit: Maximum number of notes to yield. Yields: ``NoteSummary`` instances with lightweight metadata suitable for feeds, selectors, or navigation UIs. rN SearchIndexesEQUALS indexNameSTRINGrecentstypevalue comparator fieldName fieldValuemodTimeF)rb ascending) recordTypefilterBysortByzFetching recents: limit=%dTqueryzone_id desired_keys results_limit continuationrz)Recents: yielded %d notes (limit reached)continuationMarkerz+Recents: no more continuation marker, done.)r&TITLE_ENCRYPTEDSNIPPET_ENCRYPTEDMODIFICATION_DATEDELETEDFOLDERFIRST_ATTACHMENT_UTI_ENCRYPTEDFIRST_ATTACHMENT_THUMBNAIL&FIRST_ATTACHMENT_THUMBNAIL_ORIENTATION ATTACHMENTSr r r rLOGGERdebugrKrkr _coerce_keysminrecords isinstancer_summary_from_recordgetattr) rNrTrmrkyieldedcont remainingresprecsummaryr2r2r3r\vsj           zNotesService.recents folder_idccs||dkrdSd}td|d}td|||j|dD]}|j|kr;|js;|V|d7}||kr;td|dSqdS) z Yield recent note summaries that belong to ``folder_id``. This is a convenience helper that filters the global recents feed rather than scanning the full Notes history. rNriz1Fetching recents in folder: folder_id=%s limit=%drSrz3Recents in folder: yielded %d notes (limit reached))maxrzr{r\r is_deleted)rNrrTemittedwindownr2r2r3recents_in_folders& zNotesService.recents_in_folder)sincer Optional[str]c cs||r tddStd|rd|nd|jjtttjg| t j t j t j t jt jt jg|dddD]}|jD] }t|trK||Vq>q9dS) ao Yield note summaries from the Notes changes feed. Args: since: Optional sync token from ``sync_cursor()``. When provided, the iterator yields only note records changed since that cursor. Yields: ``NoteSummary`` instances for full exports, indexing jobs, or local cache refreshes. z6Skipping Notes full scan because sync token is currentNzIterating all notes%s since=FzoneIDdesiredRecordTypes desiredKeys syncTokenreversezone_req)_matches_current_sync_cursorrzr{rKchangesrrr'r$r|r&rqrrrsrtruryr~rrr)rNrzonerr2r2r3iter_alls8     zNotesService.iter_allIterable[NoteFolder]c cstjtg}tdtddtddddgd}d }td  |jj |t | |d |d }|j D]9}t |trh|j}||jd}t|jtpJddd }|d urTd nt|} t||| d dV|j||q/t|dd }|sxtdd Sq)z Yield top-level Notes folders as ``NoteFolder`` models. Use this to build folder navigation or resolve folder IDs before calling ``in_folder()``. rXrYrZr[ parentlessr]r`)rfrgNzFetching foldersTrirjTitleEncryptedr2r_)idnamehas_subfolderscountrpz+Folders: no more continuation marker, done.)r&rq_HAS_SUBFOLDER_FIELDr r r rzr{rKrkrr|r~rr recordName_decode_encryptedfields get_valuer get_fieldboolr)rL setdefault) rNrmrkrrrrr has_sub_valuehas_subr2r2r3folderssZ       zNotesService.folders Optional[int]ccsd}td|||jjtttjg|t j t j t j t j t jt jt jt jt jg dddD]>}|jD]8}t|ts;q3||}t|jdpHd}|sP||krQq3||V|d7}|rk||krktd |d Sq3q.d S) z Yield non-deleted notes in ``folder_id``, ordered newest first. Args: folder_id: Folder identifier from ``folders()`` or note metadata. limit: Optional maximum number of notes to yield. rz/Fetching notes in folder: folder_id=%s limit=%sT)rrrrrDeletedFrz1Notes in folder: yielded %d notes (limit reached)N)rzr{rKrrrr'r$r|r&rqrrrsrtrurvrwrxryr~rr_extract_folder_idrrrr)rNrrTrrr rec_folder_iddeletedr2r2r3 in_folder/sR        zNotesService.in_folderF)with_attachmentsnote_idrrr$c Cs>td|||j|tjtjtjtjtjtj dgd}| |}|j r4t d|t d|jp/|d||}z|rK|jrKtd|jt|jntd|jWntyitd |rd|jrdd nd Ynwd }d }|rs|jnd }g} |r|jr|j} |r|j|| d }t|j|j|j|j|j|j|j|j |||d S)a Return a single note with decoded text and optional attachment metadata. Args: note_id: The CloudKit note record identifier. with_attachments: When ``True``, resolve ``Attachment`` metadata and include it on the returned ``Note``. Raises: NoteNotFound: If the note does not exist. NoteLockedError: If the note is passphrase-locked and its content cannot be read. z-Fetching note: note_id=%s with_attachments=%sTextDataEncryptedrmz%Note is locked and cannot be read: %szNote 'z' is locked and cannot be read.z"notes.body.decoded ok id=%s len=%dznotes.body.decoded empty id=%sznotes.body.decoded %sokemptyNattachment_ids) rtitlesnippet modified_atr folder_namer is_lockedtexthtml attachments)rzr{_lookup_note_recordr&rqrrrsrtruryrrwarningr5r_decode_note_bodyrinforlen Exceptionr_resolve_attachments_for_recordr$rrrrr) rNrrtargetr note_bodyrrrrr2r2r3get`sv      zNotesService.getcCstd|jjtdS)z Return the current Notes sync token. Persist this token and pass it back to ``iter_all(since=...)`` or ``iter_changes(since=...)`` on a later run to perform incremental syncs. z#Fetching sync cursor for Notes zone zone_name)rzr{rKcurrent_sync_tokenrrNr2r2r3 sync_cursors zNotesService.sync_cursor output_dirc Ksb||}ddlm}ddlm}|di|}||j|d}|j||d} | s/td|| S) a Export a note to HTML on disk and return the generated file path. Args: note_id: The UUID of the note to export. output_dir: Directory where the HTML file will be written. **config_kwargs: Keyword arguments forwarded to ``ExportConfig``, including ``export_mode``, ``assets_dir``, ``full_page``, ``preview_appearance``, ``pdf_object_height``, and link behavior settings. Returns: The path to the generated HTML file. Notes: By default, this produces archival output: a full HTML page with local asset downloads. r) NoteExporter ExportConfig)config)rzFailed to export note: Nr2)rrendering.exporterrrendering.optionsrrKexportr) rNrr config_kwargsrrrrexporterpathr2r2r3 export_notes   zNotesService.export_notec Ksx||}ddlm}m}ddlm}ddlm}|di|}||} | s(dS||j|| |\} } ||} | | | S)a Render a note to an HTML fragment string. Args: note_id: The UUID of the note to render. **config_kwargs: Keyword arguments forwarded to ``ExportConfig`` to tune preview selection or link behavior for the rendered HTML. Returns: An HTML fragment string. This method does not download assets or write files to disk. r)build_datasourcedecode_and_parse_noter) NoteRendererrNr2) rrrrrrrendering.rendererrrKrender) rNrrrrrrrrnoteds_rendererr2r2r3 render_notes    zNotesService.render_noteIterable[ChangeEvent]ccsr||r tddStd|rd|nd|jjtttjg| t j t j t j t jt jt jg|dddD]}}|jD]w}t|tr`t|jd pMd}|rSd nd }t|||d Vq>t|trt|d d}|r~td t|ddddddddd Vq>t|tr|j|j|jd}t d|jpd|j|jt!d|jpd|dt d|t!ddt"|idq9dS)z Yield ``ChangeEvent`` items from the Notes changes feed. Pass a sync token from ``sync_cursor()`` to process only new changes since a previous run. z8Skipping Notes change scan because sync token is currentNzIterating changes%srrFrrrrupdated)r^rrTrrrrrrrr)serverErrorCodereasonrz;CloudKit error during change enumeration: %s (%s) record=%sUNKNOWNz*CloudKit error during change enumeration: )payloadz*Unexpected record type in changes feed: %rz&Unexpected record type in changes feed record_repr)#rrzr{rKrrrr'r$r|r&rqrrrsrtruryr~rrrrrr(rrrr%rrrrerrorrrepr)rNrrr deleted_flagevt_type record_namedetailsr2r2r3 iter_changess           zNotesService.iter_changesrcCs|jS)z Return the authenticated low-level Notes CloudKit client. This is an escape hatch for advanced or unsupported operations; prefer the typed service methods above for normal use. )rKrr2r2r3rawJszNotesService.rawc CsP|sdSz |jjtd|kWSty'}z td|WYd}~dSd}~ww)z>Return whether an incremental Notes cursor is already current.Frz(Failed to preflight Notes sync token: %sN)rKrrrrzr)rNrexcr2r2r3rVs z)NotesService._matches_current_sync_cursorrrmOptional[Iterable[object]]rcCsZ|jj|g||d}|jD]}t|tr|j|kr|Sqtd|t d|)z2Return one Notes record or raise ``NoteNotFound``.) record_namesrmzNote not found: %szNote not found: ) rKlookupr|r~rrrrzrr,)rNrrmrrr2r2r3ras  z NotesService._lookup_note_recordkeysOptional[List[str]]cCsD|durdSg}|D]}t|tr||jq |t|q |SN)rr&appendr_r;)routkr2r2r3r|ss zNotesService._coerce_keysbOptional[bytes | str]cCs>|durdSt|tr |Sz|ddWStyYdSwNzutf-8replace)rr;decoder)rr2r2r3rs  zNotesService._decode_encryptedrcCsP|jd}t|tr|jr|jjS|jd}t|tr&|jr&|jdjSdS)NFolderFoldersr)rrrrr_rr)rNrfflr2r2r3rs   zNotesService._extract_folder_idc Cs|sdS||jvr|j|Sz6|jj|gdgd}d}|jD]}t|tr5|j|kr5||j d}nq||j|<t d|||WSt t tfy\d|j|<t d|YdSw)Nrrz*Folder name resolved: folder_id=%s name=%sz+Failed to resolve folder name: folder_id=%s)rLrKrr~rrrrrrrzr{rrr r)rNrrrrr2r2r3 _folder_names,       zNotesService._folder_namer%c Cs||jd}||jd}|jd}t|jdp d}||}||}tt|ddt j k}t |j |||||||dS) NrSnippetEncryptedModificationDaterFrfrr) rrrrrrr;rlowerr'PasswordProtectedNoter%r) rNrrrmodifiedrrrrr2r2r3rs,  z!NotesService._summary_from_recordOptional[NoteBody]c Cs|jd}|std|jdSz+t|}|r"t|tr"|WStd|j|r7t |ddr7t t |ddndWdSt yW}zt d|j|WYd}~dSd}~ww) zADecode TextDataEncrypted into a NoteBody (text + attachment IDs).rz*notes.body.missing TextDataEncrypted id=%sNz!notes.body.no_text id=%s bytes=%sbytes0z%notes.body.decode_failed id=%s err=%s) rrrzr{rrr rr"rrrr)rNrrnber2r2r3rs.    zNotesService._decode_note_bodyrrOptional[List[AttachmentId]]List[Attachment]c slg}g}g}|r|D]}t|dd}|r||q |jd}|rFt|drFt|dgp.g} | D]} t| dd} | rE|| || q1t} g} |D]}|| vr]| || |qMtd| || sk|sk|St}g}|rt|n| }|D]}||vr||||qxfdd|D}|r gd }z j j ||d }Wnt y}ztd ||t gd }WYd}~nd}~wwt|d gD]B}t |tsqȈ|}|stdt|ddt|ddrt|jndq|j}||}|D]}|j|<td||qq| p|D]}j|}|r%||vr%||qtdt|t|dd|S)zHydrate attachment metadata for a note. Combines attachment identifiers from CloudKit references and the decoded protobuf body. Missing records are skipped gracefully. identifierN Attachmentsr_rz/notes.attachments.candidates alias=%s lookup=%scsg|] }|jvr|qSr2)rM).0aidrr2r3 sz@NotesService._resolve_attachments_for_record..)Filename AttachmentUTIUTISize ThumbnailFirstAttachmentThumbnail PrimaryAsset PreviewImagesPreviewAppearances FallbackImage FallbackPDF PaperAssetsAttachmentIdentifierattachmentIdentifier Identifierrz-notes.attachments.lookup_failed ids=%s err=%s)r~r~z/notes.attachments.unhandled record=%s fields=%srz)notes.attachments.cached base=%s alias=%sz#Resolved %d attachments for note %s)rrrrhasattrsetaddrzr{rKrrr rr_build_attachment_from_recordlistrr_attachment_aliasesrMrr)rNrrr alias_idslookup_candidatesr"identfldrefsrefrn seen_aliasidscid seen_lookup lookup_idslookup_source_idsmissingrmrerrrec_a attachmentbase_idaliasesaliasattr2rr3rs                     z,NotesService._resolve_attachments_for_recordnames List[str]c Csf|D].}|j|}|durqt|tr'z |ddWSty&Yqwt|tr0|SqdSr )rrrrr rr;rrNrvr2r2r3_coerce_stringVs    zNotesService._coerce_stringrcCs0|g}t|gd}|r||vr|||S)N)r0r1r2)r6rRr)rrrKrr2r2r3r8es  z NotesService._attachment_aliasesOptional[Attachment]c Cst|dd}|s dS||gd}||ddg}||ddg}d}d}d}|p0||dg}|p9||dg}|pA||d}|pI||d }|pQ||d }|pZ||d g}|pd||d d g}|ph|}t|||||||dS)Nr)r$NameFileNamer%r&r'FileSizer*r.r/r+r-r(r))rfilenameutisize download_url preview_url thumbnail_url)rrR _coerce_int_coerce_asset_url_coerce_asset_url_from_listr#) rNrr"rWrXrYrZr[r\r2r2r3r6osD z*NotesService._build_attachment_from_recordcCs`|D]+}|j|}t|tr|St|trt|St|tr-|r-t|SqdSr)rrrrUfloatr;isdigitrPr2r2r3r]s     zNotesService._coerce_intcCsJ|D] }|j|}|s qt|dd}t|dd}t|tr"|SqdSNr_ downloadURL)rrrrr;)rrNrr<valurlr2r2r3r^s    zNotesService._coerce_asset_urlrcCsx|j|}|s dSt|dd}t|ttfr:|D] }d}t|tr(|d}nt|dd}t|tr9|r9|SqdSrb) rrrrr7tupledictrr;)rrr<rdtokenrer2r2r3r_s     z(NotesService._coerce_asset_url_from_listrMr# directorycCs<|jp|jp|j}|stdtd|j||j||S)N*Attachment does not expose a download URL.zDownloading attachment %s to %s) rZr[r\rrzr{rrKdownload_asset_to)rNrMrirer2r2r3_download_attachment_tos z$NotesService._download_attachment_toi chunk_sizernIterator[bytes]ccsJ|jp |jp |j}|stdtd|j||jj||dEdHdS)Nrjz%Streaming attachment %s chunk_size=%drm) rZr[r\rrzr{rrKdownload_asset_stream)rNrMrnrer2r2r3_stream_attachments zNotesService._stream_attachment)r:r;r<r=r9r>)rTrUrVrW)rr;rTrUrVrW)rrrVrW)rVr)rr;rTrrVrW)rr;rrrVr$)rVr;)rr;rr;rVr;)rr;rVr;)rrrVr)rVr)rrrVr)rr;rmrrVr)rrrVr)rr rVr)rrrVr)rrrVr)rrrVr%)rrrVr)rrrrrVr)rrrNrOrVr)rrrr;rVrO)rrrVrS)rrrNrOrVr)rrrr;rVr)rMr#rir;rVr;)rMr#rnrUrVro)'r.r/r0r1rHrIrJrGr\rrrrrrrrrpropertyrrr staticmethodr|rrrrrrrRr8r6r]r^r_rlrq __classcell__r2r2rPr3r6Ls` B (61 P T      |   *    r6)s& 8