You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
3.1 KiB
96 lines
3.1 KiB
from api import StandardNotesAPI |
|
from uuid import uuid1 |
|
|
|
class ItemManager: |
|
items = {} |
|
|
|
def mapResponseItemsToLocalItems(self, response_items, metadata_only=False): |
|
DATA_KEYS = ['content', 'enc_item_key', 'auth_hash'] |
|
|
|
for response_item in response_items: |
|
uuid = response_item['uuid'] |
|
|
|
if response_item['deleted']: |
|
if uuid in self.items: |
|
del self.items[uuid] |
|
continue |
|
|
|
response_item['dirty'] = False |
|
|
|
if uuid not in self.items: |
|
self.items[uuid] = {} |
|
|
|
for key, value in response_item.items(): |
|
if metadata_only and key in DATA_KEYS: |
|
continue |
|
self.items[uuid][key] = value |
|
|
|
def syncItems(self): |
|
dirty_items = [item for uuid, item in self.items.items() if item['dirty']] |
|
|
|
# remove keys (note: this removes them from self.items as well) |
|
for item in dirty_items: |
|
item.pop('dirty', None) |
|
item.pop('updated_at', None) |
|
|
|
response = self.sn_api.sync(dirty_items) |
|
self.mapResponseItemsToLocalItems(response['response_items']) |
|
self.mapResponseItemsToLocalItems(response['saved_items'], metadata_only=True) |
|
|
|
def getNotes(self): |
|
notes = {} |
|
sorted_items = sorted(self.items.items(), key=lambda x: x[1]['created_at']) |
|
|
|
for uuid, item in sorted_items: |
|
if item['content_type'] == 'Note': |
|
note = item['content'] |
|
text = note['text'] + '\n' |
|
count = 0 # used to remove title duplicates |
|
|
|
while True: |
|
title = note['title'] + ('' if not count else ' ' + str(count + 1)) |
|
if title in notes: |
|
count += 1 |
|
else: |
|
break |
|
|
|
notes[title] = dict(text=text, |
|
created=item['created_at'], |
|
modified=item.get('updated_at', item['created_at']), |
|
uuid=item['uuid']) |
|
return notes |
|
|
|
def touchNote(self, uuid): |
|
item = self.items[uuid] |
|
item['dirty'] = True |
|
|
|
def writeNote(self, uuid, text): |
|
item = self.items[uuid] |
|
item['content']['text'] = text.strip() |
|
item['dirty'] = True |
|
|
|
def createNote(self, name, time): |
|
uuid = str(uuid1()) |
|
content = dict(title=name, text='', references=[]) |
|
self.items[uuid] = dict(content_type='Note', |
|
dirty=True, |
|
auth_hash=None, |
|
uuid=uuid, |
|
created_at=time, |
|
updated_at=time, |
|
enc_item_key='', |
|
content=content) |
|
|
|
def renameNote(self, uuid, new_note_name): |
|
item = self.items[uuid] |
|
item['content']['title'] = new_note_name |
|
item['dirty'] = True |
|
|
|
def deleteNote(self, uuid): |
|
item = self.items[uuid] |
|
item['deleted'] = True |
|
item['dirty'] = True |
|
|
|
def __init__(self, sn_api): |
|
self.sn_api = sn_api |
|
self.syncItems()
|
|
|