python学习之操作系统相关练习题

题目要求：

在当前目录新建目录img, 里面包含多个文件, 文件名各不相同(X4G5.png)
将当前img目录所有以.png结尾的后缀名改为.jpg

自己写的：

##生成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 in range(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 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)]
        # 或者用以下的方式
        # 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')