学习python过程中,对于文件的读写是必不可少的,昨天花了十分钟写了一小段代码。
主要功能是:创建文件--按照用户输入生成随机字符串密码--写入文件
import random
from random import sample
import string
import time
import os
def random_number():
pwnumber=input("请输入需要密码个数:")
pwlength=input("请输入需要密码长度:")
if int(pwlength)<=30:
#获取时间戳
timestamp = time.strftime('%Y%m%d%H%M%S', time.localtime())
str1 = "E:\\"
#拼接路径
filename = str1 + str(time.strftime(timestamp)) + ".txt"
#存在的时候打开,不存在的时候新建
f = open(filename, "a")
for i in range(int(pwnumber)):
#获取随机数
number=random_pinyin = ''.join(sample(string.ascii_letters, 8))
#写入数据
f.write(number+'\n')
#关闭文件
f.close()
else :
print("目前只支持30位以下密码,请重试!!")
random_number()
if __name__=="__main__":
random_number()
其中需要注意的是:
(1)格式化时间戳作为文件名的时候一定不能存在空格;
(2)拼接路径时,一定要转义“\”,才能识别路径。