python：操作系统的练习题（创建目录和文件，修改文件后缀名）

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

import random
import string
import os

os.mkdir('img')
code_str = string.ascii_letters + string.digits


def gen_code(len=4):
    return ''.join(random.sample(code_str, len))


for i in range(100):
    os.mknod('/home/kiosk/PycharmProjects/untitled/img/%s.png' % (gen_code()))

from os.path import join

# 2. 将当前img目录所有以.png结尾的后缀名改为.jpg。

files = [name for name in os.listdir('img/')]

j = [os.path.splitext(i)[0] for i in files]
for i in j:
    old = os.path.join('/home/kiosk/PycharmProjects/untitled/img/', i + '.png')
    new = os.path.join('/home/kiosk/PycharmProjects/untitled/img/', i + '.jpg')
    os.rename(old, new)