src/box/return_node.py

import logging


class ReturnNode:
    def __init__(self, box, generator):
        self.box = box
        self.generator = generator
        logging.debug("Constructed return node")

    def to_python(self, indent="    "):
        logging.debug("Generating Python for return node")
        result = indent + "return"

        return_vals = []

        for port in self.box.input_data_flow_ports:
            input_port = self.generator._find_destination_connection(port, "left")
            input_box = self.generator.port_box_map[input_port]
            return_vals.append(
                self.generator._get_output_data_name(input_box, input_port)
            )

        logging.debug("  Returning " + str(len(return_vals)) + " values")

        for i, val in enumerate(return_vals):
            result += " " + val
            if i < len(return_vals) - 1:
                result += ", "
        result += "\n"

        return result