Method replace_string

aura/analyzers/python/rewrite_ast.py:289–330

Rewrites an expression `"some_string".replace("s", "a")` AST structure: :: aura.analyzers.python.nodes.Call( func=aura.analyzers.python.nodes.Attribute( source=aura.analyzers.python.nodes.String(value='some_string'),

(self, context)

            return

    def replace_string(self, context):  # TODO: add test
        """
        Rewrites an expression `"some_string".replace("s", "a")`
        AST structure:

        ::
            aura.analyzers.python.nodes.Call(
              func=aura.analyzers.python.nodes.Attribute(
                source=aura.analyzers.python.nodes.String(value='some_string'),
                attr='replace',
                action='Load'
              ),
              args=[
                aura.analyzers.python.nodes.String(value='s'),
                aura.analyzers.python.nodes.String(value='a')
              ],
              kwargs={}
            )
        """
        # We are looking for a function call
        if type(context.node) != Call:
            return
        # Function target is an attribute with `replace` attribute name
        func: Attribute = context.node.func
        if not (type(func) == Attribute and func.attr == "replace"):
            return

        replace_source = func.source
        # Source of the replace must be String
        if type(replace_source) != String:
            return

        # Check that replace args are also strings
        if len(context.node.args) < 2 or type(context.node.args[0]) != String or type(context.node.args[1]) != String:
            return

        # Rewrite the node by applying the replace operation
        # TODO: check docs if replace takes additional (kw)arguments
        data = str(replace_source).replace(str(context.node.args[0]), str(context.node.args[1]))
        new_node = String(value=data)
        new_node.enrich_from_previous(context.node)
        context.replace(new_node)

    def unary_op(self, context):
        if not type(context.node) in (dict, OrderedDict):