generate_TableFunctionsFactory_init.TextEncodingDictTransformer Class Reference

Inheritance diagram for generate_TableFunctionsFactory_init.TextEncodingDictTransformer:

Collaboration diagram for generate_TableFunctionsFactory_init.TextEncodingDictTransformer:

Public Member Functions
def	visit_udtf_node
Public Member Functions inherited from generate_TableFunctionsFactory_init.AstTransformer
def	visit_udtf_node
def	visit_composed_node
def	visit_arg_node
def	visit_primitive_node
def	visit_template_node
def	visit_annotation_node
Public Member Functions inherited from generate_TableFunctionsFactory_init.AstVisitor
def	visit_udtf_node
def	visit_composed_node
def	visit_arg_node
def	visit_primitive_node
def	visit_annotation_node
def	visit_template_node

Detailed Description

Definition at line 840 of file generate_TableFunctionsFactory_init.py.

Member Function Documentation

def generate_TableFunctionsFactory_init.TextEncodingDictTransformer.visit_udtf_node	(		self,
			udtf_node
	)

* Add default_input_id to Column(List)<TextEncodingDict> without one

Definition at line 841 of file generate_TableFunctionsFactory_init.py.

References run_benchmark_import.type.

    def visit_udtf_node(self, udtf_node):
        """
        * Add default_input_id to Column(List)<TextEncodingDict> without one
        """
        udtf_node = super(type(self), self).visit_udtf_node(udtf_node)
        # add default input_id
        default_input_id = None
        for idx, t in enumerate(udtf_node.inputs):

            if not isinstance(t.type, ComposedNode):
                continue
            if default_input_id is not None:
                pass
            elif t.type.is_column_text_encoding_dict() or t.type.is_column_array_text_encoding_dict():
                default_input_id = AnnotationNode('input_id', 'args<%s>' % (idx,))
            elif t.type.is_column_list_text_encoding_dict():
                default_input_id = AnnotationNode('input_id', 'args<%s, 0>' % (idx,))

        for t in udtf_node.outputs:
            if isinstance(t.type, ComposedNode) and t.type.is_any_text_encoding_dict():
                for a in t.annotations:
                    if a.key == 'input_id':
                        break
                else:
                    if default_input_id is None:
                        raise TypeError('Cannot parse line "%s".\n'
                                        'Missing TextEncodingDict input?' %
                                        (udtf_node.line))
                    t.annotations.append(default_input_id)

        return udtf_node