提取APK图标脚本

import os
import re
import shutil
import xml.dom.minidom
from xml.dom.minidom import parse

apk_path = 'apks/'
icons_path = 'icons/'


def decomp(apk_name):
    os.system("apktool d %s%s" % (apk_path, apk_name))


def search_for_icon(base_path, icon_str):
    if icon_str is None or icon_str == "":
        return None
    matches = re.match(r'@(.*)\/(.*)', icon_str)
    path_name = matches.group(1)
    icon_name = matches.group(2)
    ls = os.listdir(base_path)
    ls.reverse()
    for p in ls:
        if p.startswith(path_name):
            path = os.path.join(base_path, p, icon_name+".png")
            if os.path.exists(path):
                return path
    return None


def parse_for_icon(apk_name):
    base_path = apk_name[:-4]
    dom = parse(base_path+"/AndroidManifest.xml")
    collection = dom.documentElement
    tag = collection.getElementsByTagName("application")[0]
    # Something like: @mipmap/ic_launcher
    icon_android_path = tag.getAttribute("android:icon")
    return search_for_icon("%s/res/" % base_path, icon_android_path)


def del_file(path):
    list = os.listdir(path)
    for l in list:
        c_path = os.path.join(path, l)
        if os.path.isdir(c_path):
            del_file(c_path)
        else:
            os.remove(c_path)
    os.rmdir(path)


def move_apk_out(path):
    ls = os.listdir(path)
    for i in ls:
        c_path = os.path.join(path, i)
        if os.path.isdir(c_path):
            move_apk_out(c_path)
        elif i[-3:] == 'apk':
            shutil.move(c_path, apk_path+i)


if __name__ == '__main__':
    move_apk_out(apk_path)
    apk_list = []
    list = os.listdir(apk_path)
    for l in list:
        path = os.path.join(apk_path, l)
        if os.path.isfile(path) and l[-3:] == 'apk':
            apk_list.append(l)
    print(apk_list)
    for apk in apk_list:
        print("Geting icon for: %s" % apk)
        decomp(apk)
        icon_path = parse_for_icon(apk)
        if icon_path is not None and os.path.exists(icon_path):
            shutil.move(icon_path, "%s%s.png" % (icons_path, apk[:-4]))
        else:
            print("Failed to get: " + apk)
        if os.path.exists(apk[:-4]):
            del_file(apk[:-4])