aboutsummaryrefslogtreecommitdiffstats
path: root/chocula/directories/ezb.py
blob: 15730480e9cc3667eb1dc4b7c773f2c0088bb607 (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
34
35
36
37
38
39
40
41
42
43
44
45
46
47

from typing import Iterable, Optional
import json

from chocula.util import clean_str
from chocula.common import DirectoryLoader
from chocula.database import DirectoryInfo, HomepageUrl


class EzbLoader(DirectoryLoader):
    """
    CSV Columns:

    """

    source_slug = "ezb"

    def open_file(self) -> Iterable:
        return open(self.config.ezb.filepath, 'r')

    def parse_record(self, row) -> Optional[DirectoryInfo]:

        if not row:
            return None
        row = json.loads(row)

        info = DirectoryInfo(
            directory_slug=self.source_slug,
            issne=row.get('issne'),
            issnp=row.get('issnp'),
            custom_id=row['ezb_id'],
            name=clean_str(row['title']),
            publisher=clean_str(row.get('publisher')),
        )

        info.extra = dict()
        for k in ('ezb_color', 'subjects', 'keywords', 'zdb_id',
                    'first_volume', 'first_issue', 'first_year',
                    'appearance', 'costs'):
            if row.get(k):
                info.extra[k] = row[k]

        url = HomepageUrl.from_url(row.get('url'))
        if url:
            info.homepage_urls.append(url)

        return info