生成优惠券,并将优惠券存入Mysql

#coding:utf-8

import random
import string
import MySQLdb


def gen_charint(filename, width =4, num=5):

    f = open(filename, 'wb')
    charint = string.digits + string.letters
    for i in range(num):
        verify = [random.choice(charint) for j in range(width)]
        verify = ''.join(verify) + '\n'
        f.write(verify)
    f.close()

def store_mysql(filepath):

    conn = MySQLdb.connect(
            host="localhost",
            port=3306,
            user='root',
            passwd='root',
            db='test',
            charset='utf8'
        )

    cursor = conn.cursor()
    # 判断表是否存在
    cursor.execute('show tables in test')
    tables = cursor.fetchall()
    findtables = False
    for table in tables:
        if 'coupon' in table:
            findtables = True
    # 下面的sql语句中的符号为反引号
    # 由于刚开始设置code的字段长度偏小,报错后,调整这里的长度不起作用,查看数据库并未相应改变,需要手动更改数据库该字段长度
    if not findtables:
        cursor.execute('''
        CREATE TABLE `coupon24`(
        `id` INT NOT NULL AUTO_INCREMENT,
        `coupon` VARCHAR(66) NOT NULL,
        PRIMARY KEY(`id`));
        ''')

    f = open(filepath, 'rb')
    for line in f.readlines():
        code = line.split()
        cursor.execute('insert into coupon24 (coupon) VALUES (%s)', code)

    conn.commit()
    cursor.close()
    conn.close()

if __name__ == '__main__':
    
    filename = 'result24.txt'
    width = 4
    num = 11
    gen_charint(filename, width, num)
    store_mysql(filename)

注意点:

1、用列表生成式生成随机字数和数字,并通过join连接,并注意加上换行符

2、创建表时,表名和字段名均要用反引号,即TAB键上方的那个按键

posted @ 2017-10-24 11:31  milian0711  阅读(1475)  评论(0编辑  收藏  举报