# -*- coding: utf-8 -*-
# __author__ = SHI
import os
import re
import base64
def findimg(content):
'''
查找网页中所有的img,类似img src='1.png'，返回1.png
:param content: 网页内容
:return: 返回找到的所有图片文件名列表
'''
patt = re.compile('<img src="(.+)" ') #正则表达式查找所有的img
grp = re.findall(patt,content)
# print(grp)
return grp
def findcomment(content):
'''
查找注释，删除注释
:param content: 网页源内容
:return: 删除注释后的网页内容
'''
patt = re.compile('(<!--.+-->)')
grp = re.findall(patt,content)
# print(grp)
for g in grp:
content = content.replace(g,'')
return content
def imgbase64(pic_path):
'''
实现图片的base64编码，返回编码字符串
:param pic_path:
:return:
'''
pic_basename = os.path.basename(pic_path)
file_ext = pic_basename.split('.')[-1].lower()
# print(file_ext)
if file_ext == "jpg" :
tag = "jpg"
elif file_ext == "jpeg" :
tag = "jpg"
elif file_ext == "png" :
tag = "png"
elif file_ext == "bmp" :
tag = "bmp"
else:
print("Unsupported image format !")
return None
with open(pic_path, "rb") as imageFile:
str_pic = base64.b64encode(imageFile.read()).decode('ascii')
# print(str_pic)
str_begin = 'data:image/' + tag + ';base64,'
result_str = str_begin + str_pic
# print(result_str)
return result_str
def readhtml():
'''
读取当前目录下的所有html文件，并查找本地图片，实现嵌入图片的base64编码
:return: 无
'''
for file in os.listdir('.'):
if file.endswith(".html"):
# print (file)
with open(file,'r+',encoding='utf-8') as html:
content = html.read() # 读取html文件内容
pics = findimg(content) # 查找所有内容中的本地图片
for pic in pics:
if os.path.exists(pic):
base64code = imgbase64(pic) # base64编码图片文件
content = content.replace(pic,base64code) # 替换html文件内容
content = findcomment(content) # 去除html文件中的注释
html.seek(0)
html.write(content) # 覆写
# 打包命令
# pyinstaller --noupx -F -w --icon=mylogo.ico imgbase64.py
if __name__ == '__main__':
readhtml()
