119 lines
4.7 KiB
Python
119 lines
4.7 KiB
Python
|
import PIL.Image as Image
|
|||
|
|
import PIL.ImageDraw as ImageDraw
|
|||
|
|
import zipfile
|
|||
|
|
import subprocess
|
|||
|
|
import re
|
|||
|
|
|
|||
|
|
class getsavexml():
|
|||
|
|
|
|||
|
|
def savexml(self,apkFilePath,xmlpath,iconSavePath):
|
|||
|
|
cmddumpid = "aapt dump xmltree "+ apkFilePath + " " + xmlpath
|
|||
|
|
print(cmddumpid)
|
|||
|
|
xmltree = subprocess.getoutput(cmddumpid)
|
|||
|
|
xmls = xmltree.splitlines()
|
|||
|
|
# find strs ,print next line
|
|||
|
|
def FindStrs(lines,strs):
|
|||
|
|
i=0
|
|||
|
|
while i < len(lines):
|
|||
|
|
if re.search(strs,lines[i]):
|
|||
|
|
tmpstr = lines[i+1]
|
|||
|
|
i += 1
|
|||
|
|
Resultstr = tmpstr.split(":")[-1].split("=")[-1].split("0x")[-1]
|
|||
|
|
return Resultstr
|
|||
|
|
else:
|
|||
|
|
i += 1
|
|||
|
|
#从apk的信息中获取前后景图片的ID号
|
|||
|
|
backimgid = FindStrs(xmls,"background")
|
|||
|
|
foreimgid = FindStrs(xmls,"foreground")
|
|||
|
|
print(backimgid)
|
|||
|
|
print(foreimgid)
|
|||
|
|
|
|||
|
|
# 直接从apk resource文件获取前后两层图片路径及ID字符串
|
|||
|
|
resource = subprocess.getoutput("aapt dump --values resources " + apkFilePath + "| grep -iE -A1 " + "\"" + backimgid + "|" + foreimgid + "\"")
|
|||
|
|
resourcelines = resource.splitlines()
|
|||
|
|
print(resourcelines)
|
|||
|
|
|
|||
|
|
# 从过滤出的字符串中获取所有相同ID的图片路径
|
|||
|
|
def Findpicpath(lines,imgid):
|
|||
|
|
i=0
|
|||
|
|
Resultstr = []
|
|||
|
|
while i < len(lines):
|
|||
|
|
if re.search(imgid,lines[i]) and re.search("string8",lines[i+1]) :
|
|||
|
|
print(lines[i+1])
|
|||
|
|
tmpstr = lines[i+1].replace("\"","")
|
|||
|
|
i += 1
|
|||
|
|
Resultstr.append(tmpstr.split()[-1])
|
|||
|
|
else:
|
|||
|
|
i += 1
|
|||
|
|
return Resultstr
|
|||
|
|
|
|||
|
|
#获取所有带前后图片ID的图片路径（相同背景或者前景的图片ID但分辨率不一样）
|
|||
|
|
backimgs = Findpicpath(resourcelines,backimgid)
|
|||
|
|
foreimgs = Findpicpath(resourcelines,foreimgid)
|
|||
|
|
print(backimgs)
|
|||
|
|
print(foreimgs)
|
|||
|
|
#获取分辨率最高的图片路径
|
|||
|
|
def getmaxsize(imgs):
|
|||
|
|
j = 0
|
|||
|
|
size=(0,0)
|
|||
|
|
zipapk = zipfile.ZipFile(apkFilePath)
|
|||
|
|
imgpath = ""
|
|||
|
|
while j < len(imgs):
|
|||
|
|
print(imgs[j])
|
|||
|
|
img = Image.open(zipapk.open(imgs[j]))
|
|||
|
|
print(imgs[j])
|
|||
|
|
print(img.size)
|
|||
|
|
if size < img.size:
|
|||
|
|
size = img.size
|
|||
|
|
imgpath = imgs[j]
|
|||
|
|
j += 1
|
|||
|
|
return imgpath
|
|||
|
|
|
|||
|
|
# 获取到文件列表后，进行比较分辨率，选取分辨率最高的张图片
|
|||
|
|
iconbackpath = getmaxsize(backimgs)
|
|||
|
|
iconforepath = getmaxsize(foreimgs)
|
|||
|
|
print(iconbackpath + " " + iconforepath)
|
|||
|
|
|
|||
|
|
#从APK文件获取最终图片
|
|||
|
|
zipapk = zipfile.ZipFile(apkFilePath)
|
|||
|
|
iconback = zipapk.open(iconbackpath)
|
|||
|
|
iconfore = zipapk.open(iconforepath)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 叠加图片，mask 设置前景为蒙版
|
|||
|
|
iconbackimg = Image.open(iconback).convert("RGBA")
|
|||
|
|
iconforeimg = Image.open(iconfore).convert("RGBA")
|
|||
|
|
iconbackimg.paste(iconforeimg,mask=iconforeimg)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 圆角图片函数，网上拷贝的
|
|||
|
|
def circle_corner(img, radii): #把原图片变成圆角，这个函数是从网上找的，原址 https://www.pyget.cn/p/185266
|
|||
|
|
"""
|
|||
|
|
圆角处理
|
|||
|
|
:param img: 源图象。
|
|||
|
|
:param radii: 半径，如：30。
|
|||
|
|
:return: 返回一个圆角处理后的图象。
|
|||
|
|
"""
|
|||
|
|
# 画圆（用于分离4个角）
|
|||
|
|
circle = Image.new('L', (radii * 2, radii * 2), 0) # 创建一个黑色背景的画布
|
|||
|
|
draw = ImageDraw.Draw(circle)
|
|||
|
|
draw.ellipse((0, 0, radii * 2, radii * 2), fill=255) # 画白色圆形
|
|||
|
|
# 原图
|
|||
|
|
img = img.convert("RGBA")
|
|||
|
|
w, h = img.size
|
|||
|
|
# 画4个角（将整圆分离为4个部分）
|
|||
|
|
alpha = Image.new('L', img.size, 255)
|
|||
|
|
alpha.paste(circle.crop((0, 0, radii, radii)), (0, 0)) # 左上角
|
|||
|
|
alpha.paste(circle.crop((radii, 0, radii * 2, radii)), (w - radii, 0)) # 右上角
|
|||
|
|
alpha.paste(circle.crop((radii, radii, radii * 2, radii * 2)), (w - radii, h - radii)) # 右下角
|
|||
|
|
alpha.paste(circle.crop((0, radii, radii, radii * 2)), (0, h - radii)) # 左下角
|
|||
|
|
# alpha.show()
|
|||
|
|
img.putalpha(alpha) # 白色区域透明可见，黑色区域不可见
|
|||
|
|
return img
|
|||
|
|
|
|||
|
|
# 圆角半径1/8边长,保存icon图片
|
|||
|
|
w,h = iconbackimg.size
|
|||
|
|
iconimg = circle_corner(iconbackimg,int(w/8))
|
|||
|
|
iconimg.save(iconSavePath)
|
|||
|
|
|