0. 引言

今天发现一个有意思的竞赛，竞赛中使用了 sqlite 数据库。

由于个人更习惯 Oracle 数据库，所以将 sqlite 数据库迁移到了 Oracle 数据库。

此文章记录一下迁移时使用的 Python 代码。

1. 代码

import os
import sqlite3import oracledb
from dotenv import find_dotenv, load_dotenv_ = load_dotenv(find_dotenv())# 连接SQLite数据库
sqlite_conn = sqlite3.connect('./dataset/博金杯比赛数据.db')
sqlite_cursor = sqlite_conn.cursor()# 连接Oracle数据库
oracle_conn = oracledb.connect(user="username", password="password", dsn="host:port/service_name")
oracle_cursor = oracle_conn.cursor()# 获取SQLite数据库中的表名
sqlite_cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
tables = sqlite_cursor.fetchall()def map_sqlite_type_to_oracle(sqlite_type):# 根据需要映射SQLite数据类型到Oracle数据类型# 这是一个更详尽的映射示例mapping = {'INTEGER': 'NUMBER(10)','TEXT': 'VARCHAR2(255)','REAL': 'FLOAT','BLOB': 'BLOB','NUMERIC': 'NUMBER'}return mapping.get(sqlite_type, 'VARCHAR2(255)')  # 默认类型为 VARCHAR2(255)# 迭代每个表
for table in tables:table_name = table[0]print(f"Migrating table: {table_name}")# 获取表结构sqlite_cursor.execute(f"PRAGMA table_info({table_name});")columns = sqlite_cursor.fetchall()# 构建并执行Oracle表的创建语句column_defs = ', '.join(f"\"{column[1]}\" {map_sqlite_type_to_oracle(column[2])}" for column in columns)create_table_query = f"CREATE TABLE {table_name} ({column_defs})"print(f"{create_table_query=}")oracle_cursor.execute(create_table_query)# 测试时想Drop掉Table时使用# oracle_cursor.execute(f"DROP TABLE {table_name}")# 准备批量插入语句col_placeholders = ', '.join([':' + str(i + 1) for i in range(len(columns))])insert_query = f"INSERT INTO {table_name} VALUES ({col_placeholders})"# 迁移数据sqlite_cursor.execute(f"SELECT * FROM {table_name};")rows = sqlite_cursor.fetchall()oracle_cursor.executemany(insert_query, rows)oracle_conn.commit()sqlite_conn.close()
oracle_conn.close()

完结！