markdown自动生成的img目录自动清理脚本

markdown编辑器支持直接复制粘贴图片了，虽说很方便，但是在img目录下生成了许多过期的图片。写个脚本自动清理下。

import os
import re

imgs = set()
links = set()

for file in os.listdir('./img'):
    imgs.add(file)

regex = r'/img/(.*)\)'
p = re.compile(regex)  

for file in os.listdir('./'):
    if ((file.split('.'))[-1] == 'md'):
        fobj = open(file,'r', encoding='utf-8');
        for fileLine in fobj:
            m = p.search(fileLine)  
            if (m):
                links.add(m.group(1))

print(links)
print('*'*20)
print(imgs)
print('*'*20)
print(imgs-links)

for file in (imgs-links):
    os.remove('./img/'+file)