aselimov/pubs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added AddEvent

Browse Source
This commit is contained in:
jgrizou 2013-07-05 17:41:48 +02:00
parent ac8ab54aae
commit 9cc893d8a7
3 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
papers/events.py
View File

@ -36,3 +36,8 @@ class RenameEvent(Event):
def __init__(self, old_citekey, new_citekey): def __init__(self, old_citekey, new_citekey):
self.old_citekey = old_citekey self.old_citekey = old_citekey
self.new_citekey = new_citekey self.new_citekey = new_citekey
class AddEvent(Event):
def __init__(self, citekey):
self.citekey = citekey

8
papers/repo.py
View File

@ -5,7 +5,7 @@ import itertools
from . import files from . import files
from .paper import PaperInRepo, NoDocumentFile from .paper import PaperInRepo, NoDocumentFile
from .events import RemoveEvent, RenameEvent from .events import RemoveEvent, RenameEvent, AddEvent
BASE_FILE = 'papers.yaml' BASE_FILE = 'papers.yaml'
BIB_DIR = 'bibdata' BIB_DIR = 'bibdata'
@ -114,9 +114,7 @@ class Repository(object):
self.citekeys.append(p.citekey) self.citekeys.append(p.citekey)
self.save_paper(p) self.save_paper(p)
self.save() self.save()
# TODO change to logging system (17/12/2012) AddEvent(p.citekey).send()
print('Added: {}'.format(p.citekey))
return p
def save_paper(self, paper, old_citekey=None, overwrite=False): def save_paper(self, paper, old_citekey=None, overwrite=False):
if (not paper.citekey in self.citekeys and if (not paper.citekey in self.citekeys and
@ -130,7 +128,7 @@ class Repository(object):
raise CiteKeyCollision('citekey {} already in use'.format( raise CiteKeyCollision('citekey {} already in use'.format(
paper.citekey)) paper.citekey))
RenameEvent(old_citekey, paper.citekey) RenameEvent(old_citekey, paper.citekey).send()
self._remove_paper(old_citekey, remove_doc=False) self._remove_paper(old_citekey, remove_doc=False)
self.add_paper(paper, overwrite=overwrite) self.add_paper(paper, overwrite=overwrite)
self._move_doc(old_citekey, paper) self._move_doc(old_citekey, paper)

2
tests/test_usecase.py
View File

@ -249,7 +249,7 @@ class TestUsecase(unittest.TestCase):
def test_first(self): def test_first(self):
correct = ['Initializing papers in /paper_first.\n', 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', '0: [Page99] L. Page et al. "The PageRank Citation Ranking Bringing Order to the Web" (1999) \n',
'', '',
'', '',