Python3:使用pymysql管理和創建數據庫

使用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()
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章