generate_TableFunctionsFactory_init.FieldAnnotationTransformer Class Reference

Inheritance diagram for generate_TableFunctionsFactory_init.FieldAnnotationTransformer:

Collaboration diagram for generate_TableFunctionsFactory_init.FieldAnnotationTransformer:

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 859 of file generate_TableFunctionsFactory_init.py.

Member Function Documentation

def generate_TableFunctionsFactory_init.FieldAnnotationTransformer.visit_udtf_node	(		self,
			udtf_node
	)

* Generate fields annotation to Cursor if non-existing

Definition at line 861 of file generate_TableFunctionsFactory_init.py.

References run_benchmark_import.type.

    def visit_udtf_node(self, udtf_node):
        """
        * Generate fields annotation to Cursor if non-existing
        """
        udtf_node = super(type(self), self).visit_udtf_node(udtf_node)

        for _, t in enumerate(udtf_node.inputs):

            if not isinstance(t.type, ComposedNode):
                continue

            if t.type.is_cursor() and t.get_annotation('fields') is None:
                fields = list(PrimitiveNode(a.get_annotation('name', 'field%s' % i)) for i, a in enumerate(t.type.inner))
                t.annotations.append(AnnotationNode('fields', fields))

        return udtf_node