宜配屋

Python生成指定数量的优惠码实操内容

yipeiwu_com5年前 (2020-03-06)Python基础

Python生成指定数量的优惠码

打开Python开发工具IDLE，新建‘codeGen.py'文件，并保存

导入需要的包，这里需要random和string，代码如下：

import string

import random

假设需要生成200个，长度为10的优惠码，实际应用这个可以作为参数进来

CODE_COUNT = 200

CODE_LEN =10

设计的优惠码由字母和数字组成，通过下面函数获取所有字母和数字，字母区分大小写

def strall():

  return (string.ascii_letters+string.digits)

string.ascii_letters是python3的写法，python2是string.letters

生成优惠码，这里用set()去除重复，10个字符很容易产生重复，重复数据set是不会添加的，利用这个特性保证了200个不重复的优惠码

def codeGen():

  CodeSet = set()

  while len(CodeSet) < CODE_COUNT:

    code = ''.join([random.choice(strall()) for i in range(CODE_LEN)])

    CodeSet.add(code)

  return CodeSet

这里最好把strall()单独拿出来用变量保存，不然每次循环都会执行一次这个函数

编写程序入口，调用优惠码生成函数打印出所有优惠码，完整代码如下：

import string

import random



#假设需要生成200个，长度为10的优惠码




CODE_COUNT = 200

CODE_LEN =10







def strall():

  return (string.ascii_letters+string.digits)




def codeGen():

  CodeSet = set()

  while len(CodeSet) < CODE_COUNT:

    code = ''.join([random.choice(strall()) for i in range(CODE_LEN)])

    CodeSet.add(code)

  return CodeSet




if __name__ == '__main__':

  

  print (codeGen())