python 用random函数生成邀请码，并存储在mysql中

本文链接：https://blog.youkuaiyun.com/BigData_ming/article/details/80125880

本文介绍如何利用Python的random模块生成邀请码，并将其存储到MySQL数据库中，详细介绍了随机数生成和Python连接MySQL数据库的基础操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

网上的习题，照着别人的代码学习学习，熟悉python。

随机生成邀请码，并存储起来。

随机：python中的random函数，具体可以看实现方式random.py

存储: 可以存在文件中，也可以存在数据库中。这里选择了mysql，具体可以看菜鸟教程 http://www.runoob.com/python3/python3-mysql.html

#-*- coding:utf-8-*-
#生成邀请码功能
import  pymysql
import string ,random
def gen_code(count,long):
    r = []
    base_string = string.digits+string.ascii_letters
    for i in range(count):
        card_code = ''
        for j in range(long):
            card_code += random.choice(base_string)
        r.append(card_code)
    return r
if __name__ == '__main__':
    #存入mysql数据库功能
    conn = pymysql.connect(host='127.0.0.1', port=3306, user='thunder', passwd='thunderisgreat', db='villa', charset='utf8')
    cursor = conn.cursor()  #创建游标对象，用于操作数据库
    r = gen_code(20,10)
    #print("test",r)
    create_sql = """CREATE TABLE IF NOT EXISTS CARD_CODE(
                                                        id  int,
                                                        code varchar(100)
                                                        )"""
    insert_sql = """INSERT INTO CARD_CODE (id,code) VALUES(%s,%s)""" #参数传递
    try:
        cursor.execute("DROP TABLE IF EXISTS CARD_CODE")
        cursor.execute(create_sql)
        for i in range(20):
            cursor.execute(insert_sql,(i+1,r[i]))   #两个参数传递给insert_sql
        conn.commit()
    except:
        print("dangerous and rollback")
        conn.rollback()
    cursor.close()
    conn.close()