Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 4984cea

Browse files
committed
some code refactoring and minor speed up (jump prediction rule)
1 parent ed20f1c commit 4984cea

1 file changed

Lines changed: 10 additions & 22 deletions

File tree

lib/request/connect.py

Lines changed: 10 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -285,35 +285,23 @@ def queryPage(value=None, place=None, content=False, getSeqMatcher=False, silent
285285
if not place:
286286
place = kb.injPlace
287287

288-
if conf.parameters.has_key("GET"):
289-
if place == "GET" and value:
290-
get = value
291-
else:
292-
get = conf.parameters["GET"]
288+
if "GET" in conf.parameters:
289+
get = conf.parameters["GET"] if place != "GET" or not value else value
293290

294-
if conf.parameters.has_key("POST"):
295-
if place == "POST" and value:
296-
post = value
297-
else:
298-
post = conf.parameters["POST"]
291+
if "POST" in conf.parameters:
292+
post = conf.parameters["POST"] if place != "POST" or not value else value
299293

300-
if conf.parameters.has_key("Cookie"):
301-
if place == "Cookie" and value:
302-
cookie = value
303-
else:
304-
cookie = conf.parameters["Cookie"]
294+
if "Cookie" in conf.parameters:
295+
cookie = conf.parameters["Cookie"] if place != "Cookie" or not value else value
296+
297+
if "User-Agent" in conf.parameters:
298+
ua = conf.parameters["User-Agent"] if place != "User-Agent" or not value else value
305299

306-
if conf.parameters.has_key("User-Agent"):
307-
if place == "User-Agent" and value:
308-
ua = value
309-
else:
310-
ua = conf.parameters["User-Agent"]
311-
312300
if conf.safUrl and conf.saFreq > 0:
313301
kb.queryCounter += 1
314302
if kb.queryCounter % conf.saFreq == 0:
315303
Connect.getPage(url=conf.safUrl, cookie=cookie, direct=True, silent=True, ua=ua)
316-
304+
317305
page, headers = Connect.getPage(get=get, post=post, cookie=cookie, ua=ua, silent=silent)
318306

319307
if content:

0 commit comments

Comments
 (0)