python生成激活码,并保存在MySQL数据库中

# coding=utf-8
import string, random
import MySQLdb
#取所有的字母(包括大小写)和数字
field = string.letters + string.digits
#得到一个随机字符串(长度为4)
def getRandom():
    return "".join(random.sample(field,4))
#连接函数,将字符串用‘-’连接起来
def concatenate(group):
    return "-".join([getRandom() for i in range(group)])
#将所有生成的字符串连接起来,组数为(4)
def generate(n):
    return [concatenate(4) for i in range(n)]

def createTable():
    db = MySQLdb.connect(host='localhost', user='root', passwd='jh13617967684', db='lyylll')
    cursor = db.cursor()
    sql = '''
    create table if not exists activationCode(id int auto_increment primary key not null, code varchar(32) not null);
    '''
    cursor.execute(sql)
    db.close()
def insertCode():
    db = MySQLdb.connect(host='localhost', user='root', passwd='xxx', db='lyylll')
    value=generate(200)
    for i in value:
        cursor = db.cursor()
        sql = "INSERT INTO activationCode(id, \
                   code) \
                   VALUES ('%s', '%s')" % \
              (value.index(i)+1, i)
        try:
            # 执行sql语句
            cursor.execute(sql)
            # 提交到数据库执行
            db.commit()
        except:
            # 发生错误时回滚
            db.rollback()

        # 关闭数据库连接

    db.close()
if __name__ == '__main__':
    insertCode()

你可能感兴趣的:(python生成激活码,并保存在MySQL数据库中)