Released version 0.1.5

Author: rohitcoder

hawk_scanner/commands/postgresql.py

@@ -24,9 +24,7 @@ def check_data_patterns(conn, patterns, profile_name, database_name):
     cursor = conn.cursor()
     cursor.execute("SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'")
     tables = [table[0] for table in cursor.fetchall()]
-
     table_count = 1
-
     results = []
     for table in tables:
         cursor.execute(f"SELECT * FROM {table}")

hawk_scanner/internals/system.py

@@ -17,6 +17,24 @@
 args, extra_args = parser.parse_known_args()
 
 
+def print_info(message):
+    console.print(f"[yellow][INFO][/yellow] {message}")
+
+def print_debug(message):
+    if args.debug:
+        console.print(f"[blue][DEBUG][/blue] {message}")
+
+def print_error(message):
+    console.print(f"[bold red]❌ {message}")
+
+def print_success(message):
+    console.print(f"[bold green]✅ {message}")
+
+def print_info(message):
+    console.print(f"[yellow][INFO][/yellow] {message}")
+def print_alert(message):
+    console.print(f"[bold red][ALERT][/bold red] {message}")
+
 def get_connection():
     if args.connection:
         if os.path.exists(args.connection):
@@ -56,24 +74,6 @@ def get_fingerprint_file():
 
 patterns = get_fingerprint_file()
 
-def print_info(message):
-    console.print(f"[yellow][INFO][/yellow] {message}")
-
-def print_debug(message):
-    if args.debug:
-        console.print(f"[blue][DEBUG][/blue] {message}")
-
-def print_error(message):
-    console.print(f"[bold red]❌ {message}")
-
-def print_success(message):
-    console.print(f"[bold green]✅ {message}")
-
-def print_info(message):
-    console.print(f"[yellow][INFO][/yellow] {message}")
-def print_alert(message):
-    console.print(f"[bold red][ALERT][/bold red] {message}")
-
 def print_banner():
     banner = r"""
                                 /T /I

requirements.txt

@@ -6,4 +6,4 @@ redis
 firebase-admin
 google-cloud-core
 google-cloud-storage
-psycopg2
+psycopg2-binary