Skip to content

Commit a1ed392

Browse files
rohitcoderrohitcoder
committed
fixed printing large values
1 parent 9c06f99 commit a1ed392

File tree

3 files changed

+20
-16
lines changed

3 files changed

+20
-16
lines changed

‎hawk_scanner/internals/system.py‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,9 @@ def match_strings(content):
177177

178178
if 'notify' in connections:
179179
redacted: bool = connections.get('notify', {}).get('redacted', False)
180-
180+
else:
181+
redacted = False
182+
181183
for pattern_name, pattern_regex in patterns.items():
182184
print_debug(f"Matching pattern: {pattern_name}")
183185
found = {}
@@ -242,6 +244,7 @@ def read_match_strings(file_path, source):
242244
with open(file_path, 'r', encoding="utf-8") as file:
243245
content = file.read()
244246
except Exception as e:
247+
print_debug(f"Error in read_match_strings: {e}")
245248
pass
246249
matched_strings = match_strings(content)
247250
return matched_strings

‎hawk_scanner/main.py‎

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -103,14 +103,15 @@ def main():
103103
i = 1
104104

105105
for result in grouped_results[group]:
106+
records_mini = ', '.join(result['matches']) if len(result['matches']) < 25 else ', '.join(result['matches'][:25]) + f" + {len(result['matches']) - 25} more records"
106107
if group == 's3':
107108
table.add_row(
108109
str(i),
109110
result['profile'],
110111
f"{result['bucket']} > {result['file_path']}",
111112
result['pattern_name'],
112113
str(len(result['matches'])),
113-
str(', '.join(result['matches'])),
114+
str(records_mini),
114115
result['sample_text'],
115116
)
116117
AlertMsg = """
@@ -127,7 +128,7 @@ def main():
127128
file_path=result['file_path'],
128129
pattern_name=result['pattern_name'],
129130
total_exposed=str(len(result['matches'])),
130-
exposed_values=', '.join(result['matches'])
131+
exposed_values=records_mini
131132
)
132133

133134
system.SlackNotify(AlertMsg)
@@ -139,7 +140,7 @@ def main():
139140
f"{result['host']} > {result['database']} > {result['table']}.{result['column']}",
140141
result['pattern_name'],
141142
str(len(result['matches'])),
142-
str(', '.join(result['matches'])),
143+
records_mini,
143144
result['sample_text'],
144145
)
145146

@@ -162,7 +163,7 @@ def main():
162163
column=result['column'],
163164
pattern_name=result['pattern_name'],
164165
total_exposed=str(len(result['matches'])),
165-
exposed_values=', '.join(result['matches'])
166+
exposed_values=records_mini
166167
)
167168

168169
system.SlackNotify(AlertMsg)
@@ -174,7 +175,7 @@ def main():
174175
f"{result['host']} > {result['database']} > {result['collection']} > {result['field']}",
175176
result['pattern_name'],
176177
str(len(result['matches'])),
177-
str(', '.join(result['matches'])),
178+
records_mini,
178179
result['sample_text'],
179180
)
180181

@@ -197,7 +198,7 @@ def main():
197198
field=result['field'],
198199
pattern_name=result['pattern_name'],
199200
total_exposed=str(len(result['matches'])),
200-
exposed_values=', '.join(result['matches'])
201+
exposed_values=records_mini
201202
)
202203

203204
system.SlackNotify(AlertMsg)
@@ -209,7 +210,7 @@ def main():
209210
f"{result['host']} > {result['database']} > {result['table']}.{result['column']}",
210211
result['pattern_name'],
211212
str(len(result['matches'])),
212-
str(', '.join(result['matches'])),
213+
records_mini,
213214
result['sample_text'],
214215
)
215216

@@ -232,7 +233,7 @@ def main():
232233
column=result['column'],
233234
pattern_name=result['pattern_name'],
234235
total_exposed=str(len(result['matches'])),
235-
exposed_values=', '.join(result['matches'])
236+
exposed_values=records_mini
236237
)
237238

238239
system.SlackNotify(AlertMsg)
@@ -244,7 +245,7 @@ def main():
244245
f"{result['host']} > {result['key']}",
245246
result['pattern_name'],
246247
str(len(result['matches'])),
247-
str(', '.join(result['matches'])),
248+
records_mini,
248249
result['sample_text'],
249250
)
250251
AlertMsg = """
@@ -261,7 +262,7 @@ def main():
261262
key=result['key'],
262263
pattern_name=result['pattern_name'],
263264
total_exposed=str(len(result['matches'])),
264-
exposed_values=', '.join(result['matches'])
265+
exposed_values=records_mini
265266
)
266267

267268
system.SlackNotify(AlertMsg)
@@ -272,7 +273,7 @@ def main():
272273
f"{result['bucket']} > {result['file_path']}",
273274
result['pattern_name'],
274275
str(len(result['matches'])),
275-
str(', '.join(result['matches'])),
276+
records_mini,
276277
result['sample_text'],
277278
)
278279

@@ -291,7 +292,7 @@ def main():
291292
file_path=result['file_path'],
292293
pattern_name=result['pattern_name'],
293294
total_exposed=str(len(result['matches'])),
294-
exposed_values=', '.join(result['matches'])
295+
exposed_values=records_mini
295296
)
296297

297298
system.SlackNotify(AlertMsg)
@@ -303,7 +304,7 @@ def main():
303304
f"{result['file_path']}",
304305
result['pattern_name'],
305306
str(len(result['matches'])),
306-
str(', '.join(result['matches'])),
307+
records_mini,
307308
result['sample_text'],
308309
)
309310
AlertMsg = """
@@ -324,7 +325,7 @@ def main():
324325
file_path=result['file_path'],
325326
pattern_name=result['pattern_name'],
326327
total_exposed=str(len(result['matches'])),
327-
exposed_values=str(', '.join(result['matches']))
328+
exposed_values=records_mini
328329
)
329330
system.SlackNotify(AlertMsg)
330331
else:

‎setup.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
VERSION = "0.2.4"
1+
VERSION = "0.2.5"
22

33
from setuptools import setup, find_packages
44

0 commit comments

Comments
 (0)