使用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()