本脚本旨获取指定文件中在网络中所有的图片,并存储在本地,并且修改文件中的url指向本地图片。
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : codekiller
# @Time : 2022/2/13 20:42
# @Email : jcl1345414527@163.com
# @File : MdImage.py
# @Description: 用来进行md文件中图片的获取,保存在本地,并修改md文件。只需要修改ua和文件的位置就行了
import re
import requests
import time
ua = 自己去浏览器寻找
headers = {"User-Agent": ua}
# 文件的位置
file_dir = r'F:\md笔记\java\JVM\\'
file_n = 'test'
file_suffix = '.md'
fileName = file_dir+file_n+file_suffix
# 存储图片的位置
img_tsrc = 'assets/'
img_src = file_dir + img_tsrc
with open(file=fileName, mode="r", encoding='utf-8') as f1:
# 用来计数
s_count = 0
fail_count = 0
sus_count = 0
lines = f1.readlines()
for i in range(0, len(lines)):
url = re.findall('<img src="(.*?)"', lines[i], re.S)
if len(url)!=0:
url = url[0]
#处理本地图片
if len(re.findall(img_tsrc+'.*', url, re.S))!=0:
print('第',i,'行:为本地图片')
s_count+=1
continue
#获取时间作为图片文件名
timestamp = int(time.time())
image_name = str(time.strftime('%Y%m%d%H%M%S', time.localtime(timestamp)))+str(i)+".png"
#请求图片
img = requests.get(url,headers=headers)
#判断是否获取
if img.status_code != 200:
print('第',i,'行:','!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!',img.status_code)
fail_count+=1
continue
#存储
target = open(img_src+image_name,'ab')
target.write(img.content)
target.close()
lines[i] = lines[i].replace(url,img_tsrc+image_name)
print('第',i,'行:',url+'成功转为'+img_tsrc+image_name)
sus_count+=1
print('---本地图片为:',s_count,'个---')
print('---获取失败为:',fail_count,'个---')
print('---转化成功为:',sus_count,'个---')
#将处理过的cNames写入新的文件中
fileName2 = file_dir+file_n+'-修改'+file_suffix
with open(file = fileName2,mode='w',encoding='utf-8') as f2:
f2.writelines(lines)
控制台输出:
报感叹号的表示转化失败,后面是获取图片的状态码;为本地图片的说明不需要转化。
转化成功的效果: