Python 练习册，每天一个小程序(0002)

来源：互联网发布：云舒网络拖欠工资编辑：程序博客网时间：2024/05/16 14:51

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

# coding: utf-8# 第 0002 题：将 0001 题生成的 200 个激活码（或者优惠券）保存到 MySQL 关系型数据库中。import uuidimport MySQLdbdef 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 codesdef 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))

0 0