-
Notifications
You must be signed in to change notification settings - Fork 9
/
Copy pathgetxmlimg.py-bak
executable file
·177 lines (163 loc) · 6.53 KB
/
getxmlimg.py-bak
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
import PIL.Image as Image
import PIL.ImageDraw as ImageDraw
import xml.etree.ElementTree as ET
import zipfile
import subprocess
import re
#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''import xml.etree.ElementTree as ET
def xml2svg(xml_file):
tree = ET.parse(xml_file)
root = tree.getroot()
svg_str = '<svg>'
for child in root:
svg_str += '<' + child.tag
for attr in child.attrib:
svg_str += ' ' + attr + '="' + child.attrib[attr] + '"'
svg_str += '>'
for sub_child in child:
svg_str += '<' + sub_child.tag
for attr in sub_child.attrib:
svg_str += ' ' + attr + '="' + sub_child.attrib[attr] + '"'
svg_str += '/>'
svg_str += '</' + child.tag + '>'
svg_str += '</svg>'
return svg_str'''
class getsavexml():
'''def xml2svg(xml_file):
tree = ET.parse(xml_file)
root = tree.getroot()
svg_str = '<svg>'
for child in root:
svg_str += '<' + child.tag
for attr in child.attrib:
svg_str += ' ' + attr + '="' + child.attrib[attr] + '"'
svg_str += '>'
for sub_child in child:
svg_str += '<' + sub_child.tag
for attr in sub_child.attrib:
svg_str += ' ' + attr + '="' + sub_child.attrib[attr] + '"'
svg_str += '/>'
svg_str += '</' + child.tag + '>'
svg_str += '</svg>'
return svg_str'''
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
# 获取到文件列表后，进行比较分辨率，选取分辨率最高的张图片
# /home/gfdgd_xi/Downloads/MT2.12.2.apk
try:
iconbackpath = getmaxsize(backimgs)
iconforepath = getmaxsize(foreimgs)
except:
if len(backimgs):
iconbackpath = backimgs[0]
else:
iconbackpath = ""
if len(foreimgs):
iconforepath = foreimgs[0]
else:
iconforepath = ""
print(iconbackpath, iconforepath)
#从APK文件获取最终图片
zipapk = zipfile.ZipFile(apkFilePath)
try:
iconback = zipapk.open(iconbackpath)
except:
iconback = None
try:
iconfore = zipapk.open(iconforepath)
except:
iconfore = None
# 叠加图片，mask 设置前景为蒙版
try:
iconbackimg = Image.open(iconback).convert("RGBA")
except:
pass
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)