Add table_prefix conf option

simo7 · simo7 · commit 1ac5932452c9 · 2024-08-02T15:44:36.000+02:00
diff --git a/Makefile b/Makefile
@@ -9,10 +9,10 @@ test: bin/sqlc-gen-python.wasm
 all: bin/sqlc-gen-python bin/sqlc-gen-python.wasm
 
 bin/sqlc-gen-python: bin go.mod go.sum $(wildcard **/*.go)
-	cd plugin && go build -o ../bin/sqlc-gen-python ./main.go
+	cd plugin/sqlc-gen-python && go build -o ../../bin/sqlc-gen-python ./main.go
 
 bin/sqlc-gen-python.wasm: bin/sqlc-gen-python
-	cd plugin && GOOS=wasip1 GOARCH=wasm go build -o ../bin/sqlc-gen-python.wasm main.go
+	cd plugin/sqlc-gen-python && GOOS=wasip1 GOARCH=wasm go build -o ../../bin/sqlc-gen-python.wasm main.go
 
 bin:
 	mkdir -p bin
diff --git a/README.md b/README.md
@@ -1,3 +1,7 @@
+## Build
+
+make bin/sqlc-gen-python
+
 ## Usage
 
 ```yaml
diff --git a/internal/config.go b/internal/config.go
@@ -9,4 +9,5 @@ type Config struct {
 	EmitPydanticModels          bool     `json:"emit_pydantic_models"`
 	QueryParameterLimit         *int32   `json:"query_parameter_limit"`
 	InflectionExcludeTableNames []string `json:"inflection_exclude_table_names"`
+	TablePrefix                 string   `json:"table_prefix"`
 }
diff --git a/internal/gen.go b/internal/gen.go
@@ -279,6 +279,9 @@ func buildModels(conf Config, req *plugin.GenerateRequest) []Struct {
 					Exclusions: conf.InflectionExcludeTableNames,
 				})
 			}
+			if conf.TablePrefix != "" {
+				structName = conf.TablePrefix + strings.ToUpper(structName[:1]) + structName[1:]
+			}
 			s := Struct{
 				Table:   plugin.Identifier{Schema: schema.Name, Name: table.Rel.Name},
 				Name:    modelName(structName, req.Settings),
diff --git a/plugin/sqlc-gen-python/main.go b/plugin/sqlc-gen-python/main.go

Original file line number	Diff line number	Diff line change
`@@ -9,4 +9,5 @@ type Config struct {`
`9`	`9`	EmitPydanticModels bool `json:"emit_pydantic_models"`
`10`	`10`	QueryParameterLimit *int32 `json:"query_parameter_limit"`
`11`	`11`	InflectionExcludeTableNames []string `json:"inflection_exclude_table_names"`
	`12`	+ TablePrefix string `json:"table_prefix"`
`12`	`13`	`}`