57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
|
from datetime import datetime, timezone
|
||
|
|
from distutils.version import StrictVersion
|
||
|
|
from enum import Enum, unique
|
||
|
|
from typing import List, Optional
|
||
|
|
from uuid import UUID
|
||
|
|
|
||
|
|
import inflection
|
||
|
|
from ereuse_utils import cli
|
||
|
|
from ereuse_utils.cli import Line
|
||
|
|
from ereuse_utils.session import DevicehubClient
|
||
|
|
|
||
|
|
from ereuse_workbench.computer import Component, Computer, DataStorage, SoundCard
|
||
|
|
from ereuse_workbench.utils import Dumpeable
|
||
|
|
|
||
|
|
|
||
|
|
@unique
|
||
|
|
class SnapshotSoftware(Enum):
|
||
|
|
"""The algorithm_software used to perform the Snapshot."""
|
||
|
|
Workbench = 'Workbench'
|
||
|
|
AndroidApp = 'AndroidApp'
|
||
|
|
Web = 'Web'
|
||
|
|
DesktopApp = 'DesktopApp'
|
||
|
|
|
||
|
|
|
||
|
|
class Snapshot(Dumpeable):
|
||
|
|
"""
|
||
|
|
Generates the Snapshot report for Devicehub by obtaining the
|
||
|
|
data from the computer, performing benchmarks and tests...
|
||
|
|
|
||
|
|
After instantiating the class, run :meth:`.computer` before any
|
||
|
|
other method.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, uuid, software, version, lshw, hwinfo):
|
||
|
|
self.type = 'Snapshot'
|
||
|
|
self.uuid = uuid
|
||
|
|
self.software = software
|
||
|
|
self.version = version
|
||
|
|
self.lshw = lshw
|
||
|
|
self.hwinfo = hwinfo
|
||
|
|
self.endTime = datetime.now(timezone.utc)
|
||
|
|
self.closed = False
|
||
|
|
self.elapsed = None
|
||
|
|
self.device = None # type: Computer
|
||
|
|
self.components = None # type: List[Component]
|
||
|
|
self._storages = None
|
||
|
|
|
||
|
|
def computer(self):
|
||
|
|
"""Retrieves information about the computer and components."""
|
||
|
|
self.device, self.components = Computer.run(self.lshw, self.hwinfo)
|
||
|
|
self._storages = tuple(c for c in self.components if isinstance(c, DataStorage))
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
"""Closes the Snapshot
|
||
|
|
"""
|
||
|
|
self.closed = True
|