jmton 发表于 2018-8-12 11:05:38

Python——使用ElementTree解析AndroidManifest.xml文件

from ActivityInfo import ActivityInfo  

  
try:
  
    import xml.etree.cElementTree as ET
  
except ImportError:
  
    import xml.etree.ElementTree as ET
  
import sys
  

  
# Manifest namespace
  
namespace = '{http://schemas.android.com/apk/res/android}'
  
# Exported activities
  
exported_activities = []
  

  

  
# XML analyze
  
def __xml_analyze__():
  
    tree = ET.parse(r"./app/AndroidManifest.xml")
  
    root = tree.getroot()
  
    # print tree.namespace
  
    # 1.get package name
  
    packagename = root.attrib["package"]
  
    # 2.get exported activity
  
    application = root.find("application")
  
    __get_exported_activities__(application, packagename)
  
    for activity in exported_activities:
  
      print activity.__get_name__(), activity.__get_package_name__(), activity.__get_permission__()
  
    # 3.save result to xml file
  
    __save_result_to_xml__()
  

  

  
# get exported activities
  
def __get_exported_activities__(application, packagename):
  
    for activity in application:
  
      # 2.1 MainActivity name
  
      if __is_main_activity__(activity):
  
            name = activity.attrib
  
            main_activtiy = ActivityInfo(name)
  
            main_activtiy.__set_package_name__(packagename)
  
            if activity.find(namespace+'permission') is not None:
  
                permission = activity.attrib
  
                main_activtiy.__set_permission__(permission)
  
            exported_activities.append(main_activtiy)
  
      else:
  
            # 2.1 exported activities except mainactivtiy
  
            exported = activity.attrib
  
            if exported:
  
                activity_name = activity.attrib
  
                exported_activity = ActivityInfo(activity_name)
  
                exported_activity.__set_package_name__(packagename)
  
                permission = activity.attrib.get(namespace+'permission', None)
  
                if permission is not None:
  
                  # permission = activity.attrib
  
                  # print permission
  
                  exported_activity.__set_permission__(permission)
  
                exported_activities.append(exported_activity)
  

  

  
# save result to xml file
  
def __save_result_to_xml__():
  
    tree = ET.ElementTree()
  
    root = ET.Element('data')
  
    tree._setroot(root)
  
    for activity in exported_activities:
  
      node = ET.Element('activity')
  
      node.set('name', activity.__get_name__())
  
      node.set('packagename', activity.__get_package_name__())
  
      root.append(node)
  
      if activity.__get_permission__() is not None:
  
            node.set('permission', activity.__get_permission__())
  
    tree.write("output.xml")
  

  

  
# the activity is or not MainActivity
  
def __is_main_activity__(activity):
  
    intent_filter = activity.find("intent-filter")
  
    if intent_filter is not None:
  
      action = intent_filter.find('action')
  
      category = intent_filter.find('category')
  
      if action is not None and category is not None \
  
                and action.attrib == 'android.intent.action.MAIN'\
  
                and category.attrib == 'android.intent.category.LAUNCHER':
  
            print 'this is main activity'
  
            return True
  
    return False
  

  

  
# main fun
  
if __name__ == "__main__":
  
    __xml_analyze__()
页: [1]
查看完整版本: Python——使用ElementTree解析AndroidManifest.xml文件