dspace-import/bundleservice.py at master · khemnavet/dspace-import · 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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
from config import ImporterConfig
from dspacerequest import DspaceBundleRequest
from dataobjects import Bundle, Bitstream, Item, AuthData
from requests import HTTPError

class BundleException(Exception):
    def __init__(self, msg):
        self.msg = msg

    def __str__(self):
        return self.msg

class BundleService:
    _self = None

    def __new__(cls, *args, **kwargs):
        if cls._self is None:
            cls._self = super().__new__(cls)
        return cls._self

    def __init__(self, config: ImporterConfig, auth_data: AuthData) -> None:
        self._bundle_request = DspaceBundleRequest(config.dspace_rest_url(), auth_data)

    def bundle_has_primary_bitstream(self, bundle: Bundle):
        return self._bundle_request.has_primary_bitstream(bundle.uuid)

    def remove_primary_bitstream(self, bundle: Bundle):
        try:
            resp = self._bundle_request.delete_primary_bitstream_flag(bundle.uuid)
        except HTTPError as err:
            print(f"Exception removing primary bitstream flag for bundle {bundle.uuid}. Error code {err.response.status_code}, reason {err.response.reason}")
            raise BundleException(f"Error removing primary bitstream flag for bundle {bundle.name}. Error code {err.response.status_code}, reason {err.response.reason}")
        except Exception as err1:
            print(f"Exception removing primary bitstream flag for bundle {bundle.uuid}. Error: {str(err1)}")
            raise BundleException(f"Error removing primary bitstream flag for bundle {bundle.name}. Error: {str(err1)}")

    def _populate_bitstream(self, bitstream):
        return Bitstream(bitstream_id=bitstream["id"], uuid=bitstream["uuid"], name=bitstream["name"])

    def bundle_bitstreams(self, bundle: Bundle):
        try:
            result = []
            curr_page = 0
            total_pages = 1
            while curr_page < total_pages:
                bitstream_json = self._bundle_request.get_bitstreams(bundle.uuid, curr_page)
                total_pages = bitstream_json["page"]["totalPages"]
                for bitstream in bitstream_json["_embedded"]["bitstreams"]:
                    result.append(self._populate_bitstream(bitstream))
                curr_page = curr_page + 1

            return result
        except HTTPError as err:
            print(f"Exception getting bitstreams for bundle {bundle.uuid}. Error code {err.response.status_code}, reason {err.response.reason}")
            raise BundleException(f"Exception getting bitstreams for bundle {bundle.name}. Error code {err.response.status_code}, reason {err.response.reason}")
        except Exception as err1:
            print(f"Exception getting bitstreams for bundle {bundle.uuid}. Error: {str(err1)}")
            raise BundleException(f"Error getting bitstreams for bundle {bundle.uuid}. Error: {str(err1)}")

    def create_bundle(self, bundle: Bundle, item: Item) -> Bundle:
        try:
            new_bundle = self._bundle_request.new_bundle(bundle.to_json_str(), item.uuid)
            bundle.uuid = new_bundle["uuid"]
            return bundle

        except HTTPError as err:
            print(f"Exception creating bundle {bundle.name} for item {item.name} ({item.uuid}). Error code {err.response.status_code}, reason {err.response.reason}")
            raise BundleException(f"Error creating bundle {bundle.name} for item {item.name} ({item.uuid}). Error code {err.response.status_code}, reason {err.response.reason}")
        except Exception as err1:
            print(f"Exception creating bundle {bundle.name} for item {item.name} ({item.uuid}). Error: {str(err1)}")
            raise BundleException(f"Error creating bundle {bundle.name} for item {item.name} ({item.uuid}). Error: {str(err1)}")

    def bundle_add_primary_bitstream(self, bundle: Bundle, bitstream: Bitstream):
        try:
            _ = self._bundle_request.add_primary_bitstream(bundle.uuid, bitstream.uuid)
        except HTTPError as err:
            print(f"Exception setting primary bitstream for bundle {bundle.name} ({bundle.uuid}) to {bitstream.name} ({bitstream.uuid}). Error code {err.response.status_code}, reason {err.response.reason}")
            raise BundleException(f"Error setting primary bitstream for bundle {bundle.name} ({bundle.uuid}) to {bitstream.name} ({bitstream.uuid}). Error code {err.response.status_code}, reason {err.response.reason}")
        except Exception as err1:
            print(f"Exception setting primary bitstream for bundle {bundle.name} ({bundle.uuid}) to {bitstream.name} ({bitstream.uuid}). Error: {str(err1)}")
            raise BundleException(f"Exception setting primary bitstream for bundle {bundle.name} ({bundle.uuid}) to {bitstream.name} ({bitstream.uuid}). Error: {str(err1)}")