patch for headers via arguments

This commit is contained in:
Somdev Sangwan
2019-01-30 12:04:40 +05:30
committed by GitHub
parent 34eaba739a
commit 479700cc5a

View File

@@ -46,17 +46,19 @@ parser.add_argument('-t', help='number of threads', dest='threads', type=int)
parser.add_argument('-l', help='levels to crawl', dest='level', type=int) parser.add_argument('-l', help='levels to crawl', dest='level', type=int)
parser.add_argument('--delay', help='delay between requests', dest='delay', type=int) parser.add_argument('--delay', help='delay between requests', dest='delay', type=int)
parser.add_argument('--timeout', help='http request timeout', dest='timeout', type=int) parser.add_argument('--timeout', help='http request timeout', dest='timeout', type=int)
parser.add_argument('--headers', help='http headers', dest='headers', action='store_true') parser.add_argument('--headers', help='http headers', dest='add_headers', nargs='?', const=True)
args = parser.parse_args() args = parser.parse_args()
if not args.target: if not args.target:
print('\n' + parser.format_help().lower()) print('\n' + parser.format_help().lower())
quit() quit()
if args.headers: if type(args.add_headers) == bool:
headers = extractHeaders(prompt()) headers = extractHeaders(prompt())
elif type(args.add_headers) == str:
headers = extractHeaders(args.add_headers)
else: else:
headers = core.config.headers from core.config import headers
target = args.target target = args.target
delay = args.delay or 0 delay = args.delay or 0