運(yùn)用Python快速的對MySQL數(shù)據(jù)庫進(jìn)行重命名

更新時(shí)間：2021年06月07日 10:58:25 作者：一擼程猿

本文介紹了如何運(yùn)用Python快速的對現(xiàn)有的數(shù)據(jù)庫進(jìn)行重命名，有此需求的朋友可以參考下

常規(guī)思路

下面的方法步驟較為繁瑣

-- 數(shù)據(jù)庫備份
mysqldump –u [UserName] –p[Password] –R [DB_Name] > [DB_Name].sql
-- 創(chuàng)建新數(shù)據(jù)庫
create database [New_DB_Name];
-- 把備份的數(shù)據(jù)導(dǎo)入到新數(shù)據(jù)庫
mysql –u [UserName] –p[Password] [New_DB_Name] < [DB_Name].sql
-- 刪除舊數(shù)據(jù)庫
drop database [DB_Name];

更快捷的方法

只需要執(zhí)行下面這條命令即可

python rename_database.py old_db_name new_db_name

我們可以使用表重命名的方法，把表重命名到新的數(shù)據(jù)庫之下。基本邏輯如下：

創(chuàng)建新數(shù)據(jù)庫
獲取舊數(shù)據(jù)庫下所有的表名
把表重命名到新的數(shù)據(jù)庫名下
刪除舊數(shù)據(jù)庫

下面使用Python代碼去實(shí)現(xiàn)，主要代碼(完整代碼見文末)：

def rename_db(old_name, new_name):
    """
    數(shù)據(jù)庫重命名
    :param old_name: 原來的數(shù)據(jù)庫名
    :param new_name: 新數(shù)據(jù)庫名
    :return: 成功返回True， 失敗返回False
    """
    # 獲取所有的表名
    sql = """SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE table_schema=%s"""
    conn, cursor = context()
    try:
        # 創(chuàng)建新數(shù)據(jù)庫名
        cursor.execute('create database if not exists {}'.format(new_name))

        cursor.execute(sql, (old_name, ))
        results = cursor.fetchall()
        # 獲取表明，循環(huán)處理放到新的數(shù)據(jù)庫名下
        for r in results:
            tb = r['TABLE_NAME']
            rename_sql = """RENAME TABLE {}.{} to {}.{}""".format(old_name, tb, new_name, tb)
            cursor.execute(rename_sql)
        # 把舊數(shù)據(jù)庫刪掉
        cursor.execute('drop database {}'.format(old_name))
    except Exception as ex:
        conn.rollback()
        print("rename_db Exception: {},{}".format(sql, ex))
        return False
    else:
        # 如果沒有發(fā)生異常，則提交事務(wù)
        conn.commit()
    finally:
        conn.close()
    return True

使用方法

1.安裝PyMySQL

pip install PyMySQL

2.修改腳本中關(guān)于數(shù)據(jù)庫賬號配置信息部分

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

3.切換到腳本所在目錄，執(zhí)行以下命令即可(數(shù)據(jù)無價(jià)，請先mysqldump備份)

python rename_database.py old_db_name new_db_name

完整代碼如下：

import sys
import pymysql


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


def context(is_dict_cursor=True, database=MYSQL_DATABASE):
    """
    創(chuàng)建數(shù)據(jù)庫連接， 數(shù)據(jù)以字典結(jié)構(gòu)返回
    :param is_dict_cursor: 是否返回字典結(jié)構(gòu)的數(shù)據(jù)
    :param database: 默認(rèn)連接的數(shù)據(jù)庫
    :return: 返回一個(gè)連接和一個(gè)浮標(biāo)
    """
    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):
    """
    數(shù)據(jù)庫重命名
    :param old_name: 原來的數(shù)據(jù)庫名
    :param new_name: 新數(shù)據(jù)庫名
    :return: 成功返回True， 失敗返回False
    """
    # 獲取所有的表名
    sql = """SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE table_schema=%s"""
    conn, cursor = context()
    try:
        conn.begin()
        # 創(chuàng)建新數(shù)據(jù)庫名
        cursor.execute('create database if not exists {}'.format(new_name))

        cursor.execute(sql, (old_name, ))
        results = cursor.fetchall()
        # 獲取表明，循環(huán)處理放到新的數(shù)據(jù)庫名下
        for r in results:
            tb = r['TABLE_NAME']
            rename_sql = """RENAME TABLE {}.{} to {}.{}""".format(old_name, tb, new_name, tb)
            cursor.execute(rename_sql)
        # 把舊數(shù)據(jù)庫刪掉
        cursor.execute('drop database {}'.format(old_name))
    except Exception as ex:
        conn.rollback()
        print("rename_db Exception: {},{}".format(sql, ex))
        return False
    else:
        # 如果沒有發(fā)生異常，則提交事務(wù)
        conn.commit()
    finally:
        conn.close()
    return True


if __name__ == '__main__':
    old_db = sys.argv[1]
    new_db = sys.argv[2]
    rename_db(old_name=old_db, new_name=new_db)

以上就是運(yùn)用Python快速的對MySQL數(shù)據(jù)庫進(jìn)行重命名的詳細(xì)內(nèi)容，更多關(guān)于python 重命名MySQL數(shù)據(jù)庫的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: