B ê¹`1ã@s†ddlmZddlmZddlmZddlmZddlmZddlmZddgZed ƒZed ƒZ Gd d„deeƒZ Gd d„dƒZ d S)é)ÚAny)Úcast)ÚDict)ÚGeneric)ÚTypeVar)ÚUnionÚStoreÚStoreKeyÚTÚDc@seZdZdZdZdS)r z»StoreKey is an object used as a key to a Store. A StoreKey is associated with the type T of the value of the key. A StoreKey is unique and cannot conflict with another key. ©N)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú __slots__r r r úU/home/kop/projects/devel/pgwui/test_venv/lib/python3.7/site-packages/_pytest/store.pyr sc@s¨eZdZdZdZddœdd„Zeeeddœdd „Zeeed œd d „Z eee e ee fd œdd„Z eeeed œdd„Z eedd œdd„Zeeed œdd„ZdS)raStore is a type-safe heterogenous mutable mapping that allows keys and value types to be defined separately from where it (the Store) is created. Usually you will be given an object which has a ``Store``: .. code-block:: python store: Store = some_object.store If a module wants to store data in this Store, it creates StoreKeys for its keys (at the module level): .. code-block:: python some_str_key = StoreKey[str]() some_bool_key = StoreKey[bool]() To store information: .. code-block:: python # Value type must match the key. store[some_str_key] = "value" store[some_bool_key] = True To retrieve the information: .. code-block:: python # The static type of some_str is str. some_str = store[some_str_key] # The static type of some_bool is bool. some_bool = store[some_bool_key] Why use this? ------------- Problem: module Internal defines an object. Module External, which module Internal doesn't know about, receives the object and wants to attach information to it, to be retrieved later given the object. Bad solution 1: Module External assigns private attributes directly on the object. This doesn't work well because the type checker doesn't know about these attributes and it complains about undefined attributes. Bad solution 2: module Internal adds a ``Dict[str, Any]`` attribute to the object. Module External stores its data in private keys of this dict. This doesn't work well because retrieved values are untyped. Good solution: module Internal adds a ``Store`` to the object. Module External mints StoreKeys for its own keys. Module External stores and retrieves its data using these keys. )Ú_storeN)ÚreturncCs i|_dS)N)r)Úselfr r rÚ__init__UszStore.__init__)ÚkeyÚvaluercCs||j|<dS)zSet a value for key.N)r)rrrr r rÚ __setitem__XszStore.__setitem__)rrcCstt|j|ƒS)zZGet the value for key. Raises ``KeyError`` if the key wasn't set before. )rr r)rrr r rÚ __getitem__\szStore.__getitem__)rÚdefaultrcCs"y||Stk r|SXdS)zNGet the value for key, or return default if the key wasn't set before.N)ÚKeyError)rrrr r rÚgetcsz Store.getcCs*y||Stk r$|||<|SXdS)zmReturn the value of key if already set, otherwise set the value of key to default and return default.N)r)rrrr r rÚ setdefaultks zStore.setdefaultcCs |j|=dS)z]Delete the value for key. Raises ``KeyError`` if the key wasn't set before. N)r)rrr r rÚ __delitem__tszStore.__delitem__cCs ||jkS)zReturn whether key was set.)r)rrr r rÚ __contains__{szStore.__contains__)r rrrrrr r rrr rrrrÚboolr r r r rrs6 N) ÚtypingrrrrrrÚ__all__r r r rr r r rÚs