python生成4位纯数字验证码

'''
思路
生成0-9999的字符串 不足4位前面补0
以生成器的方式每次调用 随机return一个
'''
import random


class A:
    def __init__(self):
        self.create = self.create_()

    def create_(self):
        code_ls = list(map(lambda x: '0' * (4 - len(str(x))) + str(x) if x < 1000 else str(x), [i for i in range(10000)]))
        while True:
            yield random.choice(code_ls)

    def get_auth_code(self):
        return next(self.create)


if __name__ == '__main__':
    a = A()
    for i in range(3):
        print(a.get_auth_code())

image.png

image.png