actionsProcessor.py

import emoji
import pallete
import node
import gvInput

class ActionsProcessor():
    """ Class containing neccessary methods for putting actions scheduled by blivet into graph"""
    def __init__(self, actions_list, node_list, edge_list, path_to_pallete):
        self.actions = actions_list
        self.node_list = node_list
        self.edge_list = edge_list
        self.pallete = pallete.Pallete(path_to_pallete)
        self.gv_input = gvInput.GvInput(node_list, edge_list, path_to_pallete)

    def process_actions(self):
        """ Main method for processing actions """
        for action in self.actions:
            found_node = self.find_node(action.device.name)
            if not found_node:
                found_node = node.Node(action.device.name, action.device.type, action.device.format, action.device.size, action.device.path)
                self.gv_input.process_node(found_node,action.device)
            self.process_action(action, found_node)

    def process_action(self, action, node):
        """ Helper function for processing actions, finds out what does each action do and sets the appropriate attributed of the node
        :param obj action The action to be processed.
        :param obj node The node to be changed."""
        if action.isFormat:
            node.add_emoji("Fmt:")
            node.addAttribute("action", "Format")
        else:
            node.add_emoji("Dev:")
            node.addAttribute("action", "Device")
        if action.isDestroy or action.isRemove:
            print("Adding action: Delete for node: " + node.getName())
            node.addAttribute("action", "delete")
            node.change_color(self.pallete.complement["2"])
            node.add_emoji(emoji.emojize(":fire:"))
        if action.isCreate or action.isAdd:
            print("Adding action: Add for node: " + node.getName())
            node.addAttribute("action", "add")
            node.change_color(self.pallete.primary["2"])
            node.add_emoji(emoji.emojize(":building_construction:"))
        if action.isResize or action.isShrink or action.isGrow:
            print("Adding action: Resize for node: " + node.getName())
            node.addAttribute("action", "resize")
            node.change_color(self.pallete.secondary_second["2"])
            node.add_emoji(emoji.emojize(":wrench:"))

    def find_node(self, dev_name):
        """ Helper function that searches node_list for a node using it's name
            :param str dev_name The name of node / device that should be looked up."""
        for found_node in self.node_list:
            if found_node.getName() == dev_name: 
                print("Found node: " + found_node.getName())
                return found_node