From 19851a88eb0f7031ff4dbe320cd01bf1057c5f54 Mon Sep 17 00:00:00 2001 From: JohnE Date: Fri, 26 Feb 2016 01:18:17 -0800 Subject: [PATCH] FIX: the path parsing and the database file name have improved parsing --- libs/android.py | 1 + libs/fingerprint.py | 6 ++++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/libs/android.py b/libs/android.py index 9e65fe5..0d5f8d2 100644 --- a/libs/android.py +++ b/libs/android.py @@ -24,6 +24,7 @@ class AndroidAppPull: if (0 < stdout.find('failed')): logging.error("Error executing adb shell") logging.info("Make sure adb is in root mode, 'adb root'") + return dir_names strings = stdout.split('\n') for sstr in strings: diff --git a/libs/fingerprint.py b/libs/fingerprint.py index b3e66e6..18ae230 100644 --- a/libs/fingerprint.py +++ b/libs/fingerprint.py @@ -1,6 +1,7 @@ # # # +import os import re import json import sqlite3 @@ -73,8 +74,9 @@ class FingerprintDB: try: # extract file name from path+filename - self.metadata['db-name'] = ToolBox.parseFilenameIncExt(filein) - except: + self.metadata['db-name'] = os.path.basename(filein) + except Exception as ex: + logging.error(ex) self.metadata['db-name'] = filein try: # read database schema, parse the schema