##生成100个4位随机码,放到列表中
code_str = string.ascii_letters + string.digits
def gen_code(len=4):return''.join(random.sample(code_str,len))
li =[gen_code()for i inrange(100)]
os.mkdir('img')for i in li:
os.mknod('img/'+ i +'.png') ##创建100个png文件
# print(os.listdir('img')) ##列出img目录下的文件内容
pngli =[i for i in os.listdir('img')if i.endswith('.png')]
pngspilt =[os.path.splitext(i)[0]for i in pngli]for i in pngspilt:
os.path.join('img',i+'.jpg')
老师的:
import os
import string
import random
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 inrange(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)]
# 或者用以下的方式
# pngfiles =filter(lambda filename:filename.endswith(old_suffix),os.listdir(dirname))
# 3.将文件名和后缀名分开,留下文件名
basefiles =[os.path.splitext(filename)[0]for filename in pngfile]print(basefiles)
# 4.文件重命名
for filename in basefiles:
#需要加上目录名
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('img','.jpg','.png')