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)