小编典典

具有一些强制键作为功能输入的字典

python

我有一个以字典为参数的函数。我将传递给各种字典的字典,这些字典比函数内部使用的字典要多。另外,我想在函数定义中查看需要哪些键。所以我写

def fun(indict=dict(apple=None, pear=None)):

但是,该函数现在接受任何输入为indict。有没有聪明的写作方法

any dictionary that has at least the keys 'apple' and 'pear' is accepted.

就像是

def fun(indict=dict(apple=NeedsToBeSpecified, pear=NeedsToBeSpecified)):

阅读 190

收藏
2021-01-20

共1个答案

小编典典

在python3.x中,您可以使用函数注释

>>> def foo(indict: dict(apple=None, pear=None)):
...     print(indict)
... 
>>> foo(dict())
{}

您甚至可以疯狂地使用现在被口译人员接受的Ellipsis文字

>>> def foo(indict: dict(apple=None, pear=None, extra_items=...)) -> int:
...     if any(x not in indict for x in ('apple', 'pear')):
...         raise ValueError('message here...')
...     print(indict)
...     return 3
... 
>>> foo({})
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 3, in foo
ValueError: message here...
>>> foo({'apple':6, 'pear':4})
{'pear': 4, 'apple': 6}
3
>>> foo({'apple':6, 'pear':4, 'carrot':30000})
{'carrot': 30000, 'pear': 4, 'apple': 6}
3

你可以从我的第一个例子中看到,注释不 执行
任何东西。您必须在函数本身中执行验证,尽管我想您可以从注解1中检查所需的键,如果您想使其保持DRY,但是仅使用2个键可能就不值得了……

在python2.x中(以及更传统的情况下),也许您只想将信息放入文档字符串中;-)-我也建议您对python3.x进行此操作,因为这是查找的传统位置用于文档…

1个keys = foo.__annotations__['indict'].keys() - {'extra_items'}

更新
请注意,现在像mypy这样的奇特事物闲坐着,这个答案可能有些过时了。您可以考虑使用TypedDictfrom进行注释mypy_extensions。如果您使用类似类型的检查器,那应该为您的用户设定期望,甚至可能帮助捕获一些错误mypy

from mypy_extensions import TypedDict

class Apple:
    """Represent an Apple."""

class Pear:
    """Represent a Pear."""

# "annotation-type" for a dictionary that has an apple and pear key whose values are Apple and Pear instances.
FruitBowl = TypedDict("FruitBowl": {"apple": Apple, "Pear": Pear})

def foo(indict: FruitBowl) -> int:
    ...
2021-01-20