72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package commontypes
|
|
|
|
// PythonTypeMap maps PostgreSQL types to Python types
|
|
var PythonTypeMap = map[string]string{
|
|
// Integer types
|
|
"integer": "int",
|
|
"int": "int",
|
|
"int4": "int",
|
|
"smallint": "int",
|
|
"int2": "int",
|
|
"bigint": "int",
|
|
"int8": "int",
|
|
"serial": "int",
|
|
"bigserial": "int",
|
|
"smallserial": "int",
|
|
|
|
// String types
|
|
"text": "str",
|
|
"varchar": "str",
|
|
"char": "str",
|
|
"character": "str",
|
|
"citext": "str",
|
|
"bpchar": "str",
|
|
"uuid": "UUID",
|
|
|
|
// Boolean
|
|
"boolean": "bool",
|
|
"bool": "bool",
|
|
|
|
// Float types
|
|
"real": "float",
|
|
"float4": "float",
|
|
"double precision": "float",
|
|
"float8": "float",
|
|
"numeric": "Decimal",
|
|
"decimal": "Decimal",
|
|
|
|
// Date/Time types
|
|
"timestamp": "datetime",
|
|
"timestamp without time zone": "datetime",
|
|
"timestamp with time zone": "datetime",
|
|
"timestamptz": "datetime",
|
|
"date": "date",
|
|
"time": "time",
|
|
"time without time zone": "time",
|
|
"time with time zone": "time",
|
|
"timetz": "time",
|
|
|
|
// Binary
|
|
"bytea": "bytes",
|
|
|
|
// JSON
|
|
"json": "dict",
|
|
"jsonb": "dict",
|
|
|
|
// Array
|
|
"array": "list",
|
|
}
|
|
|
|
// SQLToPython converts SQL types to Python types
|
|
func SQLToPython(sqlType string) string {
|
|
baseType := ExtractBaseType(sqlType)
|
|
|
|
pyType, ok := PythonTypeMap[baseType]
|
|
if !ok {
|
|
pyType = "Any"
|
|
}
|
|
|
|
// Python uses Optional[Type] for nullable, but we return the base type
|
|
return pyType
|
|
}
|