* Dont remove api key during migration * Fix return * Fix test * Make lambda more readable * Move fixtures to init, move migration test to test_init.py * Refactor config_entry data assignment * Refactor system_bridge migration tests * Fix type for debug message * Fix type for debug message * Remove duplicated unused code (rebase error) * Refactor test_migration_minor_2_to_1 to test_migration_minor_future_to_2 * Fix version check in async_migrate_entry * Update migration logic to handle future minor version * Add ConfigEntryState assertion in test_init.py * Change condition to minor_version < 2 Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Refactor system bridge migration tests * Remove minor downgrade code * Update tests/components/system_bridge/test_init.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Update tests/components/system_bridge/test_init.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Apply suggestions from code review Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Move dataclass to store requested data to data.py * Use dataclass in config flow * Move media player and sensor onto data.py dataclass * Move data and handler inside validate --------- Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
29 lines
704 B
Python
29 lines
704 B
Python
"""System Bridge integration data."""
|
|
from dataclasses import dataclass, field
|
|
|
|
from systembridgemodels.modules import (
|
|
CPU,
|
|
GPU,
|
|
Battery,
|
|
Disks,
|
|
Display,
|
|
Media,
|
|
Memory,
|
|
Process,
|
|
System,
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class SystemBridgeData:
|
|
"""System Bridge Data."""
|
|
|
|
battery: Battery = field(default_factory=Battery)
|
|
cpu: CPU = field(default_factory=CPU)
|
|
disks: Disks = None
|
|
displays: list[Display] = field(default_factory=list[Display])
|
|
gpus: list[GPU] = field(default_factory=list[GPU])
|
|
media: Media = field(default_factory=Media)
|
|
memory: Memory = None
|
|
processes: list[Process] = field(default_factory=list[Process])
|
|
system: System = None
|