文章目录
需求起源
最近在用typora写博客时遇到了一个问题,什么问题呢,就是我们在md文件中插入图片,这个图片它必须是以文件路径或网页路径的形式链接到md中。而我平常在博客中插入的图片大多都是截图,也就是说,我每次截图完成后,都要把它保存为文件,然后再以超链接的形式写到md中,这给我带来了不少麻烦。
我以前都习惯用word来做笔记,在word中,我截图后可以直接点击完成,然后Ctrl V直接粘贴到文档中,比起md的这一番操作方便不少。于是我突发奇想,如果我直接点击完成截图,再Ctrl V粘贴到md文件中,会发生什么呢?
于是我尝试了一下,发现是可以实现的,只是这个图片路径有些问题——这个图片路径是在typora安装目录下的一个文件夹内,而typora安装目录是在C盘。很显然,这会挤占系统盘的空间。
我又想起word中插入的图片,就算你把图片源文件删除,对文档中的图片也不会有影响,因为图片文件已经被写入word文档中了。于是我就想着,在md文件中是不是也有类似的操作。然后我找到这张图片把它删了,发现md文档中的图片也不见了。
于是我得出结论,word和md两种文档对于图片的处理方式是不同的,word中插入图片是直接写入文件,md中插入图片时用超链接的方式从外部引用。
那么既然图片必须要保留,我能不能换一个路径呢,不要在系统盘里,改到其他盘可不可以呢?我想着这应该是默认路径的设置问题。于是我点开了偏好设置,发现里面正好有图像的设置。一看,果然,有关于路径的设置。于是我把路径修改到其他盘,再试了试截图,原来的默认文件夹里没有动静,图片保存到了我所设置的路径中。那么保存路径的问题就完美的解决了!
紧接着,我又发现了一个问题——但凡我截了图并将其粘贴在md文档中,这文件夹内就会生成一个图片文件,而如果我不要这张图片了,我把它从md文档中删除了,但文件夹内的这张图片不会被删除。这样一来,有可能会有很多无效的图片挤占空间资源。而这些无效图片还不好清理,因为如果删错了,可能会使md文件中的图片没法加载,这又是一个棘手的问题。
这时我突然想到,md文件中插入图片时写入的是图片的超链接,也就是图片的路径,那么我们可以用程序来比对文件夹中的图片的路径和md文件中写入的图片的路径,如果能匹配上的就不能删,如果所有都匹配不上的就可以删。
需求分析
现在我们明确了代码需求——获取所有图片文件的路径以及所有md文件的内容,如果图片文件的路径在所有md文件中都没有被引用,那么这张图片就可以删除。
OK,我们现在来拆分需求,进行模块化处理。
首先,我们需要获取所有图片文件的绝对路径,这是很简单的,通过os库可以获取所有图片文件名,再在前面用join加上文件夹路径即可。
其次,我们需要获取所有md文件的绝对路径,这也很好解决,通过os库的walk函数可以遍历所有文件,在所有文件中找到.md结尾的文件即可
再次,我们需要判断图片路径是否包含在md文件中。这里我们一定要想清楚它的逻辑关系,我们给定一个md文件的内容,再遍历所有图片路径来判断,如果其中一个图片路径包含在它的内容中,那么这个图片就百分百不能删,紧接着再换一个md文件接着判断,如此循环。可以看出这里需要嵌套两层循环。同时,我们需要把判断得到的不能删的图片用一个列表存储起来。
最后,根据不能删的图片列表,把所有不在该列表中的图片删除即可。
代码实现
导入需要的包
import os
设置函数
设置获取文件路径的函数
需要图片和md文件的路径,可以设置同一个函数来满足两种文件的获取
# 遍历目录下所有文件,返回文件路径列表
# 输入的是绝对路径则返回绝对路径,输入的是相对路径则返回相对路径
def getFileList(filetype, path='./',):
file_list = list()
for dirpath, dirnames, filenames in os.walk(path):
for filename in filenames:
if filename.endswith(filetype):
file = os.path.join(dirpath, filename)
file_list.append(file)
return file_list
设置读取md文件内容的函数
# 读取文件,默认以文本读取,编码类型为utf-8
def readFromFile(file, readmode='r', encoding='utf-8'):
with open(file, readmode, encoding=encoding) as f:
text = f.read()
return text
主函数部分
读取文件路径
md_path = os.getcwd()
pic_path = os.path.join(os.getcwd(), 'md_pic')
md_list = getFileList(path=md_path, filetype='.md')
pic_list = getFileList(path=pic_path, filetype=('.png', '.jpg', '.jpeg'))
判断是否能删,存入列表
alive_pic_list = list()
for md in md_list:
text = readFromFile(md)
for pic in pic_list:
if pic in text:
alive_pic_list.append(pic) # 但凡pic在md中,这个pic就不能删
删除可删的图片
for pic in pic_list:
if not (pic in alive_pic_list):
os.remove(pic)
完成主函数部分
def main():
md_path = os.getcwd()
pic_path = os.path.join(os.getcwd(), 'md_pic')
md_list = getFileList(path=md_path, filetype='.md')
pic_list = getFileList(path=pic_path, filetype=('.png', '.jpg', '.jpeg'))
alive_pic_list = list()
for md in md_list:
text = readFromFile(md)
for pic in pic_list:
if pic in text:
alive_pic_list.append(pic) # 但凡pic在md中,这个pic就不能删
for pic in pic_list:
if not (pic in alive_pic_list):
os.remove(pic)
最终代码
import os
# 遍历目录下所有文件,返回文件路径列表
# 输入的是绝对路径则返回绝对路径,输入的是相对路径则返回相对路径
def getFileList(filetype, path='./',):
file_list = list()
for dirpath, dirnames, filenames in os.walk(path):
for filename in filenames:
if filename.endswith(filetype):
file = os.path.join(dirpath, filename)
file_list.append(file)
return file_list
# 读取文件,默认以文本读取,编码类型为utf-8
def readFromFile(file, readmode='r', encoding='utf-8'):
with open(file, readmode, encoding=encoding) as f:
text = f.read()
return text
def main():
md_path = os.getcwd()
pic_path = os.path.join(os.getcwd(), 'md_pic')
md_list = getFileList(path=md_path, filetype='.md')
pic_list = getFileList(path=pic_path, filetype=('.png', '.jpg', '.jpeg'))
alive_pic_list = list()
for md in md_list:
text = readFromFile(md)
for pic in pic_list:
if pic in text:
alive_pic_list.append(pic) # 但凡pic在md中,这个pic就不能删
for pic in pic_list:
if not (pic in alive_pic_list):
os.remove(pic)
if __name__ == '__main__':
main()