Python 多次使用相同的参数保留为字符串

Question

我是 python 的新手，正在尝试弄清楚如何让这段代码接受多个相同的参数 -ex。我试着把它变成 action='append' 但这给了我一个错误，说它需要是一个字符串而不是一个列表。现在多次使用 -ex 它只使用它的最后一个实例例子。 python3 scrunch.py --no-rep -ex 123 -ex 456 5 5 abcdefg123456 输出 12345 输出仍然包含 123.

import string
import argparse
import itertools
from math import factorial



def number_permutations(n, k):
 return factorial(n)/factorial(n-k)

def main():
    parser = argparse.ArgumentParser(description='Generate password combinations from charset'
    'with length in range: [MIN..MAX]. '
    'If any of -u, -l, -d, -hex option is set, then charset is ignored, but still must be provided.')

    parser = argparse.ArgumentParser(epilog="Example 1: "
    'python3 scrunch.py --no-rep -ex ABC 3 5 ABCDEFGH123456 '
    'Example 2: '
    'python3 scrunch.py --no-rep -f wordlist.txt -u -l -d 6 6 DUMMYCHARSET')

    parser.add_argument('-f', metavar='file', help='output text file name. Prints to STDOUT by default')
    parser.add_argument('-u', action='store_true', help='Use all uppercase characters. Overrides charset')
    parser.add_argument('-l', action='store_true', help='Use all lowercase characters. Overrides charset')
    parser.add_argument('-d', action='store_true', help='Use all decimal numbers. Overrides charset')
    parser.add_argument('-hex', action='store_true', help='Use all hex numbers. Overrides charset')
    parser.add_argument('-ex', help='Exclude char sequence')
    parser.add_argument('--no-digit-start', action='store_true', help='Do not start with digit')
    parser.add_argument('minlen', metavar='MIN', type=int,
                        help='Min length of combination')
    parser.add_argument('maxlen', metavar='MAX', type=int,
                        help='Max length of combination')
    parser.add_argument('charset', nargs=1)
    parser.add_argument('--no-rep', action='store_true', help='Any symbol can occur only once.')

    args = parser.parse_args()
    print (args.ex)
    a = ''.join(args.charset)
    b = ''
    if args.u:
     b += string.ascii_uppercase
    if args.l:
     b += string.ascii_lowercase
    if args.hex:
     b += string.hexdigits
    if args.d:
     b += string.digits

    if b:
      a = b

    s = sorted(a)
    print ("Sorted charset: {}".format(s))
    num = 0
    # Calculate number of possible combinations
    for L in range(args.minlen, args.maxlen + 1):
     num += number_permutations(len(s), L)
    print ("Number of permutations: %d" % num)

    if args.f:
     f = open(args.f, 'w')

    for L in range(args.minlen, args.maxlen + 1):
     if args.no_digit_start and args.no_rep:
      gen = permutations_no_digit_start(s, L)
     elif args.no_rep:
      gen = itertools.permutations(s, L)
     else:
      gen  = itertools.product(s, repeat=L)
     for x in gen:
      y = ''.join(x)
      if args.ex and args.ex in y:
       continue
      if args.no_digit_start and y[0].isdigit():
       continue
      if args.f:
       f.write("%s\n" % ''.join(x))
      else:
       print (''.join(x))

main()

Answer 1

替换行

parser.add_argument('-ex', help='Exclude char sequence')

至

parser.add_argument('-ex', action='append', help='Exclude char sequence')

替换if检查

if args.ex and args.ex in y:
  continue

至

bad_pattern = False
for ex in args.ex:
  if ex and ex in y:
    bad_pattern = True
    break
if bad_pattern:
  continue

说明：默认只使用-ex的最后一个值。那就是 456。但是 action='append' 允许设置值列表。每次使用 -ex 都会将新值附加到列表中。之后，您必须通过迭代 for ex in args.ex

来检查 args.ex 列表中的每个值

Python 多次使用相同的参数保留为字符串

Python use same argument several times keep as string

python

argparse