python - 检查 type() 并更改 str() 或 int()

Question

为了清晰起见，已编辑:

我正在尝试解决category为None的问题，该问题导致错误:TypeError:不兼容的集合类型:None不是类似列表

我试图在创建此对象 category 的实例时检查是否有字符串、int 或对象，以便将其传递到表中的 ORM。我不断遇到错误(如上所示)，因此我想创建一个检查，让对象通过(如果它们是对象)。

我一直在重新编写代码，但并没有完全达到我希望的效果。任何帮助/解释将不胜感激!

这里上课

class BaseAPI(object):
    def create_element(self, element_text, category): # OBJ
        new_element = Element(element_text, category)
        self.session.add(new_element)
        self.session.commit()
        print(element_text)


 class ConvenienceAPI(BaseAPI):
    def create_element(self, element_text, category_name):
        category = category_name
        if type(category) == str:
            category = self.retrieve_category(category_name)
        elif type(category) == int:
            print('Sorry integers not accepted')
        else:
            return super(ConvenienceAPI, self).create_element(element_text, category)

在 test.py 文件中，我创建了元素和类别的实例:

api = ConvenienceAPI()
    sa = api.create_category('Situation Awareness')
    api.create_element('gathering information', sa)

    api.create_category('Leadership')
    api.create_element('gathering information', 'Leadership')

添加运行 test.py 后出现回溯错误:

<小时/>

Traceback (most recent call last):
  File "/usr/local/lib/python2.7/site-packages/nose/case.py", line 197, in runTest
    self.test(*self.arg)
  File "/Users/ack/code/venv/NotssDB/notssdb/test/test.py", line 41, in test1
    sa3 = api.create_element('projecting and anticipating future state', sa)
  File "/Users/ack/code/venv/NotssDB/notssdb/api/convenience.py", line 22, in create_element
    return super(ConvenienceAPI, self).create_element(element_text, category)
  File "/Users/ack/code/venv/NotssDB/notssdb/api/object.py", line 192, in create_element
    new_element = Element(element_text, category)
  File "<string>", line 4, in __init__
  File "/usr/local/lib/python2.7/site-packages/sqlalchemy/orm/state.py", line 306, in _initialize_instance
    manager.dispatch.init_failure(self, args, kwargs)
  File "/usr/local/lib/python2.7/site-packages/sqlalchemy/util/langhelpers.py", line 60, in __exit__
    compat.reraise(exc_type, exc_value, exc_tb)
  File "/usr/local/lib/python2.7/site-packages/sqlalchemy/orm/state.py", line 303, in _initialize_instance
    return manager.original_init(*mixed[1:], **kwargs)
  File "/Users/ack/code/venv/NotssDB/notssdb/model/base.py", line 180, in __init__
    self.category = category
  File "/usr/local/lib/python2.7/site-packages/sqlalchemy/orm/attributes.py", line 224, in __set__
    instance_dict(instance), value, None)
  File "/usr/local/lib/python2.7/site-packages/sqlalchemy/orm/attributes.py", line 1022, in set
    lambda adapter, i: adapter.adapt_like_to_iterable(i))
  File "/usr/local/lib/python2.7/site-packages/sqlalchemy/orm/attributes.py", line 1038, in _set_iterable
    new_values = list(adapter(new_collection, iterable))
  File "/usr/local/lib/python2.7/site-packages/sqlalchemy/orm/attributes.py", line 1022, in <lambda>
    lambda adapter, i: adapter.adapt_like_to_iterable(i))
  File "/usr/local/lib/python2.7/site-packages/sqlalchemy/orm/collections.py", line 636, in adapt_like_to_iterable
    given, wanted))
TypeError: Incompatible collection type: None is not list-like

-------------------- >> begin captured stdout << ---------------------
--start db Session--
Situation Awareness
(u'Leadership', 1)

Answer 1

在test.py的第一个测试用例中:

sa = api.create_category('Situation Awareness')
api.create_element('gathering information', sa)

sa等于函数api.create_category()的返回值。我假设这个函数没有定义的返回值，这会导致 sa成为None 。您在该测试的下一行真正调用的是 api.create_element('gathering information', None) ，导致当 sqlalchemy 需要可迭代的内容时，您的意外类型“None”被传递给它。

简单的解决方案

• 不要分配 create_category() 的结果到一个变量，因为没有任何变量，而是以与领导示例相同的方式使用 API。基本上不要做类似sa = api.create_category('Situation Awareness')的事情
• 编辑 create_category()函数返回其输入类别。例如，create_category("Situation Awareness")应该返回“态势感知”，而不是 None .

问题的根源

类型安全。您对传递的变量类型做出的假设不一定正确。特别是，请看:

 class ConvenienceAPI(BaseAPI):
    def create_element(self, element_text, category_name):
        category = category_name
        if type(category) == str:
            category = self.retrieve_category(category_name)
        elif type(category) == int:
            print('Sorry integers not accepted')
        else:
            return super(ConvenienceAPI, self).create_element(element_text, category)

假设在 else:条款，category是某种类型的对象。但正如您所看到的，只要 category不是 str、int 或对象，您会立即收到错误。这仍然留下了 BaseAPI 期望什么类型的对象的问题。可能的解决方案:

class ConvenienceAPI(BaseAPI):
    def create_element(self, element_text, category_name):
        category = category_name

        # if category is None, then this will fail
        # and print out the type (since None evaluates to False).
        if category:
            if isinstance(category, basestring):
                category = self.retrieve_category(category_name)
            # Is int and float the only numeric types that can happen?
            elif isinstance(category, int) or isinstance(float) : 
                print('Sorry integers not accepted')
            else:
                # Danger Zone!!
                return super(ConvenienceAPI, self).create_element(element_text, category)
        else:
            print('Sorry, category is type %s' % str(type(category)))

请注意，我们正在检查 basestring而不仅仅是str 。这是因为category可能是 unicode，这是有效文本，但它不是 str 。两者str和unicode是 basestring 的子类。阅读 https://stackoverflow.com/a/152596/5249060了解更多信息。

即使您采取了该修复，如果类别不是str，那么在危险区域会发生什么？或int ？如果它是一个对象，但不是 BaseAPI 期望的类型，会发生什么？我将在这里停止这个答案，因为有几个选项可以解决这个问题。

为了澄清我之前的一些编辑和评论，帮助此代码的方法之一是为 category 创建一个默认对象。如果不是 str , int ，或某种对象，而不是记录错误消息。然而，这实际上取决于您正在处理的工作是否存在可接受的默认值。