updated-god-object-detection.py

import ast

class AntiPatternDetector(ast.NodeVisitor):
    def __init__(self, threshold: int = 10):
        self.threshold = threshold
        self.god_objects = []

    def visit_ClassDef(self, node: ast.ClassDef) -> None:
        method_count = 0
        for child in ast.iter_child_nodes(node):
            if isinstance(child, (ast.FunctionDef, ast.AsyncFunctionDef)):
                method_count += 1
        if method_count > self.threshold:
            self.god_objects.append({
                'name': node.name,
                'lineno': node.lineno,
                'col_offset': node.col_offset
            })

def detect_anti_patterns(code: str, threshold: int = 10) -> list:
    try:
        tree = ast.parse(code)
        detector = AntiPatternDetector(threshold)
        detector.visit(tree)
        return detector.god_objects
    except Exception as e:
        print(f"Error: {e}")
        return []

def report_anti_patterns(anti_patterns: list) -> None:
    if anti_patterns:
        print("Detected anti-patterns:")
        for pattern in anti_patterns:
            print(f"- God Object: {pattern['name']} (line {pattern['lineno']}, column {pattern['col_offset']})")
    else:
        print("No anti-patterns detected.")

# example usage
code = """
class GodObject:
    def method1(self):
        pass
    def method2(self):
        pass
    def method3(self):
        pass
    def method4(self):
        pass
    def method5(self):
        pass
    def method6(self):
        pass
    def method7(self):
        pass
    def method8(self):
        pass
    def method9(self):
        pass
    def method10(self):
        pass
    def method11(self):
        pass
"""

anti_patterns = detect_anti_patterns(code)
report_anti_patterns(anti_patterns)