python常见模块之OS模块和time模块

def gen_code(len=4):
 # 随机生成4位验证码
 li = random.sample(string.ascii_letters+string.digits,len)
 return  ''.join(li)
def create_file():
 # 随机生成100个文件名
 li = { gen_code() for i in range(100)}
 os.mkdir('img')
 for name in li:
  os.mknod('img/' + name + '.png')
#create_file()
def modify_suffix(dirname,old_suffix,new_suffix):				#利用回车，出现参数解释（回车--输入""""""---再回车）
 """
 :param dirname: 要操作的目录
 :param old_suffix: 之前的后缀名
 :param new_suffix: 新的后缀名
 :return:
 """
 # 1.判断查找的目录是否存在，如果不存在 报错
 if os.path.exists(dirname):
  # 2.找出所有以old_suffix(.png)结尾的文件
  pngfile = [filename for filename in os.listdir(dirname)
 if filename.endswith(old_suffix)]
  # 3.将后缀名和文件名分开，留下文件名
  basefile = [os.path.splitext(filename)[0]
  for filename in pngfile]
  # 4.重命名文件
  for filename in basefile:
oldname = os.path.join(dirname,filename+old_suffix)
newname = os.path.join(dirname,filename+new_suffix)
os.rename(oldname,newname)
print('%s命名为%s成功' %(oldname,newname))
 else:
  print('%s 不存在，不能操作...' %(dirname))
modify_suffix('imgs','.png','.txt')