Island: Fix mypy issues in networkmap.py

This commit is contained in:
Kekoa Kaaikala 2022-09-21 18:28:58 +00:00
parent 24848e62df
commit 218b341006
1 changed files with 8 additions and 6 deletions

View File

@ -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