U 42‹i ã@szddlmZddlmZddlmZddlmZddlmZddgZedƒZed ƒZ Gd d„deeƒZ Gd d„dƒZ d S) é)Ú annotations)ÚAny)Úcast)ÚGeneric)ÚTypeVarÚStashÚStashKeyÚTÚDc@seZdZdZdZdS)rzï``StashKey`` is an object used as a key to a :class:`Stash`. A ``StashKey`` is associated with the type ``T`` of the value of the key. A ``StashKey`` is unique and cannot conflict with another key. .. versionadded:: 7.0 ©N)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú __slots__r r r ú1/tmp/pip-unpacked-wheel-7n2p7kht/_pytest/stash.pyrs c@s–eZdZdZdZddœdd„Zdddd œd d „Zddd œd d„Zddddœdd„Zddddœdd„Z ddd œdd„Z ddd œdd„Z ddœdd„Z dS)raJ``Stash`` is a type-safe heterogeneous mutable mapping that allows keys and value types to be defined separately from where it (the ``Stash``) is created. Usually you will be given an object which has a ``Stash``, for example :class:`~pytest.Config` or a :class:`~_pytest.nodes.Node`: .. code-block:: python stash: Stash = some_object.stash If a module or plugin wants to store data in this ``Stash``, it creates :class:`StashKey`\s for its keys (at the module level): .. code-block:: python # At the top-level of the module some_str_key = StashKey[str]() some_bool_key = StashKey[bool]() To store information: .. code-block:: python # Value type must match the key. stash[some_str_key] = "value" stash[some_bool_key] = True To retrieve the information: .. code-block:: python # The static type of some_str is str. some_str = stash[some_str_key] # The static type of some_bool is bool. some_bool = stash[some_bool_key] .. versionadded:: 7.0 ©Ú_storageÚNone)ÚreturncCs i|_dS)Nr©Úselfr r rÚ__init__HszStash.__init__z StashKey[T]r )ÚkeyÚvaluercCs||j|<dS)zSet a value for key.Nr)rrrr r rÚ __setitem__KszStash.__setitem__)rrcCstt|j|ƒS)zZGet the value for key. Raises ``KeyError`` if the key wasn't set before. )rr r©rrr r rÚ __getitem__OszStash.__getitem__r zT | D)rÚdefaultrcCs(z ||WStk r"|YSXdS)zNGet the value for key, or return default if the key wasn't set before.N©ÚKeyError©rrrr r rÚgetVs z Stash.getcCs0z ||WStk r*|||<|YSXdS)zmReturn the value of key if already set, otherwise set the value of key to default and return default.Nrr!r r rÚ setdefault^s  zStash.setdefaultcCs |j|=dS)z]Delete the value for key. Raises ``KeyError`` if the key wasn't set before. Nrrr r rÚ __delitem__gszStash.__delitem__ÚboolcCs ||jkS)zReturn whether key was set.rrr r rÚ __contains__nszStash.__contains__ÚintcCs t|jƒS)z)Return how many items exist in the stash.)Úlenrrr r rÚ__len__rsz Stash.__len__N) r r rrrrrrr"r#r$r&r)r r r rrs( N) Ú __future__rÚtypingrrrrÚ__all__r r rrr r r rÚs