openpilot_comma/scripts/code_stats.py

#!/usr/bin/env python3
import os
import ast
import stat
import subprocess

fouts = {x.decode('utf-8') for x in subprocess.check_output(['git', 'ls-files']).strip().split()}

pyf = []
for d in ["cereal", "common", "scripts", "selfdrive", "tools"]:
  for root, _, files in os.walk(d):
    for f in files:
      if f.endswith(".py"):
        pyf.append(os.path.join(root, f))

imps: set[str] = set()

class Analyzer(ast.NodeVisitor):
  def visit_Import(self, node):
    for alias in node.names:
      imps.add(alias.name)
    self.generic_visit(node)

  def visit_ImportFrom(self, node):
    imps.add(node.module)
    self.generic_visit(node)

tlns = 0
carlns = 0
scriptlns = 0
testlns = 0
for f in sorted(pyf):
  if f not in fouts:
    continue
  xbit = bool(os.stat(f)[stat.ST_MODE] & stat.S_IXUSR)
  src = open(f).read()
  lns = len(src.split("\n"))
  tree = ast.parse(src)
  Analyzer().visit(tree)
  print(f"{lns:5d} {f} {xbit}")
  if 'test' in f:
    testlns += lns
  elif f.startswith(('tools/', 'scripts/', 'selfdrive/debug')):
    scriptlns += lns
  elif f.startswith('selfdrive/car'):
    carlns += lns
  else:
    tlns += lns

print(f"{tlns} lines of openpilot python")
print(f"{carlns} lines of car ports")
print(f"{scriptlns} lines of tools/scripts/debug")
print(f"{testlns} lines of tests")
#print(sorted(list(imps)))