python之生成随机数和字符串

#coding:utf-8

import random
import sys

def stringList():
str1 = []
str3 = []
for i in range(97, 123):
str1.append(chr(i))
str2 = ''.join(str1)

for i in range(65, 91):
str3.append(chr(i))
str4 = ''.join(str3)
strall = str2 + str4

return strall

n = int(input("Please input u number :").strip())
'''
def idList():
x = 0
x += 1
return x
'''
def nameList():
a = random.randint(3,5)
name = ''.join(random.sample(stringList(), a))
return name

def dePartment():
list1 = ['IT', 'caiwu', 'xiaoshou']
dpt = random.choice(list1)
return dpt

def phoneList():
num = '0123456789'
phone = ''.join(random.choice(num) for i in range(11))
return phone

#把所有生成的数据写入到文件
listfile = open('memberlist.txt', 'w')
listfile.write('id    name    department        phone\n')
listfile.close()

output = sys.stdout
with open('memberlist.txt', 'a+') as f:
sys.stdout = f
for i in range(0, n):
i += 1
print('%-5d %-8s %-13s %s' % (i, nameList(), dePartment(), phoneList()))

sys.stdout = output
print('Wrote a total of %d rows of data' % i)