mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-28 10:05:05 -05:00
Matt HauffandGitHub
40db4635a8
* Enable PEP604 Ruff rules * Fix harcoded signature in test * Hack CLVMStreamable test with note to fast follow
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass, field
|
|
|
|
from chia.protocols.harvester_protocol import Plot
|
|
|
|
|
|
@dataclass
|
|
class DeltaType:
|
|
additions: dict[str, Plot] | list[str]
|
|
removals: list[str]
|
|
|
|
def __str__(self) -> str:
|
|
return f"+{len(self.additions)}/-{len(self.removals)}"
|
|
|
|
def clear(self) -> None:
|
|
self.additions.clear()
|
|
self.removals.clear()
|
|
|
|
def empty(self) -> bool:
|
|
return len(self.additions) == 0 and len(self.removals) == 0
|
|
|
|
|
|
@dataclass
|
|
class PlotListDelta(DeltaType):
|
|
additions: dict[str, Plot] = field(default_factory=dict)
|
|
removals: list[str] = field(default_factory=list)
|
|
|
|
|
|
@dataclass
|
|
class PathListDelta(DeltaType):
|
|
additions: list[str] = field(default_factory=list)
|
|
removals: list[str] = field(default_factory=list)
|
|
|
|
@staticmethod
|
|
def from_lists(old: list[str], new: list[str]) -> PathListDelta:
|
|
return PathListDelta([x for x in new if x not in old], [x for x in old if x not in new])
|
|
|
|
|
|
@dataclass
|
|
class Delta:
|
|
valid: PlotListDelta = field(default_factory=PlotListDelta)
|
|
invalid: PathListDelta = field(default_factory=PathListDelta)
|
|
keys_missing: PathListDelta = field(default_factory=PathListDelta)
|
|
duplicates: PathListDelta = field(default_factory=PathListDelta)
|
|
|
|
def empty(self) -> bool:
|
|
return self.valid.empty() and self.invalid.empty() and self.keys_missing.empty() and self.duplicates.empty()
|
|
|
|
def __str__(self) -> str:
|
|
return (
|
|
f"[valid {self.valid}, invalid {self.invalid}, keys missing: {self.keys_missing}, "
|
|
f"duplicates: {self.duplicates}]"
|
|
)
|
|
|
|
def clear(self) -> None:
|
|
self.valid.clear()
|
|
self.invalid.clear()
|
|
self.keys_missing.clear()
|
|
self.duplicates.clear()
|