initial changes to mimic prior address creation

This commit is contained in:
Blake Harnden 2020-05-01 13:39:27 -07:00
parent 7e0ead0766
commit 9a42368221
3 changed files with 70 additions and 32 deletions

View file

@ -1,7 +1,7 @@
import logging
from typing import TYPE_CHECKING, Dict, List, Optional, Set, Union
from core.api.grpc.core_pb2 import NodeType
from core.api.grpc.core_pb2 import Node, NodeType
from core.gui.images import ImageEnum, Images, TypeToImage
if TYPE_CHECKING:
@ -64,8 +64,13 @@ class NodeUtils:
RJ45_NODES = {NodeType.RJ45}
IGNORE_NODES = {NodeType.CONTROL_NET, NodeType.PEER_TO_PEER}
NODE_MODELS = {"router", "host", "PC", "mdr", "prouter"}
ROUTER_NODES = {"router", "mdr"}
ANTENNA_ICON = None
@classmethod
def is_router_node(cls, node: Node) -> bool:
return cls.is_model_node(node.type) and node.model in cls.ROUTER_NODES
@classmethod
def is_ignore_node(cls, node_type: NodeType) -> bool:
return node_type in cls.IGNORE_NODES