blob: d16d8df2d48607179e2d2a5087df5cba8ce0322b (
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
|
from typing import Iterable, Optional
import csv
from chocula.util import clean_str
from chocula.common import DirectoryLoader
from chocula.database import DirectoryInfo, HomepageUrl
class WikidataLoader(DirectoryLoader):
"""
CSV Columns:
"""
source_slug = "wikidata"
def open_file(self) -> Iterable:
return csv.DictReader(open(self.config.wikidata.filepath), delimiter='\t')
def parse_record(self, row) -> Optional[DirectoryInfo]:
if not (row.get('issn') and row.get('title')):
return None
wikidata_qid = row['item'].strip().split('/')[-1]
publisher = row['publisher_name']
if (publisher.startswith('Q') and publisher[1].isdigit()) or publisher.startswith('t1') or not publisher:
publisher = None
info =DirectoryInfo(
directory_slug=self.source_slug,
raw_issn=row['issn'],
custom_id=wikidata_qid,
name=clean_str(row['title']),
publisher=clean_str(publisher),
)
if row.get('start_year'):
info.extra['start_year'] = row['start_year']
url = HomepageUrl.from_url(row.get('websiteurl'))
if url:
info.homepage_urls.append(url)
return info
|