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

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

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

比如項(xiàng)目初期，對(duì)數(shù)據(jù)庫(kù)的命名(db_ridingroad)沒(méi)有規(guī)劃好，
然后在下面創(chuàng)建了大量的表和寫(xiě)入了大量的數(shù)據(jù)，現(xiàn)在需要對(duì)數(shù)據(jù)庫(kù)的名字進(jìn)行重命名為(db_news_website)

常規(guī)思路

下面的方法步驟較為繁瑣

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

更快捷的方法

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

python rename_database.py old_db_name new_db_name

我們可以使用表重命名的方法，把表重命名到新的數(shù)據(jù)庫(kù)之下?；具壿嬋缦拢?/p>

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

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

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

        cursor.execute(sql, (old_name, ))
        results = cursor.fetchall()
        # 獲取表明，循環(huán)處理放到新的數(shù)據(jù)庫(kù)名下
        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ù)庫(kù)刪掉
        cursor.execute('drop database {}'.format(old_name))
    except Exception as ex:
        conn.rollback()
        print("rename_db Exception: {},{}".format(sql, ex))
        return False
    else:
        # 如果沒(méi)有發(fā)生異常，則提交事務(wù)
        conn.commit()
    finally:
        conn.close()
    return True

使用方法

1.安裝PyMySQL

pip install PyMySQL

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

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

3.切換到腳本所在目錄，執(zhí)行以下命令即可(數(shù)據(jù)無(wú)價(jià)，請(qǐng)先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ù)庫(kù)連接， 數(shù)據(jù)以字典結(jié)構(gòu)返回
    :param is_dict_cursor: 是否返回字典結(jié)構(gòu)的數(shù)據(jù)
    :param database: 默認(rèn)連接的數(shù)據(jù)庫(kù)
    :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ù)庫(kù)重命名
    :param old_name: 原來(lái)的數(shù)據(jù)庫(kù)名
    :param new_name: 新數(shù)據(jù)庫(kù)名
    :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ù)庫(kù)名
        cursor.execute('create database if not exists {}'.format(new_name))

        cursor.execute(sql, (old_name, ))
        results = cursor.fetchall()
        # 獲取表明，循環(huán)處理放到新的數(shù)據(jù)庫(kù)名下
        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ù)庫(kù)刪掉
        cursor.execute('drop database {}'.format(old_name))
    except Exception as ex:
        conn.rollback()
        print("rename_db Exception: {},{}".format(sql, ex))
        return False
    else:
        # 如果沒(méi)有發(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快速的對(duì)MySQL數(shù)據(jù)庫(kù)進(jìn)行重命名的詳細(xì)內(nèi)容，更多關(guān)于python 重命名MySQL數(shù)據(jù)庫(kù)的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: