Python-Every-Day:0002——MySQL（python下的数据库应用）

在0001中我们产生了200组数据并存在了txt文本中，但现在我们要存储在以MySQL为代表的关系型数据库中

利用python对数据库的处理请移步MySQL及PyMySQL。

示例代码(源码移步Github)

提示：要想使用这个代码，请先把0001中的代码保存为coupon.py，然后运行即可，要先保证运行前建立名为coupon的数据库

# -*- coding: utf-8 -*-
"""
Created on Thu Jul 27 12:04:18 2017

@author: claus
"""
import pymysql
import coupon

if __name__ == "__main__":
    coupon_code = coupon.Generate_Coupon(200)
    coupon_code = tuple(coupon_code)
    db = pymysql.connect("localhost","root","sml197421","coupon")
    cursor = db.cursor()
    cursor.execute("DROP TABLE IF EXISTS COUPON")
    db.commit()
    cursor.execute("SELECT VERSION()")
    vid = cursor.fetchone()
    print("Database version : '%s'" % vid)
    sql = """CREATE TABLE COUPON(
    ID INT NOT NULL,
    CODE CHAR(10) NOT NULL)"""
    try:
        cursor.execute(sql)
    except:
        db.rollback()
    coupon_data=[]
    for i in range(len(coupon_code)):
        coupon_data.append([i+1,coupon_code[i]])
    cursor.executemany("INSERT INTO COUPON(ID,CODE) VALUES(%s,%s)",coupon_data)
    try:
        db.commit()
    except:
        db.rollback()
    db.close()