Function generateIfStatementFromElif

python/executor.py:324–370

(elif_node, else_nodes, traced)

def generateIfStatementFromElif(elif_node, else_nodes, traced):
    condition = generateAstExpression(elif_node["childSets"]["condition"][0])
    lineno = 1
    if 'meta' in elif_node and 'lineno' in elif_node['meta']:
        lineno = elif_node['meta']['lineno']

    if traced:
        key = ast.Name(id=SPLOOT_KEY, ctx=ast.Load())
        func = ast.Attribute(
            value=key, attr="logExpressionResultAndStartFrame", ctx=ast.Load()
        )
        args = [
            ast.Constant("PYTHON_ELIF_STATEMENT"),
            ast.Constant("condition"),
            condition,
        ]
        condition = ast.Call(func, args=args, keywords=[])

    statements = getStatementsFromBlock(elif_node["childSets"]["block"], traced)

    else_statements = []
    if len(else_nodes) != 0:
        else_statements = generateElifNestedChain(else_nodes, traced)

    if traced:
        key = ast.Name(id=SPLOOT_KEY, ctx=ast.Load())
        func = ast.Attribute(value=key, attr="endFrame", ctx=ast.Load())
        call_end_frame = ast.Call(func, args=[], keywords=[])

        # End the elif frame before starting the next else/elif block
        else_statements.insert(0, ast.Expr(call_end_frame, lineno=1, col_offset=0))

        key = ast.Name(id=SPLOOT_KEY, ctx=ast.Load())
        func = ast.Attribute(value=key, attr="startChildSet", ctx=ast.Load())
        args = [ast.Constant("block")]
        call_start_childset = ast.Call(func, args=args, keywords=[])

        key = ast.Name(id=SPLOOT_KEY, ctx=ast.Load())
        func = ast.Attribute(value=key, attr="endFrame", ctx=ast.Load())
        call_end_frame = ast.Call(func, args=[], keywords=[])

        statements.insert(0, ast.Expr(call_start_childset, lineno=1, col_offset=0))
        statements.append(ast.Expr(call_end_frame, lineno=1, col_offset=0))

    return [
        ast.If(condition, statements, else_statements, lineno=lineno),
    ]


def generateElifNestedChain(else_nodes, traced):