【Python学习成果】批量删除MySQL库（正则表达式）

import pymysql
import time
import re

try:
    # 输入数据库信息
    host = input("请输入数据库地址：")
    port = int(input("请输入数据库端口："))
    user = input("请输入数据库用户：")
    passwd = input("请输入数据库密码：")

    config = {
        'host': host,  # 数据库地址
        'port': port,  # 数据库端口
        'user': user,  # 数据库用户名
        'passwd': passwd,  # 数据库密码
        'charset': 'utf8',  # 编码方式
    }
    # 连接数据库
    mydb = pymysql.connect(local_infile=1, **config)

    # 创建游标
    mycursor = mydb.cursor()

    # 创建名为DE的数据库，并切换到DE数据库
    mycursor.execute("CREATE DATABASE IF NOT EXISTS DE")
    mycursor.execute("USE DE")

    # 创建名为DE的表格，并指定字段和数据类型
    mycursor.execute("CREATE TABLE IF NOT EXISTS DE "
                     "(id INT AUTO_INCREMENT PRIMARY KEY COMMENT '序号', "
                     "dbname VARCHAR(255) COMMENT '库名清单', "
                     "record_time TIMESTAMP COMMENT '记录时间', "
                     "current_status INT COMMENT '状态标识', "
                     "delete_time TIMESTAMP COMMENT '记录的删除时间', "
                     "rebuild_time TIMESTAMP COMMENT '记录的重建时间') charset=utf8 comment '数据库名信息表'")
    print("正在获取数据库信息，请稍等！")

    # 查询数据库列表
    mycursor.execute("SHOW DATABASES")
    databases = [db[0] for db in mycursor]

    # 查询DE表格里的库名列表
    mycursor.execute("SELECT dbname FROM DE")
    de_dbnames = [db[0] for db in mycursor]

    # 构建需要添加到DE表格里的数据列表
    insert_data = []

    for db in databases:
        # 判断当前数据库名是否在DE表格中
        if db not in de_dbnames:
            # 添加数据，包括库名、记录时间、当前状态、最近删除时间、最近重建时间
            insert_data.append((db,
                                time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()),
                                1, None, None))
        else:
            # 查询当前数据库的状态
            mycursor.execute(f"SELECT current_status FROM DE WHERE dbname = '{db}'")
            current_status = mycursor.fetchone()[0]
            # 如果状态为0，则更新状态，记录删除时间，否则不进行操作
            if current_status == 0:
                mycursor.execute(f"UPDATE DE SET current_status = 1, "
                                 f"rebuild_time = '{time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())}' "
                                 f"WHERE dbname = '{db}'")
                mydb.commit()
            # 如果状态为1，则不进行操作
            else:
                pass
    for db in de_dbnames:
        if db not in databases:
            mycursor.execute(f"SELECT current_status FROM DE WHERE dbname = '{db}'")
            current_status = mycursor.fetchone()[0]
            if current_status == 1:
                mycursor.execute(f"UPDATE DE SET current_status = 0, "
                                 f"delete_time = '{time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())}' "
                                 f"WHERE dbname = '{db}'")
                mydb.commit()
            else:
                pass

    # 使用 executemany() 方法批量插入数据
    if insert_data:
        sql = "INSERT INTO DE (" \
              "dbname, record_time, current_status, delete_time, rebuild_time" \
              ") values (%s, %s, %s, %s, %s)"
        mycursor.executemany(sql, insert_data)
        mydb.commit()

    print("数据库信息获取完成，请输入想要删除的数据库名称规则。")
    print("例如:我想删除的数据库里面有23456多个数字相连,那我就输入2345")
    print("支持使用正则表达式输入")

    # 查询DE表格里的库名列表，并输出
    mycursor.execute("SELECT dbname FROM DE")
    de_dbnames = [db[0] for db in mycursor]
    # print(de_dbnames)
    # 输入要删除的库的名称规则
    while True:
        guize = input("录入要删除的库名称规则：")
        print("名称输入3个字符以上,允许输入正则表达式。")
        if len(guize) < 3:
            print("输入的规则长度小于3,请重新输入。")  
        else:
            break

    print("即将批量删除的数据库清单如下：")
    for de_dbname in de_dbnames:
        # 如果库名是包含了guize函数的内容，则删除该数据库，并跳过后续操作
        if re.match(r'(.*)%s(.*?).*' % guize, de_dbname):
            print(re.match(r'(.*)%s(.*?).*' % guize, de_dbname).group())
            mycursor.execute(f"DROP DATABASE IF EXISTS {de_dbname}")
            mycursor.execute(f"UPDATE DE SET current_status = 0, "
                             f"delete_time = '{time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())}' "
                             f"WHERE dbname = '{de_dbname}'")
            continue
    

except Exception as e:
    print("Error:" + str(e))
finally:
    mycursor.close()
    mydb.close()
    print("批量删除完成！")