ai_code_reviewer.py

import ast
import pycodestyle


class CodeReviewer:
    def __init__(self):
        self.feedback = []

    def analyze_python_code(self, code):
        try:
            # Parse the Python code into an Abstract Syntax Tree (AST)
            tree = ast.parse(code)
        except SyntaxError as e:
            self.feedback.append(f"Syntax Error: {e}")
            return

        # Check for indentation errors and undefined variables
        self._check_indentation(tree)
        self._check_undefined_vars(tree)

        # Check code style using pycodestyle
        self._check_code_style(code)

        # Check code comments
        self._check_comments(code)

    def _check_indentation(self, tree):
        for node in ast.walk(tree):
            if isinstance(node, ast.FunctionDef):
                if node.body and not isinstance(node.body[0], ast.Expr):
                    self.feedback.append(
                        f"Function '{node.name}' should have a docstring or 'pass' statement.")
            elif isinstance(node, (ast.For, ast.While, ast.If, ast.With)):
                if not isinstance(node.body[0], ast.Expr):
                    self.feedback.append(
                        f"Indentation Error: Missing 'pass' statement for '{ast.dump(node)}'.")

    def _check_undefined_vars(self, tree):
        undefined_vars = set()
        for node in ast.walk(tree):
            if isinstance(node, ast.Name) and isinstance(node.ctx, ast.Store):
                undefined_vars.discard(node.id)
            elif isinstance(node, ast.Name) and isinstance(node.ctx, ast.Load):
                undefined_vars.add(node.id)

        for var in undefined_vars:
            self.feedback.append(f"Variable '{var}' is used but not defined.")

    def _check_code_style(self, code):
        style_guide = pycodestyle.StyleGuide()
        result = style_guide.check_code(code)
        if result.total_errors:
            self.feedback.append(
                "Code style issues found. Please check and fix them.")

    def _check_comments(self, code):
        lines = code.split('\n')
        for i, line in enumerate(lines):
            if line.strip().startswith('#'):
                # Check for empty comments or comments without space after '#'
                if len(line.strip()) == 1 or line.strip()[1] != ' ':
                    self.feedback.append(
                        f"Improve comment style in line {i + 1}: '{line.strip()}'")

    def get_feedback(self):
        return self.feedback


if __name__ == "__main__":
    # Example Python code to analyze
    python_code = """
    def add(a, b):
        result = a + b
        print(result)
    """

    code_reviewer = CodeReviewer()
    code_reviewer.analyze_python_code(python_code)

    feedback = code_reviewer.get_feedback()

    if feedback:
        print("Code Review Feedback:")
        for msg in feedback:
            print(f"- {msg}")
    else:
        print("No coding errors found. Code looks good!")