验证码

时间:2012-07-30 22:51:53

标签: python validation

我正在尝试验证字符串的有效性,以确保它是我可以传递给终端的合法命令。如果字符串通过测试,我返回True。否则,我返回False和错误消息。

我的代码非常难看,有很多嵌套的if语句 - 我该如何改进呢?

task = task.split()
if len(task) > 1: 
    if task[0] == 'svn':
        if task[1] in ALLOWED:
            if len(task[2:]) == ALLOWED[task[1]]:
                return True, task, None
            else:
                return False, "Incorrect number of arguments."
        else:
            return False, "Not a legal command."    
    else:
        return False, "Not a subversion command."
else:
    return False, "Invalid input"

2 个答案:

答案 0 :(得分:5)

而不是正面检查和嵌套if语句:

if a:
    if b:
        if c:
            foo()
        else:
            # error 3
     else:
         # error 2
else:
    # error 1

除非一切正常,否则你可以扭转逻辑并纾困:

if not a:
    # raise an exception

if not b:
    # raise an exception

if not c:
    # raise an exception

# If we get here, everything is OK.
foo()

这样可以更容易地查看哪个错误消息与哪个条件匹配。

答案 1 :(得分:2)

以下是Mark Byer如何针对您的案例专门实施答案的示例:

task = task.split()
if len(task) < 2:
    return False, "Invalid input"
if task[0] != 'svn':
    return False, "Not a subversion command."
if task[1] not in ALLOWED:
    return False, "Not a legal command."    
if len(task[2:]) != ALLOWED[task[1]]:
    return False, "Incorrect number of arguments."  
return True, task, None