Python练习006

题目：生成邀请码。注意邀请码由大小写字母以及数字随机组成，每个邀请码的长度任意，最后写入到codes.txt文件中。
代码：

import string, random

def generateCode(num, length):
	# 等同于'0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
    code = string.digits + string.ascii_letters
    with open('codes.txt', 'w') as file:
        for i in range(num):
            codes = [random.choice(code) for i in range(length)]
            file.write(''.join(codes) + '\n')
        

if __name__ == '__main__':
    generateCode(num = 200, length = 7)