devicehub-teal/ereuse_devicehub/resources/documents/device_row.py

from collections import OrderedDict

from flask import current_app

from ereuse_devicehub.resources.action.models import BenchmarkDataStorage, RateComputer, \
    TestDataStorage
from ereuse_devicehub.resources.device import models as d, states


class DeviceRow(OrderedDict):
    NUMS = {
        d.Display.t: 1,
        d.Processor.t: 2,
        d.GraphicCard.t: 2,
        d.Motherboard.t: 1,
        d.NetworkAdapter.t: 2,
        d.SoundCard.t: 2
    }

    # TODO Add more fields information
    def __init__(self, device: d.Device) -> None:
        super().__init__()
        self.device = device
        # General information about device
        self['Type'] = device.t
        if isinstance(device, d.Computer):
            self['Chassis'] = device.chassis
        else:
            self['Chassis'] = ''
        self['Tag 1'] = self['Tag 2'] = self['Tag 3'] = ''
        for i, tag in zip(range(1, 3), device.tags):
            self['Tag {}'.format(i)] = format(tag)
        self['Serial Number'] = convert_none_to_empty_str(device.serial_number)
        self['Model'] = convert_none_to_empty_str(device.model)
        self['Manufacturer'] = convert_none_to_empty_str(device.manufacturer)
        self['Registered in'] = format(device.created, '%c')
        try:
            self['Physical state'] = device.last_action_of(*states.Physical.actions()).t
        except LookupError:
            self['Physical state'] = ''
        try:
            self['Trading state'] = device.last_action_of(*states.Trading.actions()).t
        except LookupError:
            self['Trading state'] = ''
        self['Price'] = convert_none_to_empty_str(device.price)
        if isinstance(device, d.Computer):
            self['Processor'] = convert_none_to_empty_str(device.processor_model)
            self['RAM (MB)'] = convert_none_to_empty_str(device.ram_size)
            self['Data Storage Size (MB)'] = convert_none_to_empty_str(device.data_storage_size)
        rate = device.rate
        if rate:
            self['Rate'] = rate.rating
            self['Range'] = rate.rating_range
            assert isinstance(rate, RateComputer)
            self['Processor Rate'] = rate.processor
            self['Processor Range'] = rate.processor_range
            self['RAM Rate'] = rate.ram
            self['RAM Range'] = rate.ram_range
            self['Data Storage Rate'] = rate.data_storage
            self['Data Storage Range'] = rate.data_storage_range
        # More specific information about components
        if isinstance(device, d.Computer):
            self.components()

    def components(self):
        """Function to get all components information of a device."""
        assert isinstance(self.device, d.Computer)
        # todo put an input specific order (non alphabetic) & where are a list of types components
        for type in sorted(current_app.resources[d.Component.t].subresources_types):  # type: str
            max = self.NUMS.get(type, 4)
            if type not in ['Component', 'HardDrive', 'SolidStateDrive']:
                i = 1
                for component in (r for r in self.device.components if r.type == type):
                    self.fill_component(type, i, component)
                    i += 1
                    if i > max:
                        break
                while i <= max:
                    self.fill_component(type, i)
                    i += 1

    def fill_component(self, type, i, component=None):
        """Function to put specific information of components
        in OrderedDict (csv)
        :param type: type of component
        :param component: device.components
        """
        self['{} {}'.format(type, i)] = format(component) if component else ''
        self['{} {} Manufacturer'.format(type, i)] = component.serial_number if component else ''
        self['{} {} Model'.format(type, i)] = component.serial_number if component else ''
        self['{} {} Serial Number'.format(type, i)] = component.serial_number if component else ''

        """Particular fields for component GraphicCard."""
        if isinstance(component, d.GraphicCard):
            self['{} {} Memory (MB)'.format(type, i)] = component.memory

        """Particular fields for component DataStorage.t -> 
        (HardDrive, SolidStateDrive)
        """
        if isinstance(component, d.DataStorage):
            self['{} {} Size (MB)'.format(type, i)] = component.size
            self['{} {} Privacy'.format(type, i)] = component.privacy
            try:
                self['{} {} Lifetime'.format(type, i)] = component.last_action_of(
                    TestDataStorage).lifetime
            except:
                self['{} {} Lifetime'.format(type, i)] = ''
            try:
                self['{} {} Reading speed'.format(type, i)] = component.last_action_of(
                    BenchmarkDataStorage).read_speed
            except:
                self['{} {} Reading speed'.format(type, i)] = ''
            try:
                self['{} {} Writing speed'.format(type, i)] = component.last_action_of(
                    BenchmarkDataStorage).write_speed
            except:
                self['{} {} Writing speed'.format(type, i)] = ''

        """Particular fields for component Processor."""
        if isinstance(component, d.Processor):
            self['{} {} Number of cores'.format(type, i)] = component.cores
            self['{} {} Speed (GHz)'.format(type, i)] = component.speed

        """Particular fields for component RamModule."""
        if isinstance(component, d.RamModule):
            self['{} {} Size (MB)'.format(type, i)] = component.size
            self['{} {} Speed (MHz)'.format(type, i)] = component.speed

        # todo add Display, NetworkAdapter, etc...


class StockRow(OrderedDict):
    def __init__(self, device: d.Device) -> None:
        super().__init__()
        self.device = device
        self['Type'] = convert_none_to_empty_str(device.t)
        if isinstance(device, d.Computer):
            self['Chassis'] = device.chassis
        else:
            self['Chassis'] = ''
        self['Serial Number'] = convert_none_to_empty_str(device.serial_number)
        self['Model'] = convert_none_to_empty_str(device.model)
        self['Manufacturer'] = convert_none_to_empty_str(device.manufacturer)
        self['Registered in'] = format(device.created, '%c')
        try:
            self['Physical state'] = device.last_action_of(*states.Physical.actions()).t
        except LookupError:
            self['Physical state'] = ''
        try:
            self['Trading state'] = device.last_action_of(*states.Trading.actions()).t
        except LookupError:
            self['Trading state'] = ''
            self['Price'] = convert_none_to_empty_str(device.price)
            self['Processor'] = convert_none_to_empty_str(device.processor_model)
            self['RAM (MB)'] = convert_none_to_empty_str(device.ram_size)
            self['Data Storage Size (MB)'] = convert_none_to_empty_str(device.data_storage_size)
        rate = device.rate
        if rate:
            self['Rate'] = rate.rating
            self['Range'] = rate.rating_range
            assert isinstance(rate, RateComputer)
            self['Processor Rate'] = rate.processor
            self['Processor Range'] = rate.processor_range
            self['RAM Rate'] = rate.ram
            self['RAM Range'] = rate.ram_range
            self['Data Storage Rate'] = rate.data_storage
            self['Data Storage Range'] = rate.data_storage_range


def convert_none_to_empty_str(s):
    if s is None:
        return ''
    return s