Function analyze_blocks

packages/reactivity/src/scripts/ast-analyzer.py:245–429

(blocks)

def analyze_blocks(blocks):
    analysis = []

    for block in blocks:
        try:
            content = block.get("content", "")
            loc = count_lines_of_code(content)

            if block.get("type") == "code" or block.get("type") is None:
                block_defined, block_used, block_imported, block_packages = get_defined_used_variables(
                    block
                )
                block_defined_list = list(block_defined)
                block_defined_list.sort()
                block_used_list = list(block_used)
                block_used_list.sort()
                block_imported_list = list(block_imported)
                block_imported_list.sort()
                block_packages_list = list(block_packages)
                block_packages_list.sort()
                analysis.append(
                    {
                        "id": block["id"],
                        "definedVariables": block_defined_list,
                        "usedVariables": block_used_list,
                        "importedModules": block_imported_list,
                        "importedPackages": block_packages_list,
                        "linesOfCode": loc,
                    }
                )
            elif block["type"] == "sql":
                jinja_variables = extract_jinja_variables(block["content"])
                jinja_variables_list = list(jinja_variables)
                jinja_variables_list.sort()

                output_variables = []
                if (
                    "metadata" in block
                    and block["metadata"]
                    and "deepnote_variable_name" in block["metadata"]
                ):
                    output_variables = [block["metadata"]["deepnote_variable_name"]]

                analysis.append(
                    {
                        "id": block["id"],
                        "definedVariables": output_variables,
                        "usedVariables": jinja_variables_list,
                        "importedModules": [],
                        "linesOfCode": loc,
                    }
                )
            elif block["type"] == "button":
                output_variables = []
                if (
                    "metadata" in block
                    and block["metadata"]
                    and "deepnote_variable_name" in block["metadata"]