update urlclean.mixPayload

Author: Tr3jer

docs/urlclean.rst

@@ -74,10 +74,11 @@ pathSplit
 mixPayload
 ----------
 
-.. Method:: Method mixPayload(url, payloads, scope=[])
+.. Method:: Method mixPayload(url, payloads, scope=[], append=True)
 
 * **payloads** - [payload list]
 * **scope** - [path, params, query, fragment], Default By **All**
+* **append** - append payload for params/query/fragment, Default is **False**
 
 ::
 
@@ -91,3 +92,11 @@ mixPayload
     'https://www.example.com/path/index;params?a=1&b=-Symbo1-&b=2#fragment',
     'https://www.example.com/path/index;params?a=1&b=3&c=-Symbo1-#fragment']
 
+    # use append:
+    In[4]: mixPayload(url, payloads, ['query', 'params', 'fragement'], append=True)
+    Out[4]: ['https://www.example.com/path/index;params?a=1&c=3-symbo1-&b=2#fragment',
+    'https://www.example.com/path/index;params-symbo1-?a=1&b=2&c=3#fragment',
+    'https://www.example.com/path/index;params?a=1&b=2&c=3#fragment-symbo1-',
+    'https://www.example.com/path/index;params?a=1&c=3&b=2-symbo1-#fragment',
+    'https://www.example.com/path/index;params?a=1-symbo1-&c=3&b=2#fragment']
+

setup.py

@@ -7,7 +7,7 @@
     long_description = fh.read()
 
 setup(name="wsltools",
-      version="0.2.2",
+      version="0.2.3",
       description="Web Scan Lazy Tools",
       long_description=long_description,
       long_description_content_type="text/markdown",

wsltools/urlclean.py

@@ -155,7 +155,7 @@ def mixPayload(url, payloads, scope=[], append=False):
 	:param url: Generic URL.
 	:param payloads: Payloads list
 	:param scope: [optional] list, e.g. [params, path, query, fragment]
-	:param append: append payload for query and fragment, default is replace
+	:param append: append payload for params/query/fragment, default is replace
 	:return: list
 	'''
 
@@ -171,7 +171,10 @@ def mixPayload(url, payloads, scope=[], append=False):
 
 	for p in payloads:
 		if (not scope or 'params' in scope) and u.params:
-			result.add(urlunparse([u.scheme, u.netloc, u.path, p, u.query, u.fragment]))
+			if append:
+				result.add(urlunparse([u.scheme, u.netloc, u.path, u.params + p, u.query, u.fragment]))
+			else:
+				result.add(urlunparse([u.scheme, u.netloc, u.path, p, u.query, u.fragment]))
 
 
 		if not scope or 'path' in scope:
@@ -187,10 +190,8 @@ def mixPayload(url, payloads, scope=[], append=False):
 		if not scope or 'query' in scope:
 			for k in qs_dict.keys():
 				tmp_dict = copy.deepcopy(qs_dict)
-				if append:
-					tmp_dict[k] = tmp_dict[k] + p
-				else:
-					tmp_dict[k] = p
+				tmp_dict[k] = tmp_dict[k] + p if append else p
+
 				tmp_qs = unquote(urlencode(tmp_dict))
 
 				result.add(urlunparse([u.scheme, u.netloc, u.path, u.params, tmp_qs, u.fragment]))