845 lines
24 KiB
Python
845 lines
24 KiB
Python
# encoding=utf8
|
||
"""芋道系统数据库迁移工具
|
||
|
||
Author: dhb52 (https://gitee.com/dhb52)
|
||
|
||
pip install simple-ddl-parser
|
||
"""
|
||
|
||
import argparse
|
||
import pathlib
|
||
import re
|
||
import time
|
||
from abc import ABC, abstractmethod
|
||
from typing import Dict, Generator, Optional, Tuple, Union
|
||
|
||
from simple_ddl_parser import DDLParser
|
||
|
||
PREAMBLE = """/*
|
||
Yskj-admin Database Transfer Tool
|
||
|
||
Source Server Type : MySQL
|
||
|
||
Target Server Type : {db_type}
|
||
|
||
Date: {date}
|
||
*/
|
||
|
||
"""
|
||
|
||
|
||
def load_and_clean(sql_file: str) -> str:
|
||
"""加载源 SQL 文件，并清理内容方便下一步 ddl 解析
|
||
|
||
Args:
|
||
sql_file (str): sql文件路径
|
||
|
||
Returns:
|
||
str: 清理后的sql文件内容
|
||
"""
|
||
REPLACE_PAIR_LIST = (
|
||
(" CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci ", " "),
|
||
(" KEY `", " INDEX `"),
|
||
("UNIQUE INDEX", "UNIQUE KEY"),
|
||
("b'0'", "'0'"),
|
||
("b'1'", "'1'"),
|
||
)
|
||
|
||
content = open(sql_file).read()
|
||
for replace_pair in REPLACE_PAIR_LIST:
|
||
content = content.replace(*replace_pair)
|
||
content = re.sub(r"ENGINE.*COMMENT", "COMMENT", content)
|
||
content = re.sub(r"ENGINE.*;", ";", content)
|
||
return content
|
||
|
||
|
||
class Convertor(ABC):
|
||
def __init__(self, src: str, db_type) -> None:
|
||
self.src = src
|
||
self.db_type = db_type
|
||
self.content = load_and_clean(self.src)
|
||
self.table_script_list = re.findall(r"CREATE TABLE [^;]*;", self.content)
|
||
|
||
@abstractmethod
|
||
def translate_type(self, type: str, size: Optional[Union[int, Tuple[int]]]) -> str:
|
||
"""字段类型转换
|
||
|
||
Args:
|
||
type (str): 字段类型
|
||
size (Optional[Union[int, Tuple[int]]]): 字段长度描述, 如varchar(255), decimal(10,2)
|
||
|
||
Returns:
|
||
str: 类型定义
|
||
"""
|
||
pass
|
||
|
||
@abstractmethod
|
||
def gen_create(self, table_ddl: Dict) -> str:
|
||
"""生成 create 脚本
|
||
|
||
Args:
|
||
table_ddl (Dict): 表DDL
|
||
|
||
Returns:
|
||
str: 生成脚本
|
||
"""
|
||
pass
|
||
|
||
@abstractmethod
|
||
def gen_pk(self, table_name: str) -> str:
|
||
"""生成主键定义
|
||
|
||
Args:
|
||
table_name (str): 表名
|
||
|
||
Returns:
|
||
str: 生成脚本
|
||
"""
|
||
pass
|
||
|
||
@abstractmethod
|
||
def gen_index(self, ddl: Dict) -> str:
|
||
"""生成索引定义
|
||
|
||
Args:
|
||
table_ddl (Dict): 表DDL
|
||
|
||
Returns:
|
||
str: 生成脚本
|
||
"""
|
||
pass
|
||
|
||
@abstractmethod
|
||
def gen_comment(self, table_sql: str, table_name: str) -> str:
|
||
"""生成字段/表注释
|
||
|
||
Args:
|
||
table_sql (str): 原始表SQL
|
||
table_name (str): 表名
|
||
|
||
Returns:
|
||
str: 生成脚本
|
||
"""
|
||
pass
|
||
|
||
@abstractmethod
|
||
def gen_insert(self, table_name: str) -> str:
|
||
"""生成 insert 语句块
|
||
|
||
Args:
|
||
table_name (str): 表名
|
||
|
||
Returns:
|
||
str: 生成脚本
|
||
"""
|
||
pass
|
||
|
||
def gen_dual(self) -> str:
|
||
"""生成虚拟 dual 表
|
||
|
||
Returns:
|
||
str: 生成脚本, 默认返回空脚本, 表示当前数据库无需手工创建
|
||
"""
|
||
return ""
|
||
|
||
@staticmethod
|
||
def inserts(table_name: str, script_content: str) -> Generator:
|
||
PREFIX = f"INSERT INTO `{table_name}`"
|
||
|
||
# 收集 `table_name` 对应的 insert 语句
|
||
for line in script_content.split("\n"):
|
||
if line.startswith(PREFIX):
|
||
head, tail = line.replace(PREFIX, "").split(" VALUES ", maxsplit=1)
|
||
head = head.strip().replace("`", "").lower()
|
||
tail = tail.strip().replace(r"\"", '"')
|
||
# tail = tail.replace("b'0'", "'0'").replace("b'1'", "'1'")
|
||
yield f"INSERT INTO {table_name.lower()} {head} VALUES {tail}"
|
||
|
||
@staticmethod
|
||
def index(ddl: Dict) -> Generator:
|
||
"""生成索引定义
|
||
|
||
Args:
|
||
ddl (Dict): 表DDL
|
||
|
||
Yields:
|
||
Generator[str]: create index 语句
|
||
"""
|
||
|
||
def generate_columns(columns):
|
||
keys = [
|
||
f"{col['name'].lower()}{' ' + col['order'].lower() if col['order'] != 'ASC' else ''}"
|
||
for col in columns[0]
|
||
]
|
||
return ", ".join(keys)
|
||
|
||
for no, index in enumerate(ddl["index"], 1):
|
||
columns = generate_columns(index["columns"])
|
||
table_name = ddl["table_name"].lower()
|
||
yield f"CREATE INDEX idx_{table_name}_{no:02d} ON {table_name} ({columns})"
|
||
|
||
@staticmethod
|
||
def filed_comments(table_sql: str) -> Generator:
|
||
for line in table_sql.split("\n"):
|
||
match = re.match(r"^`([^`]+)`.* COMMENT '([^']+)'", line.strip())
|
||
if match:
|
||
field = match.group(1)
|
||
comment_string = match.group(2).replace("\\n", "\n")
|
||
yield field, comment_string
|
||
|
||
def table_comment(self, table_sql: str) -> str:
|
||
match = re.search(r"COMMENT \= '([^']+)';", table_sql)
|
||
return match.group(1) if match else None
|
||
|
||
def print(self):
|
||
"""打印转换后的sql脚本到终端"""
|
||
print(
|
||
PREAMBLE.format(
|
||
db_type=self.db_type,
|
||
date=time.strftime("%Y-%m-%d %H:%M:%S"),
|
||
)
|
||
)
|
||
|
||
dual = self.gen_dual()
|
||
if dual:
|
||
print(
|
||
f"""-- ----------------------------
|
||
-- Table structure for dual
|
||
-- ----------------------------
|
||
{dual}
|
||
"""
|
||
)
|
||
|
||
error_scripts = []
|
||
for table_sql in self.table_script_list:
|
||
ddl = DDLParser(table_sql.replace("`", "")).run()
|
||
|
||
# 如果parse失败, 需要跟进
|
||
if len(ddl) == 0:
|
||
error_scripts.append(table_sql)
|
||
continue
|
||
|
||
table_ddl = ddl[0]
|
||
table_name = table_ddl["table_name"]
|
||
|
||
# 忽略 quartz 的内容
|
||
if table_name.lower().startswith("qrtz"):
|
||
continue
|
||
|
||
# 为每个表生成个5个基本部分
|
||
create = self.gen_create(table_ddl)
|
||
pk = self.gen_pk(table_name)
|
||
index = self.gen_index(table_ddl)
|
||
comment = self.gen_comment(table_sql, table_name)
|
||
inserts = self.gen_insert(table_name)
|
||
|
||
# 组合当前表的DDL脚本
|
||
script = f"""{create}
|
||
|
||
{pk}
|
||
|
||
{index}
|
||
|
||
{comment}
|
||
|
||
{inserts}
|
||
"""
|
||
|
||
# 清理
|
||
script = re.sub("\n{3,}", "\n\n", script).strip() + "\n"
|
||
|
||
print(script)
|
||
|
||
# 将parse失败的脚本打印出来
|
||
if error_scripts:
|
||
for script in error_scripts:
|
||
print(script)
|
||
|
||
|
||
class PostgreSQLConvertor(Convertor):
|
||
def __init__(self, src):
|
||
super().__init__(src, "PostgreSQL")
|
||
|
||
def translate_type(self, type: str, size: Optional[Union[int, Tuple[int]]]):
|
||
"""类型转换"""
|
||
|
||
type = type.lower()
|
||
|
||
if type == "varchar":
|
||
return f"varchar({size})"
|
||
if type == "int":
|
||
return "int4"
|
||
if type == "bigint" or type == "bigint unsigned":
|
||
return "int8"
|
||
if type == "datetime":
|
||
return "timestamp"
|
||
if type == "bit":
|
||
return "bool"
|
||
if type in ("tinyint", "smallint"):
|
||
return "int2"
|
||
if type == "text":
|
||
return "text"
|
||
if type in ("blob", "mediumblob"):
|
||
return "bytea"
|
||
if type == "decimal":
|
||
return (
|
||
f"numeric({','.join(str(s) for s in size)})" if len(size) else "numeric"
|
||
)
|
||
|
||
def gen_create(self, ddl: Dict) -> str:
|
||
"""生成 create"""
|
||
|
||
def _generate_column(col):
|
||
name = col["name"].lower()
|
||
if name == "deleted":
|
||
return "deleted int2 NOT NULL DEFAULT 0"
|
||
|
||
type = col["type"].lower()
|
||
full_type = self.translate_type(type, col["size"])
|
||
nullable = "NULL" if col["nullable"] else "NOT NULL"
|
||
default = f"DEFAULT {col['default']}" if col["default"] is not None else ""
|
||
return f"{name} {full_type} {nullable} {default}"
|
||
|
||
table_name = ddl["table_name"].lower()
|
||
columns = [f"{_generate_column(col).strip()}" for col in ddl["columns"]]
|
||
filed_def_list = ",\n ".join(columns)
|
||
script = f"""-- ----------------------------
|
||
-- Table structure for {table_name}
|
||
-- ----------------------------
|
||
DROP TABLE IF EXISTS {table_name};
|
||
CREATE TABLE {table_name} (
|
||
{filed_def_list}
|
||
);"""
|
||
|
||
return script
|
||
|
||
def gen_index(self, ddl: Dict) -> str:
|
||
return "\n".join(f"{script};" for script in self.index(ddl))
|
||
|
||
def gen_comment(self, table_sql: str, table_name: str) -> str:
|
||
"""生成字段及表的注释"""
|
||
|
||
script = ""
|
||
for field, comment_string in self.filed_comments(table_sql):
|
||
script += (
|
||
f"COMMENT ON COLUMN {table_name}.{field} IS '{comment_string}';" + "\n"
|
||
)
|
||
|
||
table_comment = self.table_comment(table_sql)
|
||
if table_comment:
|
||
script += f"COMMENT ON TABLE {table_name} IS '{table_comment}';\n"
|
||
|
||
return script
|
||
|
||
def gen_pk(self, table_name) -> str:
|
||
"""生成主键定义"""
|
||
return f"ALTER TABLE {table_name} ADD CONSTRAINT pk_{table_name} PRIMARY KEY (id);\n"
|
||
|
||
def gen_insert(self, table_name: str) -> str:
|
||
"""生成 insert 语句，以及根据最后的 insert id+1 生成 Sequence"""
|
||
|
||
inserts = list(Convertor.inserts(table_name, self.content))
|
||
## 生成 insert 脚本
|
||
script = ""
|
||
last_id = 0
|
||
if inserts:
|
||
inserts_lines = "\n".join(inserts)
|
||
script += f"""\n\n-- ----------------------------
|
||
-- Records of {table_name.lower()}
|
||
-- ----------------------------
|
||
-- @formatter:off
|
||
BEGIN;
|
||
{inserts_lines}
|
||
COMMIT;
|
||
-- @formatter:on"""
|
||
match = re.search(r"VALUES \((\d+),", inserts[-1])
|
||
if match:
|
||
last_id = int(match.group(1))
|
||
|
||
# 生成 Sequence
|
||
script += (
|
||
"\n\n"
|
||
+ f"""DROP SEQUENCE IF EXISTS {table_name}_seq;
|
||
CREATE SEQUENCE {table_name}_seq
|
||
START {last_id + 1};"""
|
||
)
|
||
|
||
return script
|
||
|
||
def gen_dual(self) -> str:
|
||
return """DROP TABLE IF EXISTS dual;
|
||
CREATE TABLE dual
|
||
(
|
||
id int2
|
||
);
|
||
|
||
COMMENT ON TABLE dual IS '数据库连接的表';
|
||
|
||
-- ----------------------------
|
||
-- Records of dual
|
||
-- ----------------------------
|
||
-- @formatter:off
|
||
INSERT INTO dual VALUES (1);
|
||
-- @formatter:on"""
|
||
|
||
|
||
class OracleConvertor(Convertor):
|
||
def __init__(self, src):
|
||
super().__init__(src, "Oracle")
|
||
|
||
def translate_type(self, type: str, size: Optional[Union[int, Tuple[int]]]):
|
||
"""类型转换"""
|
||
type = type.lower()
|
||
|
||
if type == "varchar":
|
||
return f"varchar2({size if size < 4000 else 4000})"
|
||
if type == "int":
|
||
return "number"
|
||
if type == "bigint" or type == "bigint unsigned":
|
||
return "number"
|
||
if type == "datetime":
|
||
return "date"
|
||
if type == "bit":
|
||
return "number(1,0)"
|
||
if type in ("tinyint", "smallint"):
|
||
return "smallint"
|
||
if type == "text":
|
||
return "clob"
|
||
if type in ("blob", "mediumblob"):
|
||
return "blob"
|
||
if type == "decimal":
|
||
return (
|
||
f"number({','.join(str(s) for s in size)})" if len(size) else "number"
|
||
)
|
||
|
||
def gen_create(self, ddl) -> str:
|
||
"""生成 CREATE 语句"""
|
||
|
||
def generate_column(col):
|
||
name = col["name"].lower()
|
||
if name == "deleted":
|
||
return "deleted number(1,0) DEFAULT 0 NOT NULL"
|
||
|
||
type = col["type"].lower()
|
||
full_type = self.translate_type(type, col["size"])
|
||
nullable = "NULL" if col["nullable"] else "NOT NULL"
|
||
default = f"DEFAULT {col['default']}" if col["default"] is not None else ""
|
||
# Oracle 中 size 不能作为字段名
|
||
field_name = '"size"' if name == "size" else name
|
||
# Oracle DEFAULT 定义在 NULLABLE 之前
|
||
return f"{field_name} {full_type} {default} {nullable}"
|
||
|
||
table_name = ddl["table_name"].lower()
|
||
columns = [f"{generate_column(col).strip()}" for col in ddl["columns"]]
|
||
field_def_list = ",\n ".join(columns)
|
||
script = f"""-- ----------------------------
|
||
-- Table structure for {table_name}
|
||
-- ----------------------------
|
||
CREATE TABLE {table_name} (
|
||
{field_def_list}
|
||
);"""
|
||
|
||
# oracle INSERT '' 不能通过 NOT NULL 校验
|
||
script = script.replace("DEFAULT '' NOT NULL", "DEFAULT '' NULL")
|
||
|
||
return script
|
||
|
||
def gen_index(self, ddl: Dict) -> str:
|
||
return "\n".join(f"{script};" for script in self.index(ddl))
|
||
|
||
def gen_comment(self, table_sql: str, table_name: str) -> str:
|
||
script = ""
|
||
for field, comment_string in self.filed_comments(table_sql):
|
||
script += (
|
||
f"COMMENT ON COLUMN {table_name}.{field} IS '{comment_string}';" + "\n"
|
||
)
|
||
|
||
table_comment = self.table_comment(table_sql)
|
||
if table_comment:
|
||
script += f"COMMENT ON TABLE {table_name} IS '{table_comment}';\n"
|
||
|
||
return script
|
||
|
||
def gen_pk(self, table_name: str) -> str:
|
||
"""生成主键定义"""
|
||
return f"ALTER TABLE {table_name} ADD CONSTRAINT pk_{table_name} PRIMARY KEY (id);\n"
|
||
|
||
def gen_index(self, ddl: Dict) -> str:
|
||
return "\n".join(f"{script};" for script in self.index(ddl))
|
||
|
||
def gen_insert(self, table_name: str) -> str:
|
||
"""拷贝 INSERT 语句"""
|
||
inserts = []
|
||
for insert_script in Convertor.inserts(table_name, self.content):
|
||
# 对日期数据添加 TO_DATE 转换
|
||
insert_script = re.sub(
|
||
r"('\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}')",
|
||
r"to_date(\g<1>, 'SYYYY-MM-DD HH24:MI:SS')",
|
||
insert_script,
|
||
)
|
||
inserts.append(insert_script)
|
||
|
||
## 生成 insert 脚本
|
||
script = ""
|
||
last_id = 0
|
||
if inserts:
|
||
inserts_lines = "\n".join(inserts)
|
||
script += f"""\n\n-- ----------------------------
|
||
-- Records of {table_name.lower()}
|
||
-- ----------------------------
|
||
-- @formatter:off
|
||
{inserts_lines}
|
||
COMMIT;
|
||
-- @formatter:on"""
|
||
match = re.search(r"VALUES \((\d+),", inserts[-1])
|
||
if match:
|
||
last_id = int(match.group(1))
|
||
|
||
# 生成 Sequence
|
||
script += f"""
|
||
|
||
CREATE SEQUENCE {table_name}_seq
|
||
START WITH {last_id + 1};"""
|
||
|
||
return script
|
||
|
||
|
||
class SQLServerConvertor(Convertor):
|
||
"""_summary_
|
||
|
||
Args:
|
||
Convertor (_type_): _description_
|
||
"""
|
||
|
||
def __init__(self, src):
|
||
super().__init__(src, "Microsoft SQL Server")
|
||
|
||
def translate_type(self, type: str, size: Optional[Union[int, Tuple[int]]]):
|
||
"""类型转换"""
|
||
|
||
type = type.lower()
|
||
|
||
if type == "varchar":
|
||
return f"nvarchar({size if size < 4000 else 4000})"
|
||
if type == "int":
|
||
return "int"
|
||
if type == "bigint" or type == "bigint unsigned":
|
||
return "bigint"
|
||
if type == "datetime":
|
||
return "datetime2"
|
||
if type == "bit":
|
||
return "varchar(1)"
|
||
if type in ("tinyint", "smallint"):
|
||
return "tinyint"
|
||
if type == "text":
|
||
return "nvarchar(max)"
|
||
if type in ("blob", "mediumblob"):
|
||
return "varbinary(max)"
|
||
if type == "decimal":
|
||
return (
|
||
f"numeric({','.join(str(s) for s in size)})" if len(size) else "numeric"
|
||
)
|
||
|
||
def gen_create(self, ddl: Dict) -> str:
|
||
"""生成 create"""
|
||
|
||
def _generate_column(col):
|
||
name = col["name"].lower()
|
||
if name == "id":
|
||
return "id bigint NOT NULL PRIMARY KEY IDENTITY"
|
||
if name == "deleted":
|
||
return "deleted bit DEFAULT 0 NOT NULL"
|
||
|
||
type = col["type"].lower()
|
||
full_type = self.translate_type(type, col["size"])
|
||
nullable = "NULL" if col["nullable"] else "NOT NULL"
|
||
default = f"DEFAULT {col['default']}" if col["default"] is not None else ""
|
||
return f"{name} {full_type} {default} {nullable}"
|
||
|
||
table_name = ddl["table_name"].lower()
|
||
columns = [f"{_generate_column(col).strip()}" for col in ddl["columns"]]
|
||
filed_def_list = ",\n ".join(columns)
|
||
script = f"""-- ----------------------------
|
||
-- Table structure for {table_name}
|
||
-- ----------------------------
|
||
DROP TABLE IF EXISTS {table_name}
|
||
GO
|
||
CREATE TABLE {table_name} (
|
||
{filed_def_list}
|
||
)
|
||
GO"""
|
||
|
||
return script
|
||
|
||
def gen_comment(self, table_sql: str, table_name: str) -> str:
|
||
"""生成字段及表的注释"""
|
||
|
||
script = ""
|
||
|
||
for field, comment_string in self.filed_comments(table_sql):
|
||
script += f"""EXEC sp_addextendedproperty
|
||
'MS_Description', N'{comment_string}',
|
||
'SCHEMA', N'dbo',
|
||
'TABLE', N'{table_name}',
|
||
'COLUMN', N'{field}'
|
||
GO
|
||
|
||
"""
|
||
|
||
table_comment = self.table_comment(table_sql)
|
||
if table_comment:
|
||
script += f"""EXEC sp_addextendedproperty
|
||
'MS_Description', N'{table_comment}',
|
||
'SCHEMA', N'dbo',
|
||
'TABLE', N'{table_name}'
|
||
GO
|
||
|
||
"""
|
||
return script
|
||
|
||
def gen_pk(self, table_name: str) -> str:
|
||
"""生成主键定义"""
|
||
return ""
|
||
|
||
def gen_index(self, ddl: Dict) -> str:
|
||
"""生成 index"""
|
||
return "\n".join(f"{script}\nGO" for script in self.index(ddl))
|
||
|
||
def gen_insert(self, table_name: str) -> str:
|
||
"""生成 insert 语句"""
|
||
|
||
# 收集 `table_name` 对应的 insert 语句
|
||
inserts = []
|
||
for insert_script in Convertor.inserts(table_name, self.content):
|
||
# SQLServer: 字符串前加N，hack，是否存在替换字符串内容的风险
|
||
insert_script = insert_script.replace(", '", ", N'").replace(
|
||
"VALUES ('", "VALUES (N')"
|
||
)
|
||
# 删除 insert 的结尾分号
|
||
insert_script = re.sub(";$", r"\nGO", insert_script)
|
||
inserts.append(insert_script)
|
||
|
||
## 生成 insert 脚本
|
||
script = ""
|
||
if inserts:
|
||
inserts_lines = "\n".join(inserts)
|
||
script += f"""\n\n-- ----------------------------
|
||
-- Records of {table_name.lower()}
|
||
-- ----------------------------
|
||
-- @formatter:off
|
||
BEGIN TRANSACTION
|
||
GO
|
||
SET IDENTITY_INSERT {table_name.lower()} ON
|
||
GO
|
||
{inserts_lines}
|
||
SET IDENTITY_INSERT {table_name.lower()} OFF
|
||
GO
|
||
COMMIT
|
||
GO
|
||
-- @formatter:on"""
|
||
|
||
return script
|
||
|
||
def gen_dual(self) -> str:
|
||
return """DROP TABLE IF EXISTS dual
|
||
GO
|
||
CREATE TABLE dual
|
||
(
|
||
id int
|
||
)
|
||
GO
|
||
|
||
EXEC sp_addextendedproperty
|
||
'MS_Description', N'数据库连接的表',
|
||
'SCHEMA', N'dbo',
|
||
'TABLE', N'dual'
|
||
GO
|
||
|
||
-- ----------------------------
|
||
-- Records of dual
|
||
-- ----------------------------
|
||
-- @formatter:off
|
||
INSERT INTO dual VALUES (1)
|
||
GO
|
||
-- @formatter:on"""
|
||
|
||
|
||
class DM8Convertor(Convertor):
|
||
def __init__(self, src):
|
||
super().__init__(src, "DM8")
|
||
|
||
def translate_type(self, type: str, size: Optional[Union[int, Tuple[int]]]):
|
||
"""类型转换"""
|
||
type = type.lower()
|
||
|
||
if type == "varchar":
|
||
return f"varchar({size})"
|
||
if type == "int":
|
||
return "int"
|
||
if type == "bigint" or type == "bigint unsigned":
|
||
return "bigint"
|
||
if type == "datetime":
|
||
return "datetime"
|
||
if type == "bit":
|
||
return "bit"
|
||
if type in ("tinyint", "smallint"):
|
||
return "smallint"
|
||
if type == "text":
|
||
return "text"
|
||
if type == "blob":
|
||
return "blob"
|
||
if type == "mediumblob":
|
||
return "varchar(10240)"
|
||
if type == "decimal":
|
||
return (
|
||
f"decimal({','.join(str(s) for s in size)})" if len(size) else "decimal"
|
||
)
|
||
|
||
def gen_create(self, ddl) -> str:
|
||
"""生成 CREATE 语句"""
|
||
|
||
def generate_column(col):
|
||
name = col["name"].lower()
|
||
if name == "id":
|
||
return "id bigint NOT NULL PRIMARY KEY IDENTITY"
|
||
|
||
type = col["type"].lower()
|
||
full_type = self.translate_type(type, col["size"])
|
||
nullable = "NULL" if col["nullable"] else "NOT NULL"
|
||
default = f"DEFAULT {col['default']}" if col["default"] is not None else ""
|
||
return f"{name} {full_type} {default} {nullable}"
|
||
|
||
table_name = ddl["table_name"].lower()
|
||
columns = [f"{generate_column(col).strip()}" for col in ddl["columns"]]
|
||
field_def_list = ",\n ".join(columns)
|
||
script = f"""-- ----------------------------
|
||
-- Table structure for {table_name}
|
||
-- ----------------------------
|
||
CREATE TABLE {table_name} (
|
||
{field_def_list}
|
||
);"""
|
||
|
||
# oracle INSERT '' 不能通过 NOT NULL 校验
|
||
script = script.replace("DEFAULT '' NOT NULL", "DEFAULT '' NULL")
|
||
|
||
return script
|
||
|
||
def gen_index(self, ddl: Dict) -> str:
|
||
return "\n".join(f"{script};" for script in self.index(ddl))
|
||
|
||
def gen_comment(self, table_sql: str, table_name: str) -> str:
|
||
script = ""
|
||
for field, comment_string in self.filed_comments(table_sql):
|
||
script += (
|
||
f"COMMENT ON COLUMN {table_name}.{field} IS '{comment_string}';" + "\n"
|
||
)
|
||
|
||
table_comment = self.table_comment(table_sql)
|
||
if table_comment:
|
||
script += f"COMMENT ON TABLE {table_name} IS '{table_comment}';\n"
|
||
|
||
return script
|
||
|
||
def gen_pk(self, table_name: str) -> str:
|
||
"""生成主键定义"""
|
||
return ""
|
||
|
||
def gen_index(self, ddl: Dict) -> str:
|
||
return "\n".join(f"{script};" for script in self.index(ddl))
|
||
|
||
def gen_insert(self, table_name: str) -> str:
|
||
"""拷贝 INSERT 语句"""
|
||
inserts = list(Convertor.inserts(table_name, self.content))
|
||
|
||
## 生成 insert 脚本
|
||
script = ""
|
||
if inserts:
|
||
inserts_lines = "\n".join(inserts)
|
||
script += f"""\n\n-- ----------------------------
|
||
-- Records of {table_name.lower()}
|
||
-- ----------------------------
|
||
-- @formatter:off
|
||
SET IDENTITY_INSERT {table_name.lower()} ON;
|
||
{inserts_lines}
|
||
COMMIT;
|
||
SET IDENTITY_INSERT {table_name.lower()} OFF;
|
||
-- @formatter:on"""
|
||
|
||
return script
|
||
|
||
|
||
class KingbaseConvertor(PostgreSQLConvertor):
|
||
def __init__(self, src):
|
||
super().__init__(src)
|
||
self.db_type = "Kingbase"
|
||
|
||
def gen_create(self, ddl: Dict) -> str:
|
||
"""生成 create"""
|
||
|
||
def _generate_column(col):
|
||
name = col["name"].lower()
|
||
if name == "deleted":
|
||
return "deleted int2 NOT NULL DEFAULT 0"
|
||
|
||
type = col["type"].lower()
|
||
full_type = self.translate_type(type, col["size"])
|
||
nullable = "NULL" if col["nullable"] else "NOT NULL"
|
||
default = f"DEFAULT {col['default']}" if col["default"] is not None else ""
|
||
return f"{name} {full_type} {nullable} {default}"
|
||
|
||
table_name = ddl["table_name"].lower()
|
||
columns = [f"{_generate_column(col).strip()}" for col in ddl["columns"]]
|
||
filed_def_list = ",\n ".join(columns)
|
||
script = f"""-- ----------------------------
|
||
-- Table structure for {table_name}
|
||
-- ----------------------------
|
||
DROP TABLE IF EXISTS {table_name};
|
||
CREATE TABLE {table_name} (
|
||
{filed_def_list}
|
||
);"""
|
||
|
||
# Kingbase INSERT '' 不能通过 NOT NULL 校验
|
||
script = script.replace("NOT NULL DEFAULT ''", "NULL DEFAULT ''")
|
||
|
||
return script
|
||
|
||
|
||
class OpengaussConvertor(KingbaseConvertor):
|
||
def __init__(self, src):
|
||
super().__init__(src)
|
||
self.db_type = "OpenGauss"
|
||
|
||
|
||
def main():
|
||
parser = argparse.ArgumentParser(description="芋道系统数据库转换工具")
|
||
parser.add_argument(
|
||
"type",
|
||
type=str,
|
||
help="目标数据库类型",
|
||
choices=["postgres", "oracle", "sqlserver", "dm8", "kingbase", "opengauss"],
|
||
)
|
||
args = parser.parse_args()
|
||
|
||
sql_file = pathlib.Path("../mysql/ruoyi-vue-pro.sql").resolve().as_posix()
|
||
convertor = None
|
||
if args.type == "postgres":
|
||
convertor = PostgreSQLConvertor(sql_file)
|
||
elif args.type == "oracle":
|
||
convertor = OracleConvertor(sql_file)
|
||
elif args.type == "sqlserver":
|
||
convertor = SQLServerConvertor(sql_file)
|
||
elif args.type == "dm8":
|
||
convertor = DM8Convertor(sql_file)
|
||
elif args.type == "kingbase":
|
||
convertor = KingbaseConvertor(sql_file)
|
||
elif args.type == "opengauss":
|
||
convertor = OpengaussConvertor(sql_file)
|
||
else:
|
||
raise NotImplementedError(f"不支持目标数据库类型: {args.type}")
|
||
|
||
convertor.print()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|