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)