Python 练习册 0001~0003（保存激活码）

第 0001 题：做为 Apple Store App 独立开发者，你要搞限时促销，为你的应用生成激活码（或者优惠券），使用 Python 如何生成 200 个激活码（或者优惠券）？

第 0002 题：将 0001 题生成的 200 个激活码（或者优惠券）保存到 MySQL 关系型数据库中。

第 0003 题：将 0001 题生成的 200 个激活码（或者优惠券）保存到 Redis 非关系型数据库中。

import random
import pymysql
import pymongo
import string

# 生成200个激活码
def create_code():
    code_list = list()
    for j in range(0, 200):
        code_num = ''

        for i in range(0, 15):
            n = random.choice(string.ascii_letters + '1234567890')
            code_num += n

        code_list.append(code_num)
    return code_list

code_list = create_code()

# 存储到mysql

conn = pymysql.connect(
    host='localhost',
    port=3306,
    user='root',
    passwd='root',
    db='actcode'
)
cursor = conn.cursor()
for code_num in code_list:
    sql = 'INSERT INTO code(code_num) VALUES (\'{}\')'.format(code_num)
    cursor.execute(sql)
conn.commit()
conn.close()

# 存储到mongodb
conn = pymongo.MongoClient('localhost', 27017)
db = conn['test']
code= db['code']
for code_num in code_list:
    db.code.insert({'code': code_num})