Update robust.py

2022-11-17 06:57:59 +00:00 · 2022-11-17 06:57:59 +00:00 · 01acf672f4
parent f3d35a3dc3
commit 01acf672f4
1 changed files with 6 additions and 2 deletions
--- a/robust.py
+++ b/robust.py
@ -19,7 +19,11 @@ def parser():
 def runrobust(args):
 	found=[]
 	def bust(href):
-		headers={"cookie":" ".join(args.cookie),"user-agent":" ".join(args.uagent)}
+		headers={}
+		if args.cookie:
+			headers["cookie"]=" ".join(args.cookie)
+		if args.uagent:
+			headers["user-agent"]=" ".join(args.uagent)
 		try:
 			r=requests.get(href)
 			splitl=r.text.split("\n")
@ -62,5 +66,5 @@ def runrobust(args):
 			t.start()
 		for t in threads:
 			t.join()
-			
+
 parser()