Skip to content

Commit 96a7209

Browse files
authored
Update port_scanner.py
1 parent bfa596c commit 96a7209

File tree

1 file changed

+15
-13
lines changed

1 file changed

+15
-13
lines changed

port_scanner.py

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -95,11 +95,15 @@ def run_nmap(args):
9595
styled_output = "\n".join(colored_output)
9696

9797
console.print(
98-
Panel(styled_output, title="nmap " + " ".join(args), border_style="cyan")
98+
Panel(styled_output, title="nmap " + " ".join(args), border_style="cyan", width=100)
9999
)
100100

101101

102-
def parse_args(parser):
102+
def parse_args():
103+
parser = argparse.ArgumentParser(
104+
prog=PROGRAM, description=DESCRIPTION, add_help=False
105+
)
106+
103107
parser.add_argument(
104108
"-v", "--version", action="version", version=f"%(prog)s version {VERSION}"
105109
)
@@ -112,25 +116,23 @@ def parse_args(parser):
112116
parser.add_argument(
113117
"-n", "--nmap", nargs=argparse.REMAINDER, help="run nmap with custom arguments"
114118
)
115-
116-
return parser.parse_args()
117-
118-
119-
def main():
120-
parser = argparse.ArgumentParser(
121-
prog=PROGRAM, description=DESCRIPTION, add_help=False
122-
)
123-
args = parse_args(parser)
124119

125-
if len(sys.argv) == 1 or args.help:
120+
if len(sys.argv) == 1 or '--help' in sys.argv or '-h' in sys.argv:
126121
console.print(
127122
Panel(
128123
parser.format_help(),
129124
title=" ".join(sys.argv),
130125
border_style="cyan",
126+
width=80,
131127
)
132128
)
133-
return
129+
sys.exit()
130+
131+
return parser.parse_args()
132+
133+
134+
def main():
135+
args = parse_args()
134136

135137
if args.update:
136138
update()

0 commit comments

Comments
 (0)