loop_parallel_ast.py 文件源码

def visit_Subscript(self, node):
        """
        tag variables that are followed by index as array
        """
        try:
            if node.value.id not in self.array_name:
                self.array_name.append(node.value.id)
        except:
            pass

        for field, value in ast.iter_fields(node):

            flag_cache = self.flag
            #index_cache = self.index_name
            #self.index_name = node.value

            if field == 'slice':
                self.flag = None
            elif field == 'ctx':
                self.flag = None

            if isinstance(value, list):
                for item in value:
                    if isinstance(item, ast.AST):
                        self.visit(item)
            elif isinstance(value, ast.AST):
                self.visit(value)

            self.flag = flag_cache
            #self.index_name = index_cache