Use StrEnum

simo7 · simo7 · commit ea89ec18596e · 2024-11-30T11:09:02.000+01:00
diff --git a/internal/gen.go b/internal/gen.go
@@ -954,7 +954,7 @@ func buildModelsTree(ctx *pyTmplCtx, i *importer) *pyast.Node {
 			Name: e.Name,
 			Bases: []*pyast.Node{
 				poet.Name("str"),
-				poet.Attribute(poet.Name("enum"), "Enum"),
+				poet.Attribute(poet.Name("enum"), "StrEnum"),
 			},
 		}
 		if e.Comment != "" {
diff --git a/internal/printer/printer_test.go b/internal/printer/printer_test.go
@@ -60,7 +60,7 @@ func TestPrinter(t *testing.T) {
 												Name: &ast.Name{Id: "enum"},
 											},
 										},
-										Attr: "Enum",
+										Attr: "StrEnum",
 									},
 								},
 							},

Original file line number	Diff line number	Diff line change
`@@ -954,7 +954,7 @@ func buildModelsTree(ctx pyTmplCtx, i importer) *pyast.Node {`
`954`	`954`	`Name: e.Name,`
`955`	`955`	`Bases: []*pyast.Node{`
`956`	`956`	`poet.Name("str"),`
`957`		`- poet.Attribute(poet.Name("enum"), "Enum"),`
	`957`	`+ poet.Attribute(poet.Name("enum"), "StrEnum"),`
`958`	`958`	`},`
`959`	`959`	`}`
`960`	`960`	`if e.Comment != "" {`