From 662e299bbaadb60e8e27cde7933aa16491a5985b Mon Sep 17 00:00:00 2001 From: Rohit kumar Date: Thu, 18 Jul 2024 14:08:36 +0530 Subject: [PATCH] Releasing exclude_columns support in mysql --- hawk_scanner/internals/system.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/hawk_scanner/internals/system.py b/hawk_scanner/internals/system.py index 6ac4a56..c7aa4d0 100644 --- a/hawk_scanner/internals/system.py +++ b/hawk_scanner/internals/system.py @@ -12,11 +12,13 @@ import shutil import os, cv2 import tarfile +import pkg_resources data_sources = ['s3', 'mysql', 'redis', 'firebase', 'gcs', 'fs', 'postgresql', 'mongodb', 'slack', 'couchdb', 'gdrive', 'gdrive_workspace', 'text'] data_sources_option = ['all'] + data_sources def parse_args(args=None): + version = pkg_resources.require("hawk_scanner")[0].version parser = argparse.ArgumentParser(description='🦅 A powerful scanner to scan your Filesystem, S3, MySQL, PostgreSQL, MongoDB, Redis, Google Cloud Storage and Firebase storage for PII and sensitive data.') parser.add_argument('command', nargs='?', choices=data_sources_option, help='Command to execute') parser.add_argument('--connection', action='store', help='YAML Connection file path') @@ -28,6 +30,7 @@ def parse_args(args=None): parser.add_argument('--debug', action='store_true', help='Enable debug mode') parser.add_argument('--no-write', action='store_true', help='Do not write previous alerts to file, this may flood you with duplicate alerts') parser.add_argument('--shutup', action='store_true', help='Suppress the Hawk Eye banner 🫣', default=False) + parser.add_argument('--version', action='version', version='%(prog)s v' + version) parser.add_argument('--hawk-thuu', action='store_true', help="Delete all spitted files during testing phase forcefully") return parser.parse_args(args)