Function extractDjangoModelsAST

src/ast/extract-python.ts:644–664

(
  filePath: string,
  content: string
)

      type: f.type || "unknown",
      flags: f.flags || [],
    })) as SchemaField[],
    relations: (m.relations || []).map((r: any) => `${r.name}: ${r.target}`),
    orm: "sqlalchemy" as const,
    confidence: "ast" as const,
  }));
}

/**
 * Extract Django ORM models from a Python file using AST.
 * Handles models.Model subclasses with CharField, ForeignKey, etc.
 */
export async function extractDjangoModelsAST(
  filePath: string,
  content: string
): Promise<SchemaModel[] | null> {
  const result = await runPythonWithStdin(PYTHON_DJANGO_SCRIPT, content, filePath);
  if (!result || !Array.isArray(result) || result.length === 0) return null;

  return result.map((m: any) => ({
    name: m.name,
    fields: (m.fields || []).map((f: any) => ({
      name: f.name,
      type: f.type || "unknown",
      flags: f.flags || [],