74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
# Copyright 2018 Huawei Technologies Co.,LTD.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from cyborg import objects
|
|
from cyborg.objects import device
|
|
from cyborg.objects import fields
|
|
|
|
|
|
def get_fake_devices_as_dict():
|
|
device1 = {
|
|
"id": 1,
|
|
"vendor": "0xABCD",
|
|
"uuid": "1c6c9033-560d-4a7a-bb8e-94455d1e7825",
|
|
"hostname": "test-node-1",
|
|
"vendor_board_info": "fake_vendor_info",
|
|
"model": "miss model info",
|
|
"type": "FPGA",
|
|
"std_board_info": "{'class': 'Fake class', 'device_id': '0xabcd'}"
|
|
}
|
|
device2 = {
|
|
"id": 2,
|
|
"vendor": "0xDCBA",
|
|
"uuid": "1c6c9033-560d-4a7a-bb8e-94455d1e7826",
|
|
"hostname": "test-node-2",
|
|
"vendor_board_info": "fake_vendor_info",
|
|
"model": "miss model info",
|
|
"type": "GPU",
|
|
"std_board_info": "{'class': 'Fake class', 'device_id': '0xdcba'}"
|
|
}
|
|
return [device1, device2]
|
|
|
|
|
|
def _convert_from_dict_to_obj(device_dict):
|
|
obj_device = device.Device()
|
|
for field in device_dict.keys():
|
|
obj_device[field] = device_dict[field]
|
|
return obj_device
|
|
|
|
|
|
def _convert_to_db_device(device_dict):
|
|
for name, field in objects.Device.fields.items():
|
|
if name in device_dict:
|
|
continue
|
|
if field.nullable:
|
|
device_dict[name] = None
|
|
elif field.default != fields.UnspecifiedDefault:
|
|
device_dict[name] = field.default
|
|
else:
|
|
raise Exception('fake_db_device needs help with %s' % name)
|
|
return device_dict
|
|
|
|
|
|
def get_db_devices():
|
|
devices_list = get_fake_devices_as_dict()
|
|
db_devices = list(map(_convert_to_db_device, devices_list))
|
|
return db_devices
|
|
|
|
|
|
def get_fake_devices_objs():
|
|
devices_list = get_fake_devices_as_dict()
|
|
obj_devices = list(map(_convert_from_dict_to_obj, devices_list))
|
|
return obj_devices
|