Issue #55: mult operator (*) in argument proxies.

dirhunt/management.py

@@ -14,7 +14,8 @@
 from dirhunt.exceptions import DirHuntError, catch
 from dirhunt.output import output_urls
 from dirhunt.sources import SOURCE_CLASSES, get_source_name
-from dirhunt.utils import lrange, catch_keyboard_interrupt, force_url, read_file_lines, value_is_file_path, flat_list
+from dirhunt.utils import lrange, catch_keyboard_interrupt, force_url, read_file_lines, value_is_file_path, flat_list, \
+    multiplier_args
 from colorama import init
 
 init(autoreset=True)
@@ -134,6 +135,7 @@ def hunt(urls, threads, exclude_flags, include_flags, interesting_extensions, in
         raise BadOptionUsage('--exclude-flags and --include-flags are mutually exclusive.')
     welcome()
     urls = flat_list(urls)
+    proxies = multiplier_args(proxies)
     if not urls:
         click.echo('•_•) OOPS! Add urls to analyze.\nFor example: dirhunt http://domain/path\n\n'
                    'Need help? Then use dirhunt --help', err=True)

dirhunt/tests/test_utils.py

@@ -3,7 +3,7 @@
 import requests_mock
 
 from dirhunt.tests._compat import patch, Mock
-from dirhunt.utils import force_url, SCHEMES, catch_keyboard_interrupt, flat_list
+from dirhunt.utils import force_url, SCHEMES, catch_keyboard_interrupt, flat_list, multiplier_args
 
 
 class TestForceUrl(unittest.TestCase):
@@ -49,3 +49,14 @@ def test_without_sublists(self):
 
     def test_with_sublists(self):
         self.assertEqual(flat_list([1, [2, 3], 4]), [1, 2, 3, 4])
+
+
+class TestMultiplierArgs(unittest.TestCase):
+    def test_without_multiplier(self):
+        self.assertEqual(multiplier_args(['foo', 'bar']), ['foo', 'bar'])
+
+    def test_multiplier(self):
+        self.assertEqual(multiplier_args(['foo', 'bar*3']), ['foo'] + (['bar'] * 3))
+
+    def test_invalid_multiplier(self):
+        self.assertEqual(multiplier_args(['foo', 'bar*spam']), ['foo', 'bar*spam'])

dirhunt/utils.py

@@ -13,6 +13,7 @@
 
 SCHEMES = ['http', 'https']
 DEFAULT_SCHEME = 'http'
+ARGUMENT_MULT = re.compile('(.+)\*(\d+)$')
 
 
 def lrange(start, end):
@@ -89,3 +90,14 @@ def flat_list(values):
         else:
             items.append(value)
     return items
+
+
+def multiplier_arg(argument):
+    matchs = ARGUMENT_MULT.match(argument)
+    if matchs is None:
+        return argument
+    return [matchs.group(1)] * int(matchs.group(2))
+
+
+def multiplier_args(arguments):
+    return flat_list([multiplier_arg(argument) for argument in arguments])