aboutsummaryrefslogtreecommitdiffstats
path: root/fuzzycat/journals.py
blob: bd76b7f265cc7a1553cb67ce1190838732199b0b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# coding: utf-8

"""
Journal name matching. Includes names from issn database and abbreviations.
"""

import shelve

class JournalLookup:
    """
    Lookup allows to lookup journals, using a database of real journal names.

        >>> lookup = JournalLookup()
        >>> lookup["Philosophica"]
        {'1857-9272', '2232-299X', '2232-3007', '2232-3015'}

    """
    def __init__(self, namedb='names'):
        """
        Note that shelve appends "db" to the name automatically.
        """
        self.db = shelve.open(namedb)

    def __getitem__(self, v):
        return self.db[v]

    def get(self, v, cleanup_pipeline=None):
        if not cleanup_pipeline:
            return self.db.get(v)
        return self.db.get(cleanup_pipeline(v))

    def close(self):
        self.db.close()