feat(sdk): add Python generator

Implements Python SDK generator with two-level fallback:
1. Native openapi-python-client if installed
2. Docker openapitools/openapi-generator-cli as fallback

Includes tests following _Good convention for Available and Generate.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Snider 2026-01-29 01:15:57 +00:00
parent 3a7349ce5f
commit ea5b2fabaf
2 changed files with 134 additions and 0 deletions

View file

@ -0,0 +1,76 @@
package generators
import (
"context"
"fmt"
"os"
"os/exec"
"path/filepath"
)
// PythonGenerator generates Python SDKs from OpenAPI specs.
type PythonGenerator struct{}
// NewPythonGenerator creates a new Python generator.
func NewPythonGenerator() *PythonGenerator {
return &PythonGenerator{}
}
// Language returns the generator's target language identifier.
func (g *PythonGenerator) Language() string {
return "python"
}
// Available checks if generator dependencies are installed.
func (g *PythonGenerator) Available() bool {
_, err := exec.LookPath("openapi-python-client")
return err == nil
}
// Install returns instructions for installing the generator.
func (g *PythonGenerator) Install() string {
return "pip install openapi-python-client"
}
// Generate creates SDK from OpenAPI spec.
func (g *PythonGenerator) Generate(ctx context.Context, opts Options) error {
if err := os.MkdirAll(opts.OutputDir, 0755); err != nil {
return fmt.Errorf("python.Generate: failed to create output dir: %w", err)
}
if g.Available() {
return g.generateNative(ctx, opts)
}
return g.generateDocker(ctx, opts)
}
func (g *PythonGenerator) generateNative(ctx context.Context, opts Options) error {
parentDir := filepath.Dir(opts.OutputDir)
cmd := exec.CommandContext(ctx, "openapi-python-client", "generate",
"--path", opts.SpecPath,
"--output-path", opts.OutputDir,
)
cmd.Dir = parentDir
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
return cmd.Run()
}
func (g *PythonGenerator) generateDocker(ctx context.Context, opts Options) error {
specDir := filepath.Dir(opts.SpecPath)
specName := filepath.Base(opts.SpecPath)
cmd := exec.CommandContext(ctx, "docker", "run", "--rm",
"-v", specDir+":/spec",
"-v", opts.OutputDir+":/out",
"openapitools/openapi-generator-cli", "generate",
"-i", "/spec/"+specName,
"-g", "python",
"-o", "/out",
"--additional-properties=packageName="+opts.PackageName,
)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
return cmd.Run()
}

View file

@ -0,0 +1,58 @@
package generators
import (
"context"
"os"
"path/filepath"
"testing"
"time"
)
func TestPythonGenerator_Good_Available(t *testing.T) {
g := NewPythonGenerator()
// These should not panic
lang := g.Language()
if lang != "python" {
t.Errorf("expected language 'python', got '%s'", lang)
}
_ = g.Available()
install := g.Install()
if install == "" {
t.Error("expected non-empty install instructions")
}
}
func TestPythonGenerator_Good_Generate(t *testing.T) {
g := NewPythonGenerator()
if !g.Available() && !dockerAvailable() {
t.Skip("no Python generator available (neither native nor docker)")
}
// Create temp directories
tmpDir := t.TempDir()
specPath := createTestSpec(t, tmpDir)
outputDir := filepath.Join(tmpDir, "output")
opts := Options{
SpecPath: specPath,
OutputDir: outputDir,
PackageName: "testclient",
Version: "1.0.0",
}
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
defer cancel()
err := g.Generate(ctx, opts)
if err != nil {
t.Fatalf("Generate failed: %v", err)
}
// Verify output directory was created
if _, err := os.Stat(outputDir); os.IsNotExist(err) {
t.Error("output directory was not created")
}
}