1 # -*- coding: utf-8 -*-
2 # @Time : 8/25/2020 7:40 PM
3 # @Author : OhiyoX
4 # @FileName: MDImgScripts.py
5 # @Software: PyCharm
6 # @Blog ：https://blog.ohiyox.in/
7 # @Desc : 1、用于自动清除typora写作后多余的图片，简洁文件夹
8 # 2、也可以自动上传图片到图床，同时修改文档中的图片链接
9 # 3、若图片为网络图片，将自动转移到图床，同时修改文档中的图片链接
10 # 4、若图片在图床，但存放位置不规范，也会自动转移位置，同时修改文档中的图片链接
11
12
13 import os
14 import re
15 import json
16 import shutil
17 import sys
18 import time
19 from urllib.parse import quote, unquote
20
21 from datetime import datetime
22
23
24
25 # 配置
26 # {
27 # "test_mode": True, # 调试模式
28 # "date": {"year": "2019", # 设置文件夹日期，留空默认为本年
29 # "month": "1" # 设置文件夹日期，留空默认为本月
30 # },
31 # "clean_local_assets": True,
32 # "main_oss_folder_ref": "blogimg", # 设置图床文件夹
33 # "style": "!xwbp", # 设置默认上传的图片规则，以后缀!标识
34 # "relocate_oss_existing_file": True, # 对于已在图床中的图片，设置是否需要重新移动整理图片
35 # "delete": True # 对于移动图片，是否删除原位置
36 # }
37
38 with open('config.json') as cfig:
39 config = json.load(cfig)
40
41
42 class ImgMD:
43 """MD图片处理类"""
44
45 def __init__(self):
46 # 与MD文档有关的
47 self.article_filepath = None
48 self.assets_dirpath = None
49 self.content = ""
50 # 文档中的图片信息
51 self.imgs_list = []
52 self.imgs_url_list = []
53 # 图片在本地中的存储信息
54 self.assets_list = []
55
56 self.get_doc_imgs_list()
57 self.get_assets_list()
58
59 self.temp_dir = 'temp_img'
60
61 def get_content(self, article_filepath="", force=False):
62 """获得文本内容"""
63 if self.content and not force:
64 return self.content
65 else:
66 foo = False
67 count = 0
68 while not foo:
69 if count <= 5:
70 try:
71 if not config['test_mode']:
72 if not article_filepath:
73 self.article_filepath = input("Input the article path: ").strip('\"')
74 #print(self.article_filepath)
75
76 if "\"" in self.article_filepath:
77 self.article_filepath = re.search("\"(.*)\"", self.article_filepath).group(1)
78 article_filepath = self.article_filepath
79 else:
80 article_filepath = self.article_filepath
81 with open(article_filepath, 'r', encoding="UTF-8") as f:
82 self.content = f.read()
83 foo = True
84 return self.content
85 except:
86 print("Notice! File is not found or an error occurs, retry.")
87 count += 1
88 else:
89 exit('failed getting article file path.')
90
91 def get_doc_imgs_list(self, url=False, force=False):
92 """获得文档中的图片列表"""
93 if not force:
94 if self.imgs_list and not url:
95 return self.imgs_list
96 if self.imgs_url_list and url:
97 return self.imgs_url_list
98
99 content = self.get_content(force=force)
100
101 #self.imgs_url_list = re.findall('!\[.*?]\((.*?)\)', content, re.S)
102 self.imgs_url_list = re.findall('image-[0-9]{4,18}.png', content, re.S)
103 imgs_list = []
104 for img_url in self.imgs_url_list:
105 img_full_name = self.get_doc_imgs_list(img_url) # 不支持绝对路径
106 imgs_list.append(img_full_name)
107 if not url:
108 print("Done, %s img(s) found in article." % len(imgs_list))
109 #for i, img in enumerate(imgs_list):
110 # print(f"{i}: {img}")
111
112 self.imgs_list = imgs_list
113 return imgs_list
114 else:
115 return self.imgs_url_list
116
117
118 def get_assets_list(self):
119 """获得图片文件夹中的图片列表"""
120 bar = False
121 count = 0
122 while not bar and count <= 5:
123 try:
124 if not config['test_mode']:
125 self.assets_dirpath = input("input the imgs path (Leave it empty to use default folder): ")
126 if self.assets_dirpath == '':
127 self.assets_dirpath = self.article_filepath.replace('.md', '.assets')
128 for a, b, c in os.walk(self.assets_dirpath):
129 assets_list = c
130 print("Done, %s img(s) found in assets." % len(assets_list))
131 print('As below:')
132 for i, a in enumerate(assets_list):
133 print(f"{i}: {a}")
134 self.assets_list = assets_list
135 return assets_list
136 bar = True
137 except:
138 print(Exception)
139 count += 1
140 if count > 5:
141 exit(-1)
142
143 def clear_local_imgs(self):
144 """用于清除文件夹中无用的图片"""
145 flag = False
146 redundant_list = []
147 print("------------------OUTPUT assets_list 1111---------------- \n")
148 print(self.assets_list)
149 print("------------------OUTPUT imgs_list 222---------------- \n")
150 print(self.imgs_list)
151 print("------------------OUTPUT list end---------------- \n")
152 for file in self.assets_list:
153 print(file)
154 u_file = quote(file) # typora中使用了unicode-escape
155 #if str(u_file) not in str(self.imgs_list):
156 if str(self.imgs_list).find(str(u_file)) == -1 :
157 redundant_list.append(self.assets_dirpath + '\\' + file)
158 flag = True
159 if flag:
160
161 for x in redundant_list:
162 os.remove(x)
163 print(x + " is removed.")
164 else:
165 print("Scan finished, no redundant img is found in assets.")
166
167 def img_relocate(self, remote_img_ref, new_remote_img_ref, delete=True):
168 """将图床图片移动到合适位置"""
169 print('relocate_oss_existing_file process:', end='')
170
171 def delete_(_remote_img_ref):
172 # delete 指示是否删除旧位置的文件
173 if delete:
174 if bucket.delete_object(_remote_img_ref):
175 print('deleted "' + self.get_filename_from_url(_remote_img_ref) + '" in original loc.')
176 return True
177 return False
178 p_dir = re.search('(.*)/', _remote_img_ref, re.S).group(1)
179 if bucket.delete_object(p_dir):
180 # 清理掉旧的文件夹
181 print('deleted empty loc.')
182 return True
183
184 bucket = oss2.Bucket(self.auth, self.endpoint, self.oss_info['Bucket'])
185 exist = bucket.object_exists(new_remote_img_ref)
186 if exist:
187 return delete_(remote_img_ref)
188 else:
189 if bucket.copy_object(self.oss_info['Bucket'], remote_img_ref, new_remote_img_ref):
190 return delete_(remote_img_ref)
191 else:
192 return False
193
194 if __name__ == '__main__':
195 md = ImgMD()
196 if config['clean_local_assets']:
197 md.clear_local_imgs()
以上代码是根据文章开头的代码出处来修改的，请将以上文件保存成python文件 .py，然后运行即可。
标签：remote,img,self,Typora,list,多余,移除,print,ref From： https://www.cnblogs.com/xiaokangkp0602/p/18044329