no exit code as counter (#44)

Co-authored-by: umnik <[email protected]>

README.md

@@ -62,7 +62,7 @@ pip install -r requirements.txt
 
 ```bash
 $ ./exodus_analyze.py --help
-usage: exodus_analyze.py [-h] [-t] [-j] [-o OUTPUT_FILE] [-i IGNORE] apk
+usage: exodus_analyze.py [-h] [-t] [-j] [-o OUTPUT_FILE] [-i IGNORE] [-e CODE] apk
 
 positional arguments:
   apk                   the apk file to analyse
@@ -75,6 +75,8 @@ optional arguments:
                         store JSON report in file (requires -j option)
   -i IGNORE, --ignore IGNORE
                         comma-separated ids of trackers to ignore
+  -e CODE, --exit-code CODE
+                        use the CODE instead of trackers counter as exit code if trackers was detected
 ```
 
 #### Text output

exodus_analyze.py

@@ -58,7 +58,7 @@ def get_ignore_list(ignore_arg):
     return ignored, ''
 
 
-def analyze_apk(apk, json_mode, output_file, ignore_list):
+def analyze_apk(apk, json_mode, output_file, forced_code, ignore_list):
     analysis = AnalysisHelper(apk)
     analysis.load_trackers_signatures()
     if json_mode:
@@ -73,7 +73,10 @@ def analyze_apk(apk, json_mode, output_file, ignore_list):
         analysis.print_embedded_trackers()
 
     trackers_not_ignored = [t for t in analysis.detect_trackers() if t.id not in ignore_list]
-    sys.exit(len(trackers_not_ignored))
+    counter = len(trackers_not_ignored)
+    if all([counter, forced_code is not None]):
+        sys.exit(forced_code)
+    sys.exit(counter)
 
 
 def main():
@@ -105,6 +108,13 @@ def main():
         default=None,
         help='comma-separated ids of trackers to ignore'
     )
+    parser.add_argument(
+        '-e', '--exit-code',
+        metavar='CODE',
+        dest='override_code',
+        type=int,
+        help='use the CODE instead of trackers counter as exit code if trackers was detected'
+    )
 
     args = parser.parse_args()
     args_error = validate_arguments(args)
@@ -117,7 +127,7 @@ def main():
     if ignore_error:
         raise_error(parser, ignore_error)
 
-    analyze_apk(args.apk, args.json_mode, args.output_file, ignore_list)
+    analyze_apk(args.apk, args.json_mode, args.output_file, args.override_code, ignore_list)
 
 
 if __name__ == '__main__':