web-dev-qa-db-ja.com

bool引数をfabricコマンドに渡す方法

現在、私はfab -f check_remote.py func:"arg1","arg2"...を使用してfabリモートを実行しています。

ここでブール引数を送信する必要がありますが、True/Falseは文字列引数になります。ブール型として設定するにはどうすればよいですか？

pythonbooleanfabric
22
whi

私はこれを使用しています：

from distutils.util import strtobool

def func(arg1="default", arg2=False):
    if arg2:
        arg2 = bool(strtobool(arg2))

これまでのところ私のために働きます。値を解析します（大文字と小文字は区別されません）：

'y', 'yes', 't', 'true', 'on', '1'
'n', 'no', 'f', 'false', 'off', '0'

strtoboolは0または1を返すため、True/Falsebooleanに変換するにはboolが必要です。

完全を期すために、ここにstrtoboolの実装があります。

def strtobool (val):
    """Convert a string representation of truth to true (1) or false (0).

    True values are 'y', 'yes', 't', 'true', 'on', and '1'; false values
    are 'n', 'no', 'f', 'false', 'off', and '0'.  Raises ValueError if
    'val' is anything else.
    """
    val = val.lower()
    if val in ('y', 'yes', 't', 'true', 'on', '1'):
        return 1
    Elif val in ('n', 'no', 'f', 'false', 'off', '0'):
        return 0
    else:
        raise ValueError("invalid truth value %r" % (val,))

少し良いバージョン（コメントmVChrに感謝）

from distutils.util import strtobool

def _prep_bool_arg(arg): 
    return bool(strtobool(str(arg)))

def func(arg1="default", arg2=False):
    arg2 = _prep_bool_arg(arg2)
40
Engrost

私は関数を使用します：

def booleanize(value):
    """Return value as a boolean."""

    true_values = ("yes", "true", "1")
    false_values = ("no", "false", "0")

    if isinstance(value, bool):
        return value

    if value.lower() in true_values:
        return True

    Elif value.lower() in false_values:
        return False

    raise TypeError("Cannot booleanize ambiguous value '%s'" % value)

次に、タスクで：

@task
def mytask(arg):
    arg = booleanize(arg)
14
charlax

問題の関数が「ifargNis True：」ではなく「ifargN：」を使用してブール値がtrueかどうかをテストする場合、Falseには「」、Trueには「anything」を使用できます。

参照： http://docs.python.org/library/stdtypes.html#truth-value-testing

5
Ari

すべてのタスクで一貫してパターンを使用する場合（「false」、「true」はブール値）、ファブリックタスクをラップしてすべてに適用できます。

このパッケージ（私が書いたもの）を使用できます： https://pypi.python.org/pypi/boolfab/

これが（本質的に）ソースです：

from fabric.api import task as _task

def fix_boolean(f):
    def fix_bool(value):
        if isinstance(value, basestring):
            if value.lower() == 'false':
                return False
            if value.lower() == 'true':
                return True
        return value

    @wraps(f)
    def wrapper(*args, **kwargs):
        args_ = [fix_bool(arg) for arg in args]
        kwargs_ = {k: fix_bool(v) for k,v in kwargs.iteritems()}
        return f(*args_, **kwargs_)

    return wrapper


def task(f):
    return _task(fix_boolean(f))

そのため、次のようになります。

@task
def my_task(flag_a, flag_b, flag_c)
   if flag_a:
       ....

'booleanizing'引数で各タスクを汚染することなく。

5
j-a

より良い方法は、 ast.literal_eval を使用することです。

from ast import literal_eval

def my_task(flag):
    if isinstance(flag, basestring): # also supports calling from other tasks
        flag = literal_eval(flag)

これは「yes」や「no」などの値を考慮していませんが、eval ..よりも少しクリーンで安全です。

3
vmalloc

ユーザーが「False」を渡すと、CraigとAriの回答はTrue値になります（Ariの回答はこれについてより明確です）

Eval（）を使用する場合、文字列「True」と「False」は正しいブール値に評価されますが、デフォルト値を使用する場合は、それらがブール値ではなく文字列であることを確認する必要があります。

def myfunc(arg1="True", arg2=False):
    arg1 = eval(arg1)
    arg2 = eval(arg2) #error
1
AnalogBrain

デコレータを使用してこれを解決しました。デコレータを使用することで得られる柔軟性と明確さが気に入っています。

コードの要点は次のとおりです。

import ast
from fabric import utils
from fabric.api import task
from functools import wraps


def params(*types, **kwtypes):
    def decorator(function):
        @wraps(function)
        def wrapper(*args, **kwargs):
            new_args = ()
            for index, arg in enumerate(args):
                new_args += __cast(arg, types[index]),
            for kwarg in kwargs:
                kwargs[kwarg] = __cast(kwargs[kwarg], kwtypes[kwarg])
            return function(*new_args, **kwargs)
        return wrapper
    return decorator


def __evaluate(arg):
    try:
        return ast.literal_eval(arg)
    except:
        return str(arg)


def __cast(arg, arg_type):
    try:
        return arg_type(__evaluate(arg))
    except:
        utils.abort("Unable to cast '{}' to {}".format(arg, arg_type))

コードで使用すると、次のようになります。

@task
@params(int, bool, arg1=int, arg2=bool)
def test(arg1, arg2):
    print type(arg1), arg1
    print type(arg2), arg2

これは、適切なパラメータを使用してfab経由で呼び出すとどのように見えるかを示しています。

fab test:0.1,1
<type 'int'> 0
<type 'bool'> True

fab test:5,arg2=False
<type 'int'> 5
<type 'bool'> False

fab test:arg1=0,arg2=false
<type 'int'> 5
<type 'bool'> True

注：最後の例では、「false」はTrueです。これは、Pythonで予想される動作ですが、当然、直感に反する場合があります。 Falseをintとして渡すのと同様に、Pythonではint（False）== 0として0にキャストされます。

これは、パラメータが不適切なfab経由で呼び出すとどのように見えるかを示しています。

fab test:Test,False
Fatal error: Unable to cast 'Test' to <type 'int'>

Aborting.
1
unscannable

私のfabfilesで私はただします：

TRUTHY = [True, 1, '1', 'true', 't', 'yes', 'y']

@task
def my_task(my_arg=True):
    if my_arg in TRUTHY:
         # do stuff
    else:
         # do other stuff

もちろん、これはTRUTHYにない値は事実上Falseであることを意味しますが、これまでのところ、これ以上複雑なものは必要ありません。

1
katy lavallee

これは https://Gist.github.com/mgedmin/f832eed2ac0f3ce31edf に基づく動作バージョンです。古いバージョンとは異なり、これは実際にはすべての可能なデコレータパラメータとタスクエイリアスを尊重します。

from functools import wraps
from fabric import tasks

def fix_boolean(f):

    true_values = ("yes", "true", "1")
    false_values = ("no", "false", "0")

    def fix_bool(value):
        if isinstance(value, basestring):
            if value.lower() in false_values:
                return False
            if value.lower() in true_values:
                return True
        return value

    @wraps(f)
    def wrapper(*args, **kwargs):
        args_ = [fix_bool(arg) for arg in args]
        kwargs_ = {k: fix_bool(v) for k,v in kwargs.iteritems()}
        return f(*args_, **kwargs_)

    return wrapper


def task(*args, **kwargs):
    """
    The fabric.decorators.task decorator which automatically converts command line task arguments
    to a boolean representation if applicable.
    :param args:
    :param kwargs:
    :return: wrapped
    """
    invoked = bool(not args or kwargs)
    task_class = kwargs.pop("task_class", tasks.WrappedCallableTask)

    def wrapper(f):
        return task_class(fix_boolean(f), *args, **kwargs)

    return wrapper if invoked else wrapper(args[0])

要旨： https://Gist.github.com/eltismerino/a8ec8584034c8a7d087e

0
panni