xml2csv/xml2csv_nusl.py at master · petulica/xml2csv · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import os
from xml.etree import ElementTree as et
from collections import Counter

# base_path = os.path.dirname(os.path.realpath(__file__))
#
# xml_file = os.path.join(base_path, "testovaci_nusl",
#                         "https___invenio.nusl.cz_oai2d_verb=GetRecord&metadataPrefix=marcxml&identifier=oai_invenio.nusl.cz_395994.xml")
#
# tree = et.parse(xml_file)
#
# root = tree.getroot()


def identifier(root):
    fields = root.findall(".//{http://www.loc.gov/MARC21/slim}controlfield[@tag='001']")
    for field in fields:
        return field.text


def doc_type(root):
    fields = root.findall(".//{http://www.loc.gov/MARC21/slim}datafield[@tag='980']")
    for field in fields:
        for subfield in field:
            return subfield.text


def langs(root):
    fields = root.findall(".//{http://www.loc.gov/MARC21/slim}datafield[@tag='041']")
    for field in fields:
        result = []
        for subfield in field:
            result.append(subfield.text)
        # if subfield.attrib["code"] == "a" or "b":
        return result


def year(root):
    fields = root.findall(".//{http://www.loc.gov/MARC21/slim}datafield[@tag='046']")
    for field in fields:
        for subfield in field:
            if subfield.attrib["code"] == "k":
                return subfield.text


def projects(root):
    fields = root.findall(".//{http://www.loc.gov/MARC21/slim}datafield[@tag='999']")
    result = []
    for field in fields:
        if field.attrib["ind2"] == "1":
            for subfield in field:
                if subfield.attrib["code"] == "a":
                    result.append(subfield.text)
    return result

if __name__ == "__main__":
    base_path = os.path.dirname(os.path.realpath(r"E:\nusl_data_final\invenio.nusl.cz\2.marcxml"))
    xml_path = "E:/nusl_data_final/invenio.nusl.cz/10.marcxml.xml" #os.path.join(base_path, "1.xml")
    # xml_path = r"E:\nusl_data_final\invenio.nusl.cz\1.marcxml"
    tree = et.parse(xml_path)
    root = tree.getroot()
    print(identifier(root), doc_type(root), langs(root), year(root), projects(root), len(projects(root)))