Function extractSQLModelAST

src/ast/extract-python.ts:670–688

(
  filePath: string,
  content: string
)

    })) as SchemaField[],
    relations: (m.relations || []).map((r: any) =>
      `${r.name}: ${r.type === "many" ? "many" : "one"}(${r.target})`
    ),
    orm: "django" as const,
    confidence: "ast" as const,
  }));
}

/**
 * Extract SQLModel table models from a Python file using AST.
 * Detects class X(SQLModel, table=True) with typed field annotations.
 */
export async function extractSQLModelAST(
  filePath: string,
  content: string
): Promise<SchemaModel[] | null> {
  const result = await runPythonWithStdin(PYTHON_SQLMODEL_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,