第 0002 题:将 0001 题生成的 200 个激活码(或者优惠券)保存到 MySQL 关系型数据库中。
# coding: utf-8
# 第 0002 题:将 0001 题生成的 200 个激活码(或者优惠券)保存到 MySQL 关系型数据库中。
import uuid
import MySQLdb
def generate_code(num):
codes = []
i = 0
while True:
code = str(uuid.uuid4()).replace('-', '')
if code not in codes:
codes.append(code)
i += 1
if i >= num:
break
return codes
def save_to_db(codes):
conn = MySQLdb.Connect(host='127.0.0.1', port=3306, user='root', passwd='xushao', db='test')
try:
cursor = conn.cursor()
sql = "insert into codes(id, code) values(%s, %s)"
for i in range(len(codes)):
cursor.execute(sql, (i, codes[i]))
conn.commit()
cursor.close()
finally:
conn.close()
if __name__ == "__main__":
save_to_db(generate_code(200))