minor refactoring

This commit is contained in:
Miroslav Stampar 2012-04-05 12:55:26 +00:00
parent 02924eb345
commit 2223c884e5
3 changed files with 9 additions and 2 deletions

View File

@ -242,6 +242,13 @@ class Agent:
return payload
def adjustSleepTime(self, payload):
"""
Returns payload with a replaced tag for SLEEPTIME
"""
return payload.replace("[SLEEPTIME]", str(conf.timeSec)) if payload else payload
def getComment(self, request):
"""
Returns comment form for the given request

View File

@ -521,7 +521,7 @@ class Connect:
raise404 = place != PLACE.URI if raise404 is None else raise404
value = value.replace("[SLEEPTIME]", str(conf.timeSec)) if value else value
value = agent.adjustSleepTime(value)
payload = agent.extractPayload(value)
threadData = getCurrentThreadData()

View File

@ -28,7 +28,7 @@ from lib.utils.timeout import timeout
def direct(query, content=True):
select = True
query = agent.payloadDirect(query)
query = query.replace("[SLEEPTIME]", str(conf.timeSec))
query = agent.adjustSleepTime(query)
threadData = getCurrentThreadData()
if Backend.isDbms(DBMS.ORACLE) and query.startswith("SELECT ") and " FROM " not in query: