使用python的pymysql来管理MySQL和创建数据库

# 数据库地址
DB_ADDR = '127.0.0.1'
# 数据库用户名
USER_NAME = 'root'
# 数据库用户密码
USER_PASSWORD = 'root'
# 数据库名称
DB_NAME = 'hdu_rank'

con = pymysql.connect(host=DB_ADDR, user=USER_NAME, password=USER_PASSWORD) 
sql = "show databases like '%s'" % DB_NAME
with con.cursor() as cursor:
    cursor.execute(sql)
    row = cursor.fetchone()
    if row:
        is_continue = input("已经存在%s数据库,是否确定重建?\n注意原有的数据会丢失!\n[y/n]" % DB_NAME)
        is_continue = is_continue.lower()
        if is_continue == 'y':
            sql = 'DROP DATABASE %s' % DB_NAME
            cursor.execute(sql)
            con.commit()
        else:
            exit(1)

    sql = "CREATE DATABASE `%s` CHARACTER SET 'utf8mb4' COLLATE 'utf8mb4_unicode_ci';" % DB_NAME
    cursor.execute(sql)
    con.commit()
con.close()
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐