diff --git a/droidbot/app.py b/droidbot/app.py index dfb25795..f89bf764 100644 --- a/droidbot/app.py +++ b/droidbot/app.py @@ -25,7 +25,10 @@ def __init__(self, app_path, output_dir=None): if not os.path.isdir(output_dir): os.makedirs(output_dir) - from androguard.core.bytecodes.apk import APK + try: + from androguard.core.bytecodes.apk import APK + except ImportError: + from androguard.core.apk import APK self.apk = APK(self.app_path) self.package_name = self.apk.get_package() self.app_name = self.apk.get_app_name() diff --git a/setup.py b/setup.py index 2a18c5c6..55dd5d85 100644 --- a/setup.py +++ b/setup.py @@ -34,7 +34,7 @@ ], entry_points={ 'console_scripts': [ - 'droidbot=start:main', + 'droidbot=droidbot.start:main', ], }, package_data={