From 83a8e7cbe5a518beb0ac2d7af263ce1b6f116381 Mon Sep 17 00:00:00 2001 From: Noah Levitt Date: Tue, 21 Jul 2015 20:39:41 +0000 Subject: [PATCH] fix bug when --extra-header switch is not supplied --- bin/brozzler-add-site | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/bin/brozzler-add-site b/bin/brozzler-add-site index ca4c713..2386206 100755 --- a/bin/brozzler-add-site +++ b/bin/brozzler-add-site @@ -33,9 +33,10 @@ logging.basicConfig(stream=sys.stdout, level=args.log_level, format="%(asctime)s %(process)d %(levelname)s %(threadName)s %(name)s.%(funcName)s(%(filename)s:%(lineno)d) %(message)s") extra_headers = {} -for hh in args.extra_headers: - [k,v] = re.split(r":\s*", hh, 1) - extra_headers[k] = v +if args.extra_headers: + for hh in args.extra_headers: + [k,v] = re.split(r":\s*", hh, 1) + extra_headers[k] = v site = brozzler.Site(seed=args.seed, proxy=args.proxy, time_limit=int(args.time_limit) if args.time_limit else None,