forked from p15670423/monkey
Island: Fix mypy issues in networkmap.py
This commit is contained in:
parent
24848e62df
commit
218b341006
|
@ -1,6 +1,14 @@
|
|||
from dataclasses import dataclass
|
||||
from typing import Mapping, Sequence
|
||||
|
||||
from monkey_island.cc.models import Machine
|
||||
|
||||
|
||||
@dataclass
|
||||
class Arc:
|
||||
dst_machine: Machine # noqa: F821
|
||||
status: str
|
||||
|
||||
|
||||
# This is the most concise way to represent a graph:
|
||||
# Machine id as key, Arch list as a value
|
||||
|
@ -9,9 +17,3 @@ from typing import Mapping, Sequence
|
|||
@dataclass
|
||||
class NetworkMap:
|
||||
nodes: Mapping[str, Sequence[Arc]] # noqa: F821
|
||||
|
||||
|
||||
@dataclass
|
||||
class Arc:
|
||||
dst_machine: Machine # noqa: F821
|
||||
status: str
|
||||
|
|
Loading…
Reference in New Issue