Python 练习册 2-存储激活码到MySQL数据库

Python 练习册，每天一个小程序,原题来自Yixiaohan/show-me-the-code
我的代码仓库在Github

目标

做为 Apple Store App 独立开发者，你要搞限时促销，为你的应用生成激活码（或者优惠券），使用 Python 生成 200 个激活码（或者优惠券），并将激活码保存到 MySQL 关系型数据库中。

解决方案

该题目采用 python 中的 PyMySQL 模块 来连接操作MySQL数据库，代码如下：

#!/usr/bin/env python3
# -*- coding: utf-8 -*-


# 将0001题目中随机生成的验证码保存到MySQL数据库
import uuid
import pymysql


# 生成 num 个验证码，每个长度为length，可设置默认长度
def create_num(num, length=16):
    result = []
    while num > 0:
        uuid_id = uuid.uuid4()
        # 删去字符串中的'-',取出前length 个字符
        temp = str(uuid_id).replace('-', '')[:length]
        if temp not in result:
            result.append(temp)
            num -= 1
    return result


# 保存到MySQL数据库
def save_to_mysql(code):
    conn = pymysql.connect(
        host='127.0.0.1', 
        port=3306, 
        user='root', 
        passwd=None, 
        db='test')

    try:
        with conn.cursor() as cursor:
            # Create a new record
            sql = "INSERT INTO `codes` (`code`) VALUES (%s)"
            cursor.execute(sql, code)

        # connection is not autocommit by default. So you must commit to save
        # your changes.
            conn.commit()

        with conn.cursor() as cursor:
            # Read a single record
            sql = "SELECT `id`, `code` FROM `codes` WHERE `code`=%s"
            cursor.execute(sql, code)
            result = cursor.fetchone()
            print(result)
    finally:
        conn.close()

for code in create_num(200):
    save_to_mysql(code)