Skip to content

Commit 429c365

Browse files
committed
refactor: Improve default filename generation in get_common_inputs function for cidr ranges
1 parent f748e70 commit 429c365

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

bugscanx/modules/scanners_pro/host_scanner_pro.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,12 @@ def read_hosts(filename=None, cidr=None):
1717
return []
1818

1919
def get_common_inputs(input_source):
20-
output = get_input("Enter output filename", default=f"result_{os.path.basename(str(input_source))}", validate_input=False)
20+
if isinstance(input_source, str) and '/' in input_source:
21+
first_cidr = input_source.split(',')[0].strip()
22+
default_filename = f"result_{first_cidr.replace('/', '-')}.txt"
23+
else:
24+
default_filename = f"result_{os.path.basename(str(input_source))}"
25+
output = get_input("Enter output filename", default=default_filename, validate_input=False)
2126
threads = get_input("Enter threads", "number", default="50", allow_comma_separated=False)
2227
return output, threads
2328

0 commit comments

Comments
 (0)