TableFunctionsFactory_transformers.DefaultValueAnnotationTransformer Class Reference

Inheritance diagram for TableFunctionsFactory_transformers.DefaultValueAnnotationTransformer:

Collaboration diagram for TableFunctionsFactory_transformers.DefaultValueAnnotationTransformer:

Public Member Functions
def	visit_udtf_node
Public Member Functions inherited from TableFunctionsFactory_transformers.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 TableFunctionsFactory_transformers.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 306 of file TableFunctionsFactory_transformers.py.

Member Function Documentation

def TableFunctionsFactory_transformers.DefaultValueAnnotationTransformer.visit_udtf_node	(		self,
			udtf_node
	)

* Typechecks default value annotations.

Definition at line 307 of file TableFunctionsFactory_transformers.py.

References run_benchmark_import.type.

    def visit_udtf_node(self, udtf_node):
        """
        * Typechecks default value annotations.
        """
        udtf_node = super(type(self), self).visit_udtf_node(udtf_node)

        for t in udtf_node.inputs:
            for a in filter(lambda x: x.key == "default", t.annotations):
                if not t.type.is_scalar():
                    raise TransformerException(
                        'Error in function "%s", input annotation \'%s=%s\'. '
                        '\"default\" annotation is only supported for scalar types!'\
                        % (udtf_node.name, a.key, a.value)
                    )
                literal = literal_eval(a.value)
                lst = [(bool, 'is_boolean_scalar'), (int, 'is_integer_scalar'), (float, 'is_float_scalar'),
                (str, 'is_string_scalar')]

                for (cls, mthd) in lst:
                    if type(literal) is cls:
                        assert isinstance(t, tf_node.ArgNode)
                        m = getattr(t.type, mthd)
                        if not m():
                            raise TransformerException(
                                'Error in function "%s", input annotation \'%s=%s\'. '
                                'Argument is of type "%s" but value type was inferred as "%s".'
                                % (udtf_node.name, a.key, a.value, t.type.type, type(literal).__name__))
                        break

        return udtf_node