Skip to content

Commit 0a039d8

Browse files
committed
some more refactoring
1 parent d9d4e3e commit 0a039d8

File tree

9 files changed

+93
-130
lines changed

9 files changed

+93
-130
lines changed

lib/core/agent.py

Lines changed: 9 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -215,16 +215,13 @@ def cleanupPayload(self, payload, origValue=None):
215215
randStr = randomStr()
216216
randStr1 = randomStr()
217217

218-
payload = payload.replace("[RANDNUM]", str(randInt))
219-
payload = payload.replace("[RANDNUM1]", str(randInt1))
220-
payload = payload.replace("[RANDSTR]", randStr)
221-
payload = payload.replace("[RANDSTR1]", randStr1)
222-
payload = payload.replace("[DELIMITER_START]", kb.chars.start)
223-
payload = payload.replace("[DELIMITER_STOP]", kb.chars.stop)
224-
payload = payload.replace("[AT_REPLACE]", kb.chars.at)
225-
payload = payload.replace("[SPACE_REPLACE]", kb.chars.space)
226-
payload = payload.replace("[DOLLAR_REPLACE]", kb.chars.dollar)
227-
payload = payload.replace("[SLEEPTIME]", str(conf.timeSec))
218+
_ = (
219+
("[RANDNUM]", str(randInt)), ("[RANDNUM1]", str(randInt1)), ("[RANDSTR]", randStr),\
220+
("[RANDSTR1]", randStr1), ("[DELIMITER_START]", kb.chars.start), ("[DELIMITER_STOP]", kb.chars.stop),\
221+
("[AT_REPLACE]", kb.chars.at), ("[SPACE_REPLACE]", kb.chars.space), ("[DOLLAR_REPLACE]", kb.chars.dollar),\
222+
("[SLEEPTIME]", str(conf.timeSec))
223+
)
224+
payload = reduce(lambda x, y: x.replace(y[0], y[1]), _, payload)
228225

229226
if origValue is not None:
230227
payload = payload.replace("[ORIGVALUE]", origValue)
@@ -249,13 +246,8 @@ def cleanupPayload(self, payload, origValue=None):
249246

250247
return payload
251248

252-
def getComment(self, reqObj):
253-
if "comment" in reqObj:
254-
comment = reqObj.comment
255-
else:
256-
comment = ""
257-
258-
return comment
249+
def getComment(self, request):
250+
return request.comment if "comment" in request else ""
259251

260252
def nullAndCastField(self, field):
261253
"""

0 commit comments

Comments
 (0)