说明
最近在写markdown文档的过程中,经常需要插入一些图片.因为托管博客的服务器空间有限,所以上传图片到图床再插入到markdown中.有时候又插入网上的图片,这些图片随时可能失效.导致我的博客网站显示图片错误.
所以花了一点时间,用python 正则匹配 markdown中图片链接,然后下载图片保存到本地文件夹img
中.这样就不用担心图片失效啦,当我找到稳定的图床,可以随时把这些图片再上传更新,美滋滋~~~
功能
.md
文件自动搜索- 正则匹配图片链接
- 爬取图片内容
- 保存文本到本地
步骤
读取文件 => 正则匹配 => 图片下载 => 保存本地
python代码
详细见我的github地址: https://github.com/wangshub/markdown-img-backup
# coding=utf-8
import sys
import os
import re
import requests
import urllib
import urllib2
def search(path, word):
for filename in os.listdir(path):
fp = os.path.join(path, filename)
if os.path.isfile(fp) and word in filename:
print fp
download(str(fp))
elif os.path.isdir(fp):
search(fp, word)
def download(file_path):
# filename = "test"
name = file_path.split(u"/")
filename = name[-1]
f_md = open(file_path)
# all text of md file
text = f_md.read().decode('utf-8')
# regex
img_reg = r'\!{1}\[(.*?)\]\((.*?)\)'
result = re.findall('!\[(.*)\]\((.*)\)', text)
for i in range(len(result)):
img_quote = result[i][0]
img_url = result[i][1]
# download img
request = urllib2.Request(img_url)
response = urllib2.urlopen(request)
img_contents = response.read()
# img name spell
urlname = img_url.split(u"/")
img_name = filename + '_' + \
str(i) + '_' + img_quote + str(urlname[len(urlname) - 1])
print img_name, '~~~', img_url
# write to file
f_img = open('img/' + img_name, 'wb')
f_img.write(img_contents)
f_img.close()
f_md.close()
search(sys.argv[1], '.md')
食用方法
python md_image_bacup.py /path/to/your/file/
作者
Author : WangSong
E-mail : [email protected]