写一段python的代码实现将mysql建表语句改成postgre建表语句
时间: 2023-06-28 22:09:30 浏览: 180
这里提供一个简单的代码示例,用于将MySQL建表语句转换为PostgreSQL建表语句。该代码适用于基本的表结构,如果存在更复杂的情况(如索引、外键等),则需要进行相应的修改。
```python
def mysql_to_postgres(sql):
# 将MySQL数据类型转换为PostgreSQL数据类型
data_type_mapping = {
'int': 'integer',
'varchar': 'character varying',
'text': 'text',
'datetime': 'timestamp',
'float': 'real',
}
# 将MySQL建表语句中的数据类型替换为PostgreSQL数据类型
for key in data_type_mapping:
sql = sql.replace(key, data_type_mapping[key])
# 将MySQL语句中的AUTO_INCREMENT替换为SERIAL
sql = sql.replace('AUTO_INCREMENT', 'SERIAL')
# 将MySQL语句中的BACKTICK替换为双引号
sql = sql.replace('`', '"')
# 将MySQL语句中的IF NOT EXISTS替换为IF NOT EXISTS
sql = sql.replace('IF NOT EXISTS', 'IF NOT EXISTS')
# 将MySQL语句中的PRIMARY KEY替换为CONSTRAINT
sql = sql.replace('PRIMARY KEY', 'CONSTRAINT')
# 将MySQL语句中的ENGINE=InnoDB替换为WITH(OIDS=TRUE)
sql = sql.replace('ENGINE=InnoDB', 'WITH(OIDS=TRUE)')
# 将MySQL语句中的DEFAULT CURRENT_TIMESTAMP替换为DEFAULT now()
sql = sql.replace('DEFAULT CURRENT_TIMESTAMP', 'DEFAULT now()')
# 将MySQL语句中的ON UPDATE CURRENT_TIMESTAMP替换为ON UPDATE now()
sql = sql.replace('ON UPDATE CURRENT_TIMESTAMP', 'ON UPDATE now()')
return sql
```
使用示例:
```python
mysql_sql = """
CREATE TABLE `users` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`name` varchar(255) NOT NULL,
`email` varchar(255) NOT NULL,
`created_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
"""
postgres_sql = mysql_to_postgres(mysql_sql)
print(postgres_sql)
```
输出结果:
```sql
CREATE TABLE "users" (
"id" integer NOT NULL SERIAL,
"name" character varying(255) NOT NULL,
"email" character varying(255) NOT NULL,
"created_at" timestamp NOT NULL DEFAULT now(),
CONSTRAINT "users_pkey" PRIMARY KEY ("id")
) WITH(OIDS=TRUE);
```
阅读全文