@vibe-dev-kit/cli/src/scanner/analyzers/python.js

Advanced Command-line toolkit that analyzes your codebase and deploys project-aware rules, memories, commands and agents to any AI coding assistant - VDK is the world's first Vibe Development Kit

/**
 * python.js
 *
 * Analyzer for Python code to detect naming conventions,
 * patterns, and commonly used libraries or frameworks.
 */

/**
 * Analyzes Python code to detect naming conventions and patterns
 * @param {string} content - Python code content
 * @param {string} filePath - Path to the file
 * @returns {Object} Analysis results
 */
export async function analyzePython(content, filePath) {
  const analysis = {
    variables: [],
    functions: [],
    classes: [],
    patterns: [],
  };

  try {
    // Parse function definitions
    const functionPattern = /def\s+([a-zA-Z_][a-zA-Z0-9_]*)/g;
    let match;
    while ((match = functionPattern.exec(content)) !== null) {
      if (match[1]) {
        analysis.functions.push(match[1]);
      }
    }

    // Parse class definitions
    const classPattern = /class\s+([a-zA-Z_][a-zA-Z0-9_]*)/g;
    while ((match = classPattern.exec(content)) !== null) {
      if (match[1]) {
        analysis.classes.push(match[1]);
      }
    }

    // Variable assignments (simplified approach)
    const variablePattern = /^\s*([a-zA-Z_][a-zA-Z0-9_]*)\s*=/gm;
    while ((match = variablePattern.exec(content)) !== null) {
      if (match[1] && !['if', 'for', 'while', 'def', 'class'].includes(match[1])) {
        analysis.variables.push(match[1]);
      }
    }

    // Detect imports and patterns
    detectPythonFrameworks(content, analysis);
    detectPythonLibraries(content, analysis);
    detectPythonPatterns(content, analysis);

    // Deduplicate
    analysis.variables = [...new Set(analysis.variables)];
    analysis.functions = [...new Set(analysis.functions)];
    analysis.classes = [...new Set(analysis.classes)];
    analysis.patterns = [...new Set(analysis.patterns)];

    return analysis;
  } catch (error) {
    console.error(`Error analyzing Python file: ${filePath}`);
    console.error(error.message);
    return analysis;
  }
}

/**
 * Detects Python frameworks from imports
 * @param {string} content - Python code content
 * @param {Object} analysis - Analysis results to update
 */
function detectPythonFrameworks(content, analysis) {
  // Check for Django patterns
  if (
    content.includes('import django') ||
    content.includes('from django') ||
    content.includes('urls.py') ||
    content.includes('models.py') ||
    content.includes('views.py')
  ) {
    analysis.patterns.push('Django');
  }

  // Check for Flask patterns
  if (content.includes('from flask import') || content.includes('Flask(__name__)')) {
    analysis.patterns.push('Flask');
  }

  // Check for FastAPI patterns
  if (content.includes('from fastapi import') || content.includes('FastAPI(')) {
    analysis.patterns.push('FastAPI');
  }

  // Check for Pyramid patterns
  if (content.includes('from pyramid.') || content.includes('config.add_route')) {
    analysis.patterns.push('Pyramid');
  }

  // Check for SQLAlchemy patterns
  if (content.includes('import sqlalchemy') || content.includes('from sqlalchemy')) {
    analysis.patterns.push('SQLAlchemy');
  }
}

/**
 * Detects Python libraries from imports
 * @param {string} content - Python code content
 * @param {Object} analysis - Analysis results to update
 */
function detectPythonLibraries(content, analysis) {
  // Common data science libraries
  if (content.includes('import pandas') || content.includes('from pandas')) {
    analysis.patterns.push('Pandas');
  }

  if (content.includes('import numpy') || content.includes('from numpy')) {
    analysis.patterns.push('NumPy');
  }

  if (content.includes('import matplotlib') || content.includes('from matplotlib')) {
    analysis.patterns.push('Matplotlib');
  }

  if (content.includes('import tensorflow') || content.includes('from tensorflow')) {
    analysis.patterns.push('TensorFlow');
  }

  if (content.includes('import torch') || content.includes('from torch')) {
    analysis.patterns.push('PyTorch');
  }

  if (content.includes('import sklearn') || content.includes('from sklearn')) {
    analysis.patterns.push('Scikit-learn');
  }

  // Web libraries
  if (content.includes('import requests') || content.includes('from requests')) {
    analysis.patterns.push('Requests');
  }

  if (content.includes('import aiohttp') || content.includes('from aiohttp')) {
    analysis.patterns.push('AIOHTTP');
  }

  // Testing libraries
  if (content.includes('import pytest') || content.includes('from pytest')) {
    analysis.patterns.push('Pytest');
  }

  if (content.includes('import unittest') || content.includes('from unittest')) {
    analysis.patterns.push('Unittest');
  }
}

/**
 * Detects Python coding patterns
 * @param {string} content - Python code content
 * @param {Object} analysis - Analysis results to update
 */
function detectPythonPatterns(content, analysis) {
  // Detect Python version via features
  if (content.includes('async def') || content.includes('await ')) {
    analysis.patterns.push('Async/Await (Python 3.5+)');
  }

  if (content.includes('f"') || content.includes("f'")) {
    analysis.patterns.push('F-strings (Python 3.6+)');
  }

  if (content.includes(': ') && content.includes(' -> ')) {
    analysis.patterns.push('Type Hints (Python 3.5+)');
  }

  // OOP patterns
  if (content.includes('class') && content.includes('__init__')) {
    analysis.patterns.push('Object-Oriented Programming');
  }

  // Functional patterns
  let functionalCount = 0;
  if (content.includes('lambda ')) functionalCount++;
  if (content.includes('map(')) functionalCount++;
  if (content.includes('filter(')) functionalCount++;
  if (content.includes('reduce(')) functionalCount++;
  if (
    content.includes('list comprehension') ||
    (content.includes('[') &&
      content.includes('for') &&
      content.includes('in') &&
      content.includes(']'))
  )
    functionalCount++;

  if (functionalCount >= 2) {
    analysis.patterns.push('Functional Programming');
  }

  // Decorator usage
  if (content.includes('@')) {
    analysis.patterns.push('Decorators');
  }

  // Context managers
  if (content.includes('with ') && content.includes('as ')) {
    analysis.patterns.push('Context Managers');
  }

  // Data classes (Python 3.7+)
  if (content.includes('@dataclass') || content.includes('from dataclasses import dataclass')) {
    analysis.patterns.push('Data Classes (Python 3.7+)');
  }
}