def visit_BoolOp(self, node):
# Handle left and right
values = []
for value in node.values:
values.append(self.visit(value))
# TODO: Truthiness is not supported! Probably need a Union type
# TODO: Literals used as truthy value
# Handle operation
return BoolType()
def visit_UnaryOp(self, node):
# Handle operand
operand = self.visit(node.operand)
if isinstance(node.op, ast.Not):
return BoolType()
elif isinstance(operand, UnknownType):
return UnknownType()
elif type(node.op) in VALID_UNARYOP_TYPES:
op_lookup = VALID_UNARYOP_TYPES[type(node.op)]
if type(node.op) in op_lookup:
op_lookup = op_lookup[type(node.op)]
if type(operand) in op_lookup:
op_lookup = op_lookup[type(operand)]
return op_lookup(operand)
return UnknownType()
def visit_Compare(self, node):
# Handle left and right
left = self.visit(node.left)
comparators = [self.visit(compare) for compare in node.comparators]
# Handle ops
for op, right in zip(node.ops, comparators):
if isinstance(op, (ast.Eq, ast.NotEq, ast.Is, ast.IsNot)):
continue
elif isinstance(op, (ast.Lt, ast.LtE, ast.GtE, ast.Gt)):
if are_types_equal(left, right):
if isinstance(left, ORDERABLE_TYPES):
continue
elif isinstance(op, (ast.In, ast.NotIn)):
if isinstance(right, INDEXABLE_TYPES):
continue
self.report_issue("Incompatible types", {
"left": left,
"right": right,
"operation": op
})
return BoolType()
def visit_Name(self, node):
name = node.id
if name == "___":
self.report_issue("Unconnected blocks")
if isinstance(node.ctx, ast.Load):
if name == "True" or name == "False":
return BoolType()
elif name == "None":
return NoneType()
else:
variable = self.find_variable_scope(name)
builtin = get_builtin_function(name)
if not variable.exists and builtin:
return builtin
else:
state = self.load_variable(name)
return state.type
else:
variable = self.find_variable_scope(name)
if variable.exists:
return variable.state.type
else:
return UnknownType()
def get_builtin_module(name):
if name == 'matplotlib':
return ModuleType('matplotlib',
submodules={
'pyplot':
ModuleType('pyplot',
fields={
'plot':
FunctionType(name='plot',
returns=NoneType()),
'hist':
FunctionType(name='hist',
returns=NoneType()),
'scatter':
FunctionType(name='scatter',
returns=NoneType()),
'show':
FunctionType(name='show',
returns=NoneType()),
'xlabel':
FunctionType(name='xlabel',
returns=NoneType()),
'ylabel':
FunctionType(name='ylabel',
returns=NoneType()),
'title':
FunctionType(name='title',
returns=NoneType()),
})
})
elif name == 'pprint':
return ModuleType(
'pprint',
fields={'pprint': FunctionType(name='pprint', returns=NoneType())})
elif name == 'random':
return ModuleType('random',
fields={
'randint':
FunctionType(name='randint', returns=NumType())
})
elif name == 'string':
return ModuleType('string',
fields={
'letters': StrType(empty=False),
'digits': StrType(empty=False),
'ascii_letters': StrType(empty=False),
'punctuation': StrType(empty=False),
'printable': StrType(empty=False),
'whitespace': StrType(empty=False),
'ascii_uppercase': StrType(empty=False),
'ascii_lowercase': StrType(empty=False),
'hexdigits': StrType(empty=False),
'octdigits': StrType(empty=False),
})
elif name == 'turtle':
return ModuleType('turtle',
fields={
'forward':
FunctionType(name='forward', returns=NoneType()),
'backward':
FunctionType(name='backward',
returns=NoneType()),
'color':
FunctionType(name='color', returns=NoneType()),
'right':
FunctionType(name='right', returns=NoneType()),
'left':
FunctionType(name='left', returns=NoneType()),
})
elif name == 'parking':
return ModuleType('parking',
fields={
'Time':
FunctionType(name='Time', returns=TimeType()),
'now':
FunctionType(name='now', returns=TimeType()),
'Day':
FunctionType(name='Day', returns=DayType()),
'today':
FunctionType(name='today', returns=DayType()),
}),
elif name == 'math':
return ModuleType('math',
fields={
'ceil':
FunctionType(name='ceil', returns=NumType()),
'copysign':
FunctionType(name='copysign', returns=NumType()),
'fabs':
FunctionType(name='fabs', returns=NumType()),
'factorial':
FunctionType(name='factorial',
returns=NumType()),
'floor':
FunctionType(name='floor', returns=NumType()),
'fmod':
FunctionType(name='fmod', returns=NumType()),
'frexp':
FunctionType(name='frexp', returns=NumType()),
'fsum':
FunctionType(name='fsum', returns=NumType()),
'gcd':
FunctionType(name='gcd', returns=NumType()),
'isclose':
FunctionType(name='isclose', returns=BoolType()),
'isfinite':
FunctionType(name='isfinite',
returns=BoolType()),
'isinf':
FunctionType(name='isinf', returns=BoolType()),
'isnan':
FunctionType(name='isnan', returns=BoolType()),
'ldexp':
FunctionType(name='ldexp', returns=NumType()),
'modf':
FunctionType(name='modf', returns=NumType()),
'trunc':
FunctionType(name='trunc', returns=NumType()),
'log':
FunctionType(name='log', returns=NumType()),
'log1p':
FunctionType(name='log1p', returns=NumType()),
'log2':
FunctionType(name='log2', returns=NumType()),
'log10':
FunctionType(name='log10', returns=NumType()),
'pow':
FunctionType(name='pow', returns=NumType()),
'sqrt':
FunctionType(name='sqrt', returns=NumType()),
'sin':
FunctionType(name='sin', returns=NumType()),
'cos':
FunctionType(name='cos', returns=NumType()),
'tan':
FunctionType(name='tan', returns=NumType()),
'asin':
FunctionType(name='asin', returns=NumType()),
'acos':
FunctionType(name='acos', returns=NumType()),
'atan':
FunctionType(name='atan', returns=NumType()),
'atan2':
FunctionType(name='atan2', returns=NumType()),
'hypot':
FunctionType(name='hypot', returns=NumType()),
'degrees':
FunctionType(name='degrees', returns=NumType()),
'radians':
FunctionType(name='radians', returns=NumType()),
'sinh':
FunctionType(name='sinh', returns=NumType()),
'cosh':
FunctionType(name='cosh', returns=NumType()),
'tanh':
FunctionType(name='tanh', returns=NumType()),
'asinh':
FunctionType(name='asinh', returns=NumType()),
'acosh':
FunctionType(name='acosh', returns=NumType()),
'atanh':
FunctionType(name='atanh', returns=NumType()),
'erf':
FunctionType(name='erf', returns=NumType()),
'erfc':
FunctionType(name='erfc', returns=NumType()),
'gamma':
FunctionType(name='gamma', returns=NumType()),
'lgamma':
FunctionType(name='lgamma', returns=NumType()),
'pi':
NumType(),
'e':
NumType(),
'tau':
NumType(),
'inf':
NumType(),
'nan':
NumType(),
})
def get_builtin_function(name):
# Void Functions
if name == "print":
return FunctionType(name="print", returns=NoneType())
# Math Functions
elif name == "int":
return FunctionType(name="int", returns=NumType())
elif name == "abs":
return FunctionType(name="abs", returns=NumType())
elif name == "float":
return FunctionType(name="float", returns=NumType())
elif name == "len":
return FunctionType(name="len", returns=NumType())
elif name == "ord":
return FunctionType(name="ord", returns=NumType())
elif name == "pow":
return FunctionType(name="pow", returns=NumType())
elif name == "round":
return FunctionType(name="round", returns=NumType())
elif name == "sum":
return FunctionType(name="sum", returns=NumType())
# Boolean Functions
elif name == "bool":
return FunctionType(name="bool", returns=BoolType())
elif name == "all":
return FunctionType(name="all", returns=BoolType())
elif name == "any":
return FunctionType(name="any", returns=BoolType())
elif name == "isinstance":
return FunctionType(name="isinstance", returns=BoolType())
# String Functions
elif name == "input":
return FunctionType(name="input", returns=StrType())
elif name == "str":
return FunctionType(name="str", returns=StrType())
elif name == "chr":
return FunctionType(name="chr", returns=StrType())
elif name == "repr":
return FunctionType(name="repr", returns=StrType())
# File Functions
elif name == "open":
return FunctionType(name="open", returns=FileType())
# List Functions
elif name == "map":
return FunctionType(name="map", returns=ListType(empty=False))
elif name == "list":
return FunctionType(name="list",
definition=_builtin_sequence_constructor(ListType))
# Set Functions
elif name == "set":
return FunctionType(name="set",
definition=_builtin_sequence_constructor(SetType))
# Dict Functions
elif name == "dict":
return FunctionType(name="dict", returns=DictType())
# Pass through
elif name == "sorted":
return FunctionType(name="sorted", returns='identity')
elif name == "reversed":
return FunctionType(name="reversed", returns='identity')
elif name == "filter":
return FunctionType(name="filter", returns='identity')
# Special Functions
elif name == "type":
return FunctionType(name="type", returns=UnknownType())
elif name == "range":
return FunctionType(name="range",
returns=ListType(NumType(), empty=False))
elif name == "dir":
return FunctionType(name="dir",
returns=ListType(StrType(), empty=False))
elif name == "max":
return FunctionType(name="max", returns='element')
elif name == "min":
return FunctionType(name="min", returns='element')
elif name == "zip":
return FunctionType(name="zip", returns=_builtin_zip)
elif name == "__import__":
return FunctionType(name="__import__", returns=ModuleType())
elif name == "globals":
return FunctionType(name="globals",
returns=DictType(keys=StrType(),
values=UnknownType(),
empty=False))
def BoolType_any(*x):
return BoolType()
def visit_Bool(self, node):
return BoolType()