88 lines
2.2 KiB
Python
88 lines
2.2 KiB
Python
# Copyright (c) 2015-2023 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
|
|
|
|
from cgtsclient.common import base
|
|
from cgtsclient import exc
|
|
|
|
|
|
CREATION_ATTRIBUTES = ['software_version', 'compatible_version',
|
|
'required_patches']
|
|
|
|
IMPORT_ATTRIBUTES = ['path_to_iso', 'path_to_sig', 'active', 'local',
|
|
'inactive']
|
|
|
|
|
|
class Load(base.Resource):
|
|
def __repr__(self):
|
|
return "<loads %s>" % self._info
|
|
|
|
|
|
class LoadManager(base.Manager):
|
|
resource_class = Load
|
|
|
|
def list(self):
|
|
return self._list('/v1/loads/', "loads")
|
|
|
|
def get(self, load_id):
|
|
path = '/v1/loads/%s' % load_id
|
|
try:
|
|
return self._list(path)[0]
|
|
except IndexError:
|
|
return None
|
|
|
|
def _create_load(self, load, path):
|
|
if set(list(load.keys())) != set(CREATION_ATTRIBUTES):
|
|
raise exc.InvalidAttribute()
|
|
|
|
return self._create(path, load)
|
|
|
|
def create(self, load):
|
|
path = '/v1/loads/'
|
|
self._create_load(load, path)
|
|
|
|
def import_load_metadata(self, load):
|
|
path = '/v1/loads/import_load_metadata'
|
|
return self._create_load(load, path)
|
|
|
|
def import_load(self, **kwargs):
|
|
path = '/v1/loads/import_load'
|
|
local = kwargs.pop('local')
|
|
load_info = {}
|
|
|
|
for key, value in kwargs.items():
|
|
if key in IMPORT_ATTRIBUTES:
|
|
if isinstance(value, bool):
|
|
load_info[key] = str(value).lower()
|
|
else:
|
|
load_info[key] = value
|
|
else:
|
|
raise exc.InvalidAttribute(key)
|
|
|
|
if local:
|
|
return self._create(path, body=load_info)
|
|
|
|
data = {
|
|
'active': load_info.pop('active', 'false'),
|
|
'inactive': load_info.pop('inactive', 'false'),
|
|
}
|
|
|
|
json_data = self._upload_multipart(
|
|
path,
|
|
body=load_info,
|
|
data=data,
|
|
check_exceptions=True,
|
|
)
|
|
|
|
return self.resource_class(self, json_data)
|
|
|
|
def delete(self, load_id):
|
|
path = '/v1/loads/%s' % load_id
|
|
return self._delete(path)
|
|
|
|
def update(self, load_id, patch):
|
|
path = '/v1/loads/%s' % load_id
|
|
return self._update(path, patch)
|