← Back
Editing: ipstruct.cpython-311.pyc
� ��X�+ � �, � d Z dgZ G d� de� � ZdS )zyA dict subclass that supports attribute style access. Can probably be replaced by types.SimpleNamespace from Python 3.3 �Structc �n � e Zd ZdZdZd� Zd� Zd� Zd� Zd� Z d� Z d � Z d � Z d � Z d � Zd � Zd� Zdd�Zdd�ZdS )r aY A dict subclass with attribute style access. This dict subclass has a a few extra features: * Attribute style access. * Protection of class members (like keys, items) when using attribute style access. * The ability to restrict assignment to only existing keys. * Intelligent merging. * Overloaded operators. Tc �h � t � | dd� � t j | g|�R i |�� dS )a� Initialize with a dictionary, another Struct, or data. Parameters ---------- args : dict, Struct Initialize with one dict or Struct kw : dict Initialize with key, value pairs. Examples -------- >>> s = Struct(a=10,b=30) >>> s.a 10 >>> s.b 30 >>> s2 = Struct(s,c=30) >>> sorted(s2.keys()) ['a', 'b', 'c'] � _allownewTN)�object� __setattr__�dict�__init__)�self�args�kws �;/usr/lib/python3/dist-packages/ipython_genutils/ipstruct.pyr zStruct.__init__ sB � �, ���4��d�3�3�3� � �d�(�T�(�(�(�R�(�(�(�(�(� c �x � | j s|| vrt d|z � � �t � | ||� � dS )a{ Set an item with check for allownew. Examples -------- >>> s = Struct() >>> s['a'] = 10 >>> s.allow_new_attr(False) >>> s['a'] = 10 >>> s['a'] 10 >>> try: ... s['b'] = 20 ... except KeyError: ... print('this is not allowed') ... this is not allowed z8can't create new attribute %s when allow_new_attr(False)N)r �KeyErrorr � __setitem__)r �key�values r r zStruct.__setitem__3 sR � �& �~� R�#�T�/�/��J�S�P�R� R� R� ����s�E�*�*�*�*�*r c � � t |t � � r0|| j v st t |� � rt d|z � � � | � ||� � dS # t $ r}t |� � �d}~ww xY w)a� Set an attr with protection of class members. This calls :meth:`self.__setitem__` but convert :exc:`KeyError` to :exc:`AttributeError`. Examples -------- >>> s = Struct() >>> s.a = 10 >>> s.a 10 >>> try: ... s.get = 10 ... except AttributeError: ... print("you can't set a class member") ... you can't set a class member z.attr %s is a protected member of class Struct.N)� isinstance�str�__dict__�hasattrr �AttributeErrorr r )r r r �es r r zStruct.__setattr__K s� � �* �c�3� � � � �d�m�#�#�w�v�s�';�';�#�$�D�s�J�� � � $� � � �S�%� (� (� (� (� (��� $� $� $� ��#�#� #����� $���s �A � A=�)A8�8A=c �R � | | }|S # t $ r t |� � �w xY w)a� Get an attr by calling :meth:`dict.__getitem__`. Like :meth:`__setattr__`, this method converts :exc:`KeyError` to :exc:`AttributeError`. Examples -------- >>> s = Struct(a=10) >>> s.a 10 >>> type(s.get) <... 'builtin_function_or_method'> >>> try: ... s.b ... except AttributeError: ... print("I don't have that key") ... I don't have that key )r r )r r �results r � __getattr__zStruct.__getattr__n sB � �* ��#�Y�F� �M�� � &� &� &� ��%�%� %� &���s � �&c �0 � | � |� � | S )z�s += s2 is a shorthand for s.merge(s2). Examples -------- >>> s = Struct(a=10,b=30) >>> s2 = Struct(a=20,c=40) >>> s += s2 >>> sorted(s.keys()) ['a', 'b', 'c'] )�merge)r �others r �__iadd__zStruct.__iadd__� s � � � � �5����� r c �X � | � � � }|� |� � |S )z�s + s2 -> New Struct made from s.merge(s2). Examples -------- >>> s1 = Struct(a=10,b=30) >>> s2 = Struct(a=20,c=40) >>> s = s1 + s2 >>> sorted(s.keys()) ['a', 'b', 'c'] )�copyr �r r �souts r �__add__zStruct.__add__� s( � � �y�y�{�{�� � � �5����� r c �8 � | � � � }||z }|S )z�s1 - s2 -> remove keys in s2 from s1. Examples -------- >>> s1 = Struct(a=10,b=30) >>> s2 = Struct(a=40) >>> s = s1 - s2 >>> s {'b': 30} )r# r$ s r �__sub__zStruct.__sub__� s � � �y�y�{�{�� �� ��� r c �B � |� � � D ] }|| v r| |= � | S )z�Inplace remove keys from self that are in other. Examples -------- >>> s1 = Struct(a=10,b=30) >>> s2 = Struct(a=40) >>> s1 -= s2 >>> s1 {'b': 30} )�keys)r r �ks r �__isub__zStruct.__isub__� s1 � � ��� � � � �A��D�y�y���G��� r c � � i }|� � � D ]8\ }}t |t � � r|� � � }|D ]}|||< ��9|S )z�Helper function for merge. Takes a dictionary whose values are lists and returns a dict with the elements of each list as keys and the original keys as values. )�itemsr r �split)r �data�outdictr+ �lst�entrys r � __dict_invertzStruct.__dict_invert� sg � � ���Z�Z�\�\� #� #�E�A�c��#�s�#�#� "��i�i�k�k�� � #� #��!"����� #��r c � � | S �N� �r s r r z Struct.dict� s � �� r c �P � t t � | � � � � S )z�Return a copy as a Struct. Examples -------- >>> s = Struct(a=10,b=30) >>> s2 = s.copy() >>> type(s2) is Struct True )r r r# r8 s r r# z Struct.copy� s � � �d�i�i��o�o�&�&�&r c � � || v S )a hasattr function available as a method. Implemented like has_key. Examples -------- >>> s = Struct(a=10) >>> s.hasattr('a') True >>> s.hasattr('b') False >>> s.hasattr('get') False r7 )r r s r r zStruct.hasattr� s � � �d�{�r c �> � t � | d|� � dS )z�Set whether new attributes can be created in this Struct. This can be used to catch typos by verifying that the attribute user tries to change already exists in this Struct. r N)r r )r �allows r �allow_new_attrzStruct.allow_new_attr� s" � � ���4��e�4�4�4�4�4r Nc �� � t |fi |��}d� }d� }d� }d� }d� } t � | |� � } |rw|� � � } d|fd|fd|fd |fd | ffD ])\ } } | | � � � v r| | | | < | | = �*| � | � | � � � � |D ]3}|| vr || | |< � | | | | || � � | |< �4d S ) a Merge two Structs with customizable conflict resolution. This is similar to :meth:`update`, but much more flexible. First, a dict is made from data+key=value pairs. When merging this dict with the Struct S, the optional dictionary 'conflict' is used to decide what to do. If conflict is not given, the default behavior is to preserve any keys with their current value (the opposite of the :meth:`update` method's behavior). Parameters ---------- __loc_data : dict, Struct The data to merge into self __conflict_solve : dict The conflict policy dict. The keys are binary functions used to resolve the conflict and the values are lists of strings naming the keys the conflict resolution function applies to. Instead of a list of strings a space separated string can be used, like 'a b c'. kw : dict Additional key, value pairs to merge in Notes ----- The `__conflict_solve` dict is a dictionary of binary functions which will be used to solve key conflicts. Here is an example:: __conflict_solve = dict( func1=['a','b','c'], func2=['d','e'] ) In this case, the function :func:`func1` will be used to resolve keys 'a', 'b' and 'c' and the function :func:`func2` will be used for keys 'd' and 'e'. This could also be written as:: __conflict_solve = dict(func1='a b c',func2='d e') These functions will be called for each key they apply to with the form:: func1(self['a'], other['a']) The return value is used as the final merged value. As a convenience, merge() provides five (the most commonly needed) pre-defined policies: preserve, update, add, add_flip and add_s. The easiest explanation is their implementation:: preserve = lambda old,new: old update = lambda old,new: new add = lambda old,new: old + new add_flip = lambda old,new: new + old # note change of order! add_s = lambda old,new: old + ' ' + new # only for str! You can use those four words (as strings) as keys instead of defining them as functions, and the merge method will substitute the appropriate functions for you. For more complicated conflict resolution policies, you still need to construct your own functions. Examples -------- This show the default policy: >>> s = Struct(a=10,b=30) >>> s2 = Struct(a=20,c=40) >>> s.merge(s2) >>> sorted(s.items()) [('a', 10), ('b', 30), ('c', 40)] Now, show how to specify a conflict dict: >>> s = Struct(a=10,b=30) >>> s2 = Struct(a=20,b=40) >>> conflict = {'update':'a','add':'b'} >>> s.merge(s2,conflict) >>> sorted(s.items()) [('a', 20), ('b', 70)] c � � | S r6 r7 ��old�news r �<lambda>z Struct.merge.<locals>.<lambda>] � � �3� r c � � |S r6 r7 r@ s r rC z Struct.merge.<locals>.<lambda>^ rD r c � � | |z S r6 r7 r@ s r rC z Struct.merge.<locals>.<lambda>_ � � �3��9� r c � � || z S r6 r7 r@ s r rC z Struct.merge.<locals>.<lambda>` rG r c � � | dz |z S )N� r7 r@ s r rC z Struct.merge.<locals>.<lambda>a s � �3��9�s�?� r �preserve�update�add�add_flip�add_sN)r �fromkeysr# r* rL �_Struct__dict_invert)r � __loc_data__�_Struct__conflict_solver � data_dictrK rL rM rN rO �conflict_solve�inv_conflict_solve_user�name�funcr s r r z Struct.merge s^ � �n � �+�+��+�+� � '�&��&�&��,�,��,�,��2�2�� � � �t�X�6�6�� � O�&6�&;�&;�&=�&=� #� *�8� 4�x��6G� %�c�{�Z��,A� '��� 0� 6� 6� ��d� �2�7�7�9�9�9�9�4K�D�4Q�+�D�1�/��5�� � !� !�$�"4�"4�5L�"M�"M� N� N� N� � J� J�C��$��� %�c�N��S� � � /�N�3� /��S� �)�C�.� I� I��S� � � J� Jr )T)NN)�__name__� __module__� __qualname__�__doc__r r r r r r! r& r( r, rQ r r# r r= r r7 r r r r s � � � � � � � � �I�)� )� )�2+� +� +�0!$� !$� !$�F� � �8 � � � � � � � � � � � �" � � � � � � '� '� '�� � �$5� 5� 5� 5�uJ� uJ� uJ� uJ� uJ� uJr N)r\ �__all__r r r7 r r �<module>r^ s` ��� � �*��jJ� jJ� jJ� jJ� jJ�T� jJ� jJ� jJ� jJ� jJr
Save File
Cancel