Len功能不像预期的那样行为

Question

我正在构建一个函数，该函数正在基于传递的标签的dist上应用AWS实例。

预期行为： 当传递多个test_customer_id时，该函数应返回标签的字典：

{'foo': 'bar', 'Account': 'shared'}

单位测试功能的当前行为仅返回：

{'foo': 'bar', 'Account': [['test_customer_id1', 'test_customer_id2']]} 

我该如何解决此问题？

def get_acct_value(tags, customer_id, ceid):
    customer_id = [customer_id]
    
    if "Account" in tags:
        if tags["Account"] == customer_id:
            pass
        else:
            if len(customer_id) > 1:
                tags["Account"] = "shared"
            elif len(customer_id) == 1:
                tags["Account"] = customer_id
            else:
                raise exceptions.CustomerNotFoundError(f"No customer(s) found on {ceid}")
    else:
        if len(customer_id) > 1:
            tags["Account"] = "shared"
        elif len(customer_id) == 1:
            tags["Account"] = customer_id
        else:
            raise exceptions.CustomerNotFoundError(f"No customer(s) found on {ceid}")
    return tags

单元测试：

TAGS_NO_VALUE = {'foo': 'bar'}
TEST_CUSTOMER_ID_LIST = ["test_customer_id1", "test_customer_id2"]
TEST_CEID = "test_ceid"


def test_get_account_value_customer_list():
    response = tagging.get_acct_value(TAGS_NO_VALUE, TEST_CUSTOMER_ID_LIST, TEST_CEID)

    print(response)

其他单元测试：

所有三个测试都应返回：{'accult'：customer_id，'foo'：'bar'}

TEST_CUSTOMER_ID = "test_customer_id"
TAGS_UNEXPECTED_VALUE = {'Account': '', 'foo': 'bar'} 
TAGS_EXPECTED_VALUE = {'Account': customer_id, 'foo': 'bar'} 

def test_get_acct_value_no_value():

    response = tagging.get_acct_value(TAGS_NO_VALUE, 
TEST_CUSTOMER_ID, TEST_CEID)

    print(response)


def test_get_acct_value_unexpected_value():

    response = tagging.get_acct_value(TAGS_UNEXPECTED_VALUE, TEST_CUSTOMER_ID, TEST_CEID)

    print(response)

def test_get_acct_value_expected_value():

    response = tagging.get_acct_value(TAGS_EXPECTED_VALUE, TEST_CUSTOMER_ID, TEST_CEID)

    print(response)

Answer 1

我认为您在这里很复杂。让我们以不同的方式打破此功能：

def get_acct_value(tags, customer_ids, ceid):
    if len(customer_ids) == 0:
        raise exceptions.CustomerNotFoundError(f"No customer(s) found on {ceid}")
    tag = "shared" if len(customer_ids) > 1 else customer_ids[0]
    tags["Account"] = tag
    return tags

首先，我们知道，如果customer_ids列表为空，我们应该提出例外。 首先执行此操作。它被标记为“界限检查”，应在尝试处理任何数据之前完成 - 这样您就不必在代码的每个分支上重做它。

其次，我们知道，如果列表大于一个，我们希望我们的标签被“共享”，这意味着我们有一个以上的客户ID。如果我们的列表大于一个，则将其设置为具有名称tag的临时变量。如果列表正是一个，我们将其设置为唯一可用的客户ID。

最后，我们进行实际工作 - 将帐户设置为我们选择的标签。如果Len（customer_ids）＆gt; 1 else customer_id [0] 。

值得注意的是，您的近端问题是customer_ids的类型必须在中传递，必须是列表。如果这是一个孤独的价值，那么您会遇到问题。您尝试通过将其投入到列表中来解决此问题，但是如果您想接受列表或单个值，最好做类似的事情：

customer_ids = customer_ids if isinstance(customer_ids, list) else [customer_ids]

这将导致这样的事情：

def get_acct_value(tags, customer_ids, ceid):
    customer_ids = list() if customer_ids is None else customer_ids
    customer_ids = customer_ids if isinstance(customer_ids, list) else [customer_ids]
    print(f"type={type(customer_ids)} {customer_ids=}")
    if len(customer_ids) == 0:
        raise exceptions.CustomerNotFoundError(f"No customer(s) found on {ceid}")
    tags["Account"] = "shared" if len(customer_ids) > 1 else customer_ids[0]
    return tags

我添加了初始检查为了customer_ids确保它不是none，它会破坏您的第二个检查（如果不是一个，则将值转换为列表），因为list（列表）（无）投掷typeerror。

请注意，我会尽快命名此函数update_account_tags（）或类似的内容，因为它返回没有值，只是标签字典，该字典的帐户具有更新的值。

一些指导：如果您发现自己进行检查，如果a在b中，则b是字典，并且您打算用a做某事，最好的事情要做的是使用字典的函数get（）。

v = b.get(a, my_default)

my_default这可以是您想要的，默认情况下是none。因此，这些都是等效的：

v = b.get(a)
v = b[a] if a in b else None

其次，如果您发现自己正在进行这样的检查的情况：

if tags["Account"] == customer_id:
    pass
else:
   tags["Account"] = customer_id

您也可以简单地做到这一点：

tags["Account"] = customer_id

结果是相同的，并且在计算上同样复杂。 （如果customer_id被替换为get_customer_id（）之类的函数，这可能不是完全正确的，但是作为第一个本能，它会做得很好。）