5

すべての素数のセットのように振る舞うコンテナーを作成する次のコードがあります (実際にはメモ化されたブルート フォース プライム テストを隠します)。

import math

def is_prime(n):
    if n == 2 or n == 3:
        return True
    if n == 1 or n % 2 == 0:
        return False
    else:
        return all(n % i for i in xrange(3, int(1 + math.sqrt(n)), 2))


class Primes(object):

    def __init__(self):
        self.memo = {}

    def __contains__(self, n):
        if n not in self.memo:
            self.memo[n] = is_prime(n)
        return self.memo[n]

これまでのところうまくいっているようです：

>>> primes = Primes()
>>> 7 in primes
True
>>> 104729 in primes
True
>>> 100 in primes
False
>>> 100 not in primes
True

しかし、それはうまく機能していませんargparse:

>>> import argparse as ap
>>> parser = ap.ArgumentParser()
>>> parser.add_argument('prime', type=int, choices=primes, metavar='p')
_StoreAction(option_strings=[], dest='prime', nargs=None, const=None, default=None, type=<type 'int'>, choices=<__main__.Primes object at 0x7f4e21783f10>, help=None, metavar='p')
>>> parser.parse_args(['7'])
Namespace(prime=7)
>>> parser.parse_args(['11'])
Namespace(prime=11)
>>> parser.parse_args(['12'])
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib/python2.7/argparse.py", line 1688, in parse_args
    args, argv = self.parse_known_args(args, namespace)
  File "/usr/lib/python2.7/argparse.py", line 1720, in parse_known_args
    namespace, args = self._parse_known_args(args, namespace)
  File "/usr/lib/python2.7/argparse.py", line 1929, in _parse_known_args
    stop_index = consume_positionals(start_index)
  File "/usr/lib/python2.7/argparse.py", line 1885, in consume_positionals
    take_action(action, args)
  File "/usr/lib/python2.7/argparse.py", line 1778, in take_action
    argument_values = self._get_values(action, argument_strings)
  File "/usr/lib/python2.7/argparse.py", line 2219, in _get_values
    self._check_value(action, value)
  File "/usr/lib/python2.7/argparse.py", line 2267, in _check_value
    tup = value, ', '.join(map(repr, action.choices))
TypeError: argument 2 to map() must support iteration

ドキュメントはちょうどそれを言う

in 演算子をサポートするオブジェクトは、choices 値として渡すことができるため、dict オブジェクト、set オブジェクト、カスタム コンテナーなどはすべてサポートされます。

明らかに、素数の無限の「セット」を反復したくありません。では、一体なぜ私の素数をargparse試しているのでしょうか? とmapが必要なだけではありませんか？innot in

4

3 に答える 3

5

例外が発生しているソースは非常に明確です。確認してください。

if action.choices is not None and value not in action.choices:
    tup = value, ', '.join(map(repr, action.choices))
    msg = _('invalid choice: %r (choose from %s)') % tup
    raise ArgumentError(action, msg)

チェック自体は問題ありません。役立つエラー メッセージを出力しようとして、すべての可能な選択肢を表示しようとして失敗します。repr文字列として何かを返すようにイテレータを定義すると、primesそれをハックして正しいことを行うことができると思います。

于 2012-12-12T05:56:15.223 に答える
4

ドキュメントのバグでした。これはissue16468で、2019 年 8 月にPR 15566で修正されました。

書かれているライブラリでは、choices引数がコンテナだけでなく反復可能である必要があります。使用可能なオプションをリストしようとしますが、これはあなたのケースでは機能しません。__iter__何らかの情報文字列を返すだけの偽物を与えることで、ハッキングを試みることができます。

于 2012-12-12T06:00:32.727 に答える
1

choices許可されているすべての値（有限（小さな）セット）を列挙できる引数用です。ドキュメントはそれについてより明確にする必要があります。

primes無限集合です。type非プライムのValueErrorを発生させるパラメータを設定できます。

于 2012-12-12T06:15:07.603 に答える