Function calculateComplexity

website/src/lib/parser.worker.ts:528–545

(node: any)

}

function calculateComplexity(node: any): number {
  const complexityNodes = new Set([
    "if_statement", "for_statement", "while_statement", "except_clause",
    "with_statement", "boolean_operator", "list_comprehension",
    "generator_expression", "case_clause", "conditional_expression",
    "binary_expression" // for logical ops in some langs
  ]);

  let count = 1;
  function traverse(n: any) {
    if (complexityNodes.has(n.type)) count++;
    for (let i = 0; i < n.childCount; i++) {
      traverse(n.child(i));
    }
  }
  traverse(node);
  return count;
}

function getPythonDocstring(node: any): string | null {
  // Python docstring is the first expression statement if it's a string