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.
88 lines
2.6 KiB
88 lines
2.6 KiB
import datetime
|
|
|
|
# Store the next available id for all new notes
|
|
last_id = 0
|
|
|
|
|
|
class Note:
|
|
"""Represent a note in the notebook. Match against a
|
|
string in searches and store tags for each note."""
|
|
|
|
def __init__(self, memo, tags=''):
|
|
"""initialize a note with memo and optional space-separated tags.
|
|
Automatically set the note's creation date and a unique id."""
|
|
self.memo = memo
|
|
self.tags = tags
|
|
self.creation_date = datetime.date.today()
|
|
global last_id
|
|
last_id += 1
|
|
self.id = last_id
|
|
|
|
def match(self, filter):
|
|
"""Determine if this note matches the filter text.
|
|
Return True if it matches, False otherwise.
|
|
Search is case sensitive and matches both text and tags."""
|
|
return filter in self.memo or filter in self.tags
|
|
|
|
|
|
class Notebook:
|
|
"""Represent a collection of notes that can be tagged, modified and
|
|
searched."""
|
|
|
|
def __init__(self):
|
|
"""Initialize a notebook with an empty list."""
|
|
self.notes = []
|
|
|
|
def new_note(self, memo, tags=''):
|
|
"""Create a new note and add it to the list."""
|
|
self.notes.append(Note(memo, tags))
|
|
|
|
def modify_memo(self, note_id, memo):
|
|
"""Find the note with the given id and change its
|
|
memo to the given value."""
|
|
note = self._find_note(note_id)
|
|
if note:
|
|
note.memo = memo
|
|
return True
|
|
return False
|
|
|
|
def modify_tags(self, note_id, tags):
|
|
"""Find the note with the given id and change its tags
|
|
to the given value."""
|
|
note = self._find_note(note_id)
|
|
if note:
|
|
note.tags = tags
|
|
return True
|
|
return False
|
|
|
|
def search(self, filter):
|
|
"""Find all notes that match the given filter string."""
|
|
return [note for note in self.notes if note.match(filter)]
|
|
|
|
def _find_note(self, note_id):
|
|
"""Locate the note with the given id."""
|
|
for note in self.notes:
|
|
if str(note.id) == str(note_id):
|
|
return note
|
|
return None
|
|
|
|
def remove_note(self, id):
|
|
"""Remove note(s) with given id from note list"""
|
|
if self._find_note(id):
|
|
removed = []
|
|
for index, note in enumerate(self.notes):
|
|
if str(note.id) == str(id):
|
|
removed.append(self.notes.pop(index))
|
|
self._set_id()
|
|
return removed
|
|
return False
|
|
|
|
def _set_id(self):
|
|
"""set global last_id to highest id found in notebook"""
|
|
id = 1
|
|
for note in self.notes:
|
|
if note.id > id:
|
|
id = note.id
|
|
global last_id
|
|
last_id = id
|
|
|