python重命名_MySQL数据库重命名之Python脚本-白红宇

python重命名_MySQL数据库重命名之Python脚本

阅读量：6377 次

发布时间：2019-06-23

本文共 3919 字，大约阅读时间需要 13 分钟。

对数据库的表进行重命名：

RENAME TABLE old_table TO new_table;

但是MySQL并没有直接支持对数据库进行重命名

那么如何运用Python快速的对现有的数据库进行重命名呢？

比如项目初期，对数据库的命名(db_ridingroad)没有规划好，然后在下面创建了大量的表和写入了大量的数据，现在需要对数据库的名字进行重命名为(db_news_website)

常规思路

下面的方法较为繁琐

-- 数据库备份mysqldump –u [UserName] –p[Password] –R [DB_Name] > [DB_Name].sql-- 创建新数据库create database [New_DB_Name];-- 把备份的数据导入到新数据库mysql –u [UserName] –p[Password] [New_DB_Name] < [DB_Name].sql-- 删除旧数据库drop database [DB_Name];

更快捷的方法

只需要执行下面这条命令即可

python rename_database.py old_db_name new_db_name

我们可以使用表重命名的方法，把表重命名到新的数据库之下。基本逻辑如下：

1. 创建新数据库2. 获取旧数据库下所有的表名3. 把表重命名到新的数据库名下4. 删除旧数据库

下面使用Python代码去实现，主要代码(完整代码见文末)：

def rename_db(old_name, new_name):    """    数据库重命名    :param old_name: 原来的数据库名    :param new_name: 新数据库名    :return: 成功返回True， 失败返回False    """    # 获取所有的表名    sql = """SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE table_schema=%s"""    conn, cursor = context()    try:        # 创建新数据库名        cursor.execute('create database if not exists {}'.format(new_name))        cursor.execute(sql, (old_name, ))        results = cursor.fetchall()        # 获取表明，循环处理放到新的数据库名下        for r in results:            tb = r['TABLE_NAME']            rename_sql = """RENAME TABLE {}.{} to {}.{}""".format(old_name, tb, new_name, tb)            cursor.execute(rename_sql)        # 把旧数据库删掉        cursor.execute('drop database {}'.format(old_name))    except Exception as ex:        conn.rollback()        print("rename_db Exception: {},{}".format(sql, ex))        return False    else:        # 如果没有发生异常，则提交事务        conn.commit()    finally:        conn.close()    return True

使用方法

安装PyMySQL

pip install PyMySQL

修改脚本中关于数据库账号配置信息部分

MYSQL_HOST = '127.0.0.1'MYSQL_PORT = 3306MYSQL_USER = 'ridingroad'MYSQL_PASSWORD = 'xxxxyyyy'MYSQL_DATABASE = 'db_ridingroad'

切换到脚本所在目录，执行以下命令即可(数据无价，请先mysqldump备份)

python rename_database.py old_db_name new_db_name

完整代码如下：

import sysimport pymysqlMYSQL_HOST = '127.0.0.1'MYSQL_PORT = 3306MYSQL_USER = 'ridingroad'MYSQL_PASSWORD = 'xxxxyyyy'MYSQL_DATABASE = 'db_ridingroad'MYSQL_CHARSET = 'utf8'def context(is_dict_cursor=True, database=MYSQL_DATABASE):    """    创建数据库连接， 数据以字典结构返回    :param is_dict_cursor: 是否返回字典结构的数据    :param database: 默认连接的数据库    :return: 返回一个连接和一个浮标    """    try:        config = {            'host': MYSQL_HOST,            'port': MYSQL_PORT,            'user': MYSQL_USER,            'password': MYSQL_PASSWORD,            'database': database,            'charset': MYSQL_CHARSET,        }        conn = pymysql.connect(**config)        if is_dict_cursor:            cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)        else:            cursor = conn.cursor()        return conn, cursor    except Exception as ex:        print("connect database failed, {},{}".format(400, ex))        raise Exception({'code': 400, 'msg': ex})def rename_db(old_name, new_name):    """    数据库重命名    :param old_name: 原来的数据库名    :param new_name: 新数据库名    :return: 成功返回True， 失败返回False    """    # 获取所有的表名    sql = """SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE table_schema=%s"""    conn, cursor = context()    try:        conn.begin()        # 创建新数据库名        cursor.execute('create database if not exists {}'.format(new_name))        cursor.execute(sql, (old_name, ))        results = cursor.fetchall()        # 获取表明，循环处理放到新的数据库名下        for r in results:            tb = r['TABLE_NAME']            rename_sql = """RENAME TABLE {}.{} to {}.{}""".format(old_name, tb, new_name, tb)            cursor.execute(rename_sql)        # 把旧数据库删掉        cursor.execute('drop database {}'.format(old_name))    except Exception as ex:        conn.rollback()        print("rename_db Exception: {},{}".format(sql, ex))        return False    else:        # 如果没有发生异常，则提交事务        conn.commit()    finally:        conn.close()    return Trueif __name__ == '__main__':    old_db = sys.argv[1]    new_db = sys.argv[2]    rename_db(old_name=old_db, new_name=new_db)

转载地址：http://klvqa.baihongyu.com/

你可能感兴趣的文章

实现了所有主流APP的分类切换效果，可快速接入，灵活扩展（swift）

查看>>

微信自动跳转浏览器打开APP（APK）下载链接