diff --git a/papers/events.py b/papers/events.py index e904422..b83cd0c 100644 --- a/papers/events.py +++ b/papers/events.py @@ -36,3 +36,8 @@ class RenameEvent(Event): def __init__(self, old_citekey, new_citekey): self.old_citekey = old_citekey self.new_citekey = new_citekey + + +class AddEvent(Event): + def __init__(self, citekey): + self.citekey = citekey diff --git a/papers/repo.py b/papers/repo.py index 7641eeb..9856625 100644 --- a/papers/repo.py +++ b/papers/repo.py @@ -5,7 +5,7 @@ import itertools from . import files from .paper import PaperInRepo, NoDocumentFile -from .events import RemoveEvent, RenameEvent +from .events import RemoveEvent, RenameEvent, AddEvent BASE_FILE = 'papers.yaml' BIB_DIR = 'bibdata' @@ -114,9 +114,7 @@ class Repository(object): self.citekeys.append(p.citekey) self.save_paper(p) self.save() - # TODO change to logging system (17/12/2012) - print('Added: {}'.format(p.citekey)) - return p + AddEvent(p.citekey).send() def save_paper(self, paper, old_citekey=None, overwrite=False): if (not paper.citekey in self.citekeys and @@ -130,7 +128,7 @@ class Repository(object): raise CiteKeyCollision('citekey {} already in use'.format( paper.citekey)) - RenameEvent(old_citekey, paper.citekey) + RenameEvent(old_citekey, paper.citekey).send() self._remove_paper(old_citekey, remove_doc=False) self.add_paper(paper, overwrite=overwrite) self._move_doc(old_citekey, paper) diff --git a/tests/test_usecase.py b/tests/test_usecase.py index 2a348ee..828cb00 100644 --- a/tests/test_usecase.py +++ b/tests/test_usecase.py @@ -249,7 +249,7 @@ class TestUsecase(unittest.TestCase): def test_first(self): correct = ['Initializing papers in /paper_first.\n', - 'Added: Page99\n', + '', '0: [Page99] L. Page et al. "The PageRank Citation Ranking Bringing Order to the Web" (1999) \n', '', '',