pubs/papers/repo.py

import files
import color

from paper import Paper

alphabet = 'abcdefghijklmopqrstuvwxyz'


class Repository(object):

    def __init__(self):
        self.paperdir = files.find_papersdir()
        self.papers_config = files.load_papers()
        self.citekeys = self.papers_config['citekeys']
        self.numbers  = self.papers_config['numbers']

    # loading existing papers

    def paper_from_number(self, number, fatal = True):
        try:
            citekey = self.numbers[int(number)]
            paper = self.paper_from_citekey(citekey)
            paper.number = int(number)
            return paper
        except KeyError:
            if fatal:
                print('{}error{}: no paper with number {}{}{}'.format(
                    color.error, color.normal, color.citekey, citekey, color.end))
                exit(-1)
            raise IOError, 'file not found'
            
    def paper_from_citekey(self, citekey, fatal = True):
        """Load a paper by its citekey from disk, if necessary."""
        try:
            name = self.citekeys[citekey]
            paper = Paper.from_disc(name, citekey = citekey)
            paper.citekey = citekey
            return paper
        except KeyError:
            if fatal:
                print('{}error{}: no paper with citekey {}{}{}'.format(
                       color.error, color.normal, color.citekey, citekey, color.end))
                exit(-1)
            raise IOError, 'file not found'

    def paper_from_any(self, key, fatal = True):
        try:
            return self.paper_from_citekey(key, fatal = False)
        except IOError:
            try:
                return self.paper_from_number(key, fatal = False)
            except IOError:
                if fatal:
                    print('{}error{}: paper with citekey or number {}{}{} not found{}'.format(
                        color.error, color.normal, color.citekey, key, color.normal, color.end))
                    exit(-1)
                raise IOError, 'file not found'

    # creating new papers
    
    def add_paper_from_paths(self, pdfpath, bibpath):
        
        p = Paper.from_bibpdffiles(pdfpath, bibpath)        
        self.add_paper(p)

    def add_paper(self, p):
        # updating papersconfig
        p.citekey = self.create_citekey(p.bib_data)
        p.number  = self.create_number()

        self.papers_config['citekeys'][p.citekey] = p.name
        self.papers_config['numbers'][p.number] = p.citekey

        self.citekeys[p.citekey] = p.name
        self.numbers[p.number] = p.citekey
               
        # writing all to disk
        files.save_papers(self.papers_config)
        p.save_to_disc()
        print "Added: %s" % p.citekey
        return p

    def add_papers(self, bibpath):
        bib_data = Paper.import_bibdata(bibpath)
        for k in bib_data.entries:
            sub_bib = type(bib_data)(preamble=bib_data._preamble)
            sub_bib.add_entry(k, bib_data.entries[k])
            name, meta = Paper.create_meta(sub_bib, pdfpath=None)
            p = Paper(name, bib_data = sub_bib, metadata = meta)
            self.add_paper(p)
        
    def create_citekey(self, bib_data, allowed = tuple()):
        """Create a cite key unique to a given bib_data.
        
        Raises:
            KeyError if no author is defined.
        """
        article = bib_data.entries[list(bib_data.entries.keys())[0]]
        author_key = 'author'
        if not 'author' in article.persons:
            author_key = 'editor'
        try:
            first_author = article.persons[author_key][0]
        except KeyError:
            raise(ValueError,
                    'No author or editor defined: cannot generate a citekey.')
        try:
            year = article.fields['year']
        except KeyError:
            year = ''
        prefix = '{}{}'.format(first_author.last()[0][:6], year)

        letter = 0

        citekey = prefix
        while citekey in self.citekeys and citekey not in allowed:
            citekey = prefix + alphabet[letter]
            letter += 1

        return citekey
        
    def create_number(self):
        count = int(self.papers_config['count'])
        self.papers_config['count'] = count + 1
        return count