os模块练习
- 在当前目录新建目录img, 里面包含多个文件, 文件名各不相同(X4G5.png)
- 将当前img目录所有以.png结尾的后缀名改为.jpg
提示:
os.listdir(path) 返回path指定的文件夹包含的文件或文件夹的名字的列表
import random
import string
import os
def gen_code(len=4):
# 随机生成四位随机数
li = random.sample(string.ascii_letters+string.digits,len)
return ''.join(li)
# print(gen_code())
def creat_file():
# 随机生成多个文件名
li = {gen_code() for i in range(100)}
os.mkdir('img')
for name in li:
os.mknod('img/' + name + '.png')
# creat_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('img','.html','.php')