生成200個8位的驗證碼

# -*- coding: utf-8 -*-
import mysql.connector
import random,string

def readkey(num,length):
    with open('C:/Users/Administrator/Desktop/randomnum.txt', 'w+') as fo:
        for i in range(num):
            chars = string.ascii_letters + string.digits
            s = [random.choice(chars) for i in range(length)]
            fo.write(''.join(s) + '\n')
    with open('C:/Users/Administrator/Desktop/randomnum.txt', 'r') as fo:
        keylist = fo.readlines()
    return keylist

def storeinMysql(keylist):
    try:
        # 連線資料庫
        conn = mysql.connector.connect(user='root', password='password', db = 'mysql')
        cursor = conn.cursor()
    except BaseException as e:
        print('connection failed ')
    else:
        try:
            # 若沒有database randnum，則建立一個名為randnum的database
            cursor.execute('create database if not exists randnum')
            cursor.execute('use randnum')
            # 若沒有表，則建立一個名為randkey的表，包含randkey和name兩個欄位
            cursor.execute('''create table if not exists randkey3(
                                    keynum varchar(20) primary key,name varchar(20))''')
            for line in keylist:
                # print (line)
                cursor.execute('insert into randkey3 (keynum,name) values (%s, %s)', [line, 'cathy'])
                conn.commit()
        except BaseException as e:
            print(e)
    finally:
        cursor.close()
        conn.close()

if __name__ == '__main__':
    storeinMysql(readkey(200,8))
 

在生成驗證碼的時候，可以使用uuid4，直接生成list，這樣不用存檔案，更簡潔

轉自github的ilcyb的答案

import uuid

def generateActivationCode(num):
    codeList = []
    for i in range(num):
        code = str(uuid.uuid4()).replace('-','').upper()
        while code in codeList:
            code = str(uuid.uuid4()).replace('-','').upper()
        codeList.append(code)

    return codeList
 

fo.write(''.join(s) + '\n')

由於s取出來為一個list，需要用join函式，以''連線起來