DhJdZddlZddlZddlZddlmZej rddlZejdZ ejdZ eje e fZ ejejde fZejejde fZejdej" Zejd Zej(ej*e e fej,ej.e e fej,ej*e e fd fZej2d ZGd deje e fej6ZGddee e fZGddee e fZGddej>eZ GddejBeejDeZ#e$dk(rddl%Z%e%jLyy)aB This module provides custom container classes with enhanced functionality. Classes: CastedDictBase: Abstract base class for dictionaries that cast keys and values. CastedDict: Dictionary that casts keys and values to specified types. LazyCastedDict: Dictionary that lazily casts values to specified types upon access. UniqueList: List that only allows unique values, with configurable behavior on duplicates. SliceableDeque: Deque that supports slicing and enhanced equality checks. Type Aliases: KT: Type variable for dictionary keys. VT: Type variable for dictionary values. DT: Type alias for a dictionary with keys of type KT and values of type VT. KT_cast: Type alias for a callable that casts dictionary keys. VT_cast: Type alias for a callable that casts dictionary values. HT: Type variable for hashable values in UniqueList. T: Type variable for generic types. DictUpdateArgs: Union type for arguments that can be used to update a dictionary. OnDuplicate: Literal type for handling duplicate values in UniqueList. Usage: - CastedDict and LazyCastedDict can be used to create dictionaries with automatic type casting. - UniqueList ensures all elements are unique and can raise an error on duplicates. - SliceableDeque extends deque with slicing support and enhanced equality checks. Examples: >>> d = CastedDict(int, int) >>> d[1] = 2 >>> d['3'] = '4' >>> d.update({'5': '6'}) >>> d.update([('7', '8')]) >>> d {1: 2, 3: 4, 5: 6, 7: 8} >>> l = UniqueList(1, 2, 3) >>> l.append(4) >>> l.append(4) >>> l.insert(0, 4) >>> l.insert(0, 5) >>> l[1] = 10 >>> l [5, 10, 2, 3, 4] >>> d = SliceableDeque([1, 2, 3, 4, 5]) >>> d[1:4] SliceableDeque([2, 3, 4]) N)typesKTVT.HT)boundTz(_typeshed.SupportsKeysAndGetItem[KT, VT])ignoreraisec eZdZUdZeeed<eeed< ddeedeede eefded df d Z de e je jfde jd dfd Z d e jd e jd dffd ZxZS)CastedDictBasea4 Abstract base class for dictionaries that cast keys and values. Attributes: _key_cast (KT_cast[KT]): Callable to cast dictionary keys. _value_cast (VT_cast[VT]): Callable to cast dictionary values. Methods: __init__(key_cast: KT_cast[KT] = None, value_cast: VT_cast[VT] = None, *args: DictUpdateArgs[KT, VT], **kwargs: VT) -> None: Initializes the dictionary with optional key and value casting callables. update(*args: DictUpdateArgs[types.Any, types.Any], **kwargs: types.Any) -> None: Updates the dictionary with the given arguments. __setitem__(key: types.Any, value: types.Any) -> None: Sets the item in the dictionary, casting the key if a key cast callable is provided. _key_cast _value_castNkey_cast value_castargskwargsreturncD||_||_|j|i|y)a Initializes the CastedDictBase with optional key and value casting callables. Args: key_cast (KT_cast[KT], optional): Callable to cast dictionary keys. Defaults to None. value_cast (VT_cast[VT], optional): Callable to cast dictionary values. Defaults to None. *args (DictUpdateArgs[KT, VT]): Arguments to initialize the dictionary. **kwargs (VT): Keyword arguments to initialize the dictionary. N)rrupdate)selfrrrrs y/var/www/fastuser/data/www/generator.snapmosaic.io/flask_app/venv/lib/python3.12/site-packages/python_utils/containers.py__init__zCastedDictBase.__init__us&*&! T$V$cf|r|j||r|jD] \}}|||< yy)a  Updates the dictionary with the given arguments. Args: *args (DictUpdateArgs[types.Any, types.Any]): Arguments to update the dictionary. **kwargs (types.Any): Keyword arguments to update the dictionary. N)ritems)rrrkeyvalues rrzCastedDictBase.updates?  FMM4 $lln " U!S  " rrrc^|j|j|}t| ||S) Sets the item in the dictionary, casting the key if a key cast callable is provided. Args: key (types.Any): The key to set in the dictionary. value (types.Any): The value to set in the dictionary. rsuper __setitem__rrr __class__s rr#zCastedDictBase.__setitem__s/ >> %..%Cw"3..r)NN)__name__ __module__ __qualname____doc__KT_castr__annotations__VT_castrDictUpdateArgsrrAnyrr# __classcell__r%s@rr r ]s(r{!%"&%"+%BK%b"f% %  %  %2"#EIIuyy$89"EJYY" "$ /uyy / /t / /rr cXeZdZdZdej dej ddffd ZxZS) CastedDicta7 Custom dictionary that casts keys and values to the specified typing. Note that you can specify the types for mypy and type hinting with: CastedDict[int, int](int, int) >>> d: CastedDict[int, int] = CastedDict(int, int) >>> d[1] = 2 >>> d['3'] = '4' >>> d.update({'5': '6'}) >>> d.update([('7', '8')]) >>> d {1: 2, 3: 4, 5: 6, 7: 8} >>> list(d.keys()) [1, 3, 5, 7] >>> list(d) [1, 3, 5, 7] >>> list(d.values()) [2, 4, 6, 8] >>> list(d.items()) [(1, 2), (3, 4), (5, 6), (7, 8)] >>> d[3] 4 # Casts are optional and can be disabled by passing None as the cast >>> d = CastedDict() >>> d[1] = 2 >>> d['3'] = '4' >>> d.update({'5': '6'}) >>> d.update([('7', '8')]) >>> d {1: 2, '3': '4', '5': '6', '7': '8'} rrrNc`|j|j|}t| ||y)z.Sets `key` to `cast(value)` in the dictionary.N)rr"r#r$s rr#zCastedDict.__setitem__s.    '$$U+E C'r)r&r'r(r)typingr.r#r/r0s@rr2r2s0 D(vzz(&**(((rr2ceZdZdZdej dej ddffd Zdej deffd Zdejeje efddfffd Z dejeddfffd Z xZS) LazyCastedDicta Custom dictionary that casts keys and lazily casts values to the specified typing. Note that the values are cast only when they are accessed and are not cached between executions. Note that you can specify the types for mypy and type hinting with: LazyCastedDict[int, int](int, int) >>> d: LazyCastedDict[int, int] = LazyCastedDict(int, int) >>> d[1] = 2 >>> d['3'] = '4' >>> d.update({'5': '6'}) >>> d.update([('7', '8')]) >>> d {1: 2, 3: '4', 5: '6', 7: '8'} >>> list(d.keys()) [1, 3, 5, 7] >>> list(d) [1, 3, 5, 7] >>> list(d.values()) [2, 4, 6, 8] >>> list(d.items()) [(1, 2), (3, 4), (5, 6), (7, 8)] >>> d[3] 4 # Casts are optional and can be disabled by passing None as the cast >>> d = LazyCastedDict() >>> d[1] = 2 >>> d['3'] = '4' >>> d.update({'5': '6'}) >>> d.update([('7', '8')]) >>> d {1: 2, '3': '4', '5': '6', '7': '8'} >>> list(d.keys()) [1, '3', '5', '7'] >>> list(d.values()) [2, '4', '6', '8'] >>> list(d.items()) [(1, 2), ('3', '4'), ('5', '6'), ('7', '8')] >>> d['3'] '4' rrrNc`|j|j|}t| ||y)r Nr!r$s rr#zLazyCastedDict.__setitem__s, >> %..%C C'rc|j|j|}t| |}|j|j|}|S)a Gets the item from the dictionary, casting the value if a value cast callable is provided. Args: key (types.Any): The key to get from the dictionary. Returns: VT: The value from the dictionary. )rr" __getitem__rr$s rr9zLazyCastedDict.__getitem__sM >> %..%C#C(    '$$U+E rc#K|jt| Ed{yt| D]\}}||j|fy71w)a Returns a generator of the dictionary's items, casting the values if a value cast callable is provided. Yields: types.Generator[types.Tuple[KT, VT], None, None]: A generator of the dictionary's items. N)rr"rr$s rrzLazyCastedDict.items+sX    #w} & &#gmo 3 U4++E222 3 'sAA2Ac#K|jt| Ed{yt| D]}|j|y7,w)z Returns a generator of the dictionary's values, casting the values if a value cast callable is provided. Yields: types.Generator[VT, None, None]: A generator of the dictionary's values. N)rr"valuesrrr%s rr<zLazyCastedDict.values<sQ    #w~' ' ') .&&u-- . (sAA-A)r&r'r(r)rr.r#rr9 GeneratorTuplerrr<r/r0s@rr6r6s+Z (uyy ( (t (uyyR*3 RV,dD8 93" .D$7 . .rr6ceZdZUdZej eed<dddedeffdZ dejd ed d ffd Z d ed d ffd Z ded e fdZej dejded d fdZej dedej&ed d fdZdej(eejfdej(ej&eefd d ffd Zdej(ejefd d ffd ZxZS) UniqueLista A list that only allows unique values. Duplicate values are ignored by default, but can be configured to raise an exception instead. >>> l = UniqueList(1, 2, 3) >>> l.append(4) >>> l.append(4) >>> l.insert(0, 4) >>> l.insert(0, 5) >>> l[1] = 10 >>> l [5, 10, 2, 3, 4] >>> l = UniqueList(1, 2, 3, on_duplicate='raise') >>> l.append(4) >>> l.append(4) Traceback (most recent call last): ... ValueError: Duplicate value: 4 >>> l.insert(0, 4) Traceback (most recent call last): ... ValueError: Duplicate value: 4 >>> 4 in l True >>> l[0] 1 >>> l[1] = 4 Traceback (most recent call last): ... ValueError: Duplicate value: 4 _setr ) on_duplicaterrCc~||_t|_t||D]}|j |y)a  Initializes the UniqueList with optional duplicate handling behavior. Args: *args (HT): Initial values for the list. on_duplicate (OnDuplicate, optional): Behavior on duplicates. Defaults to 'ignore'. N)rCsetrBr"rappend)rrCrargr%s rrzUniqueList.__init__ps<)E   C KK  rindexrrNc||jvr|jdk(rtd|y|jj|t|||y)aP Inserts a value at the specified index, ensuring uniqueness. Args: index (types.SupportsIndex): The index to insert the value at. value (HT): The value to insert. Raises: ValueError: If the value is a duplicate and `on_duplicate` is set to 'raise'. r Duplicate value: N)rBrC ValueErroraddr"insertrrHrr%s rrMzUniqueList.insertsS DII   G+ #4UG!<== e ue$rc||jvr|jdk(rtd|y|jj|t||y)z Appends a value to the list, ensuring uniqueness. Args: value (HT): The value to append. Raises: ValueError: If the value is a duplicate and `on_duplicate` is set to 'raise'. r rJN)rBrCrKrLr"rFr=s rrFzUniqueList.appendsQ DII   G+ #4UG!<== e uritemc||jvS)z Checks if the list contains the specified item. Args: item (HT): The item to check for. Returns: bool: True if the item is in the list, False otherwise. )rB)rrPs r __contains__zUniqueList.__contains__styy  rindicesr<cyNrrSr<s rr#zUniqueList.__setitem__rcyrUrVrWs rr#zUniqueList.__setitem__rXrct|trtjtjt |}|j dk(r tdt||jz}|r|t||k7rtd||jj|nitjt |}||jvr&|||k7r|j dk(rtd|y|jj|t|=tjt|tjtj t |y)a Sets the item(s) at the specified index/indices, ensuring uniqueness. Args: indices (types.Union[slice, types.SupportsIndex]): The index or slice to set the value(s) at. values (types.Union[types.Iterable[HT], HT]): The value(s) to set. Raises: RuntimeError: If `on_duplicate` is 'ignore' and setting slices. ValueError: If the value(s) are duplicates and `on_duplicate` is set to 'raise'. r z^ignore mode while setting slices introduces ambiguous behaviour and is therefore not supportedzDuplicate values: r rJN) isinstanceslicercastIterablerrC RuntimeErrorrErBlistrKrrLr"r#List)rrSr< duplicatesr%s rr#zUniqueList.__setitem__s$ gu %ZZr 2F;F  H,"? ),F dii(?JfT'](;; #5j\!BCC II  V $ZZF+F"vg'>$$/$'8%ABB IIMM& !  JJug & 5::b>6(J rct|tr&||D]}|jj|n|jj||t||y)z Deletes the item(s) at the specified index/indices. Args: index (types.Union[types.SupportsIndex, slice]): The index or slice to delete the item(s) at. N)r[r\rBremover" __delitem__rNs rrezUniqueList.__delitem__sY eU #e (   ' ( II  T%[ ) E"r)r&r'r(r)rSetrr+ OnDuplicater SupportsIndexrMrFboolrRr4overloadr#r\r^Unionrer/r0s@rrArALseB ))B- %-"&%E//%%t%*B4( ! ! ! __**46  __&+nnR&8 + UE$7$778+  ENN2.23+   + Z#[[!4!4e!;<# ##rrAc*eZdZdZej dejdefdZ ej de ddfdZ dejeje fdejedfffd Z dejde ffd Zd dedeffd ZxZS) SliceableDequea9 A deque that supports slicing and enhanced equality checks. Methods: __getitem__(index: types.Union[types.SupportsIndex, slice]) -> types.Union[T, 'SliceableDeque[T]']: Returns the item or slice at the given index. __eq__(other: types.Any) -> bool: Checks equality with another object, allowing for comparison with lists, tuples, and sets. pop(index: int = -1) -> T: Removes and returns the item at the given index. Only supports index 0 and the last index. rHrcyrUrVrrHs rr9zSliceableDeque.__getitem__s>> d = SliceableDeque[int]([1, 2, 3, 4, 5]) >>> d[1:4] SliceableDeque([2, 3, 4]) >>> d = SliceableDeque[str](['a', 'b', 'c']) >>> d[-2:] SliceableDeque(['b', 'c']) c3(K|] }| ywrUrV).0irs r z-SliceableDeque.__getitem__..3s!La$q'!Ls)r[r\rSlenr%ranger"r9)rrHstartstopstepr%s` rr9zSliceableDeque.__getitem__sV. eU # % c$i 8 E4>>!L5d3K!LL L7&u- -rotherct|trt||k(St|trt||k(St|trt||k(St||S)a Checks equality with another object, allowing for comparison with lists, tuples, and sets. Args: other (types.Any): The object to compare with. Returns: bool: True if the objects are equal, False otherwise. )r[r`tuplerEr"__eq__)rr{r%s rr~zSliceableDeque.__eq__7s_ eT ":& & u %;%' ' s #t9% %7>%( (rc|dk(rt|S|dt|dz hvrt| St d)a Removes and returns the item at the given index. Only supports index 0 and the last index. Args: index (int, optional): The index of the item to remove. Defaults to -1. Returns: T: The removed item. Raises: IndexError: If the index is not 0 or the last index. Examples: >>> d = SliceableDeque([1, 2, 3]) >>> d.pop(0) 1 >>> d.pop() 3 rrz=Only index 0 and the last index (`N-1` or `-1`) are supported)r"popleftrvpop IndexError)rrHr%s rrzSliceableDeque.popKsL, A:7?$ $ r3t9q=) )7;=   r)r)r&r'r(r)r4rjrrhr r9r\rkr.rir~intrr/r0s@rrmrms  __?!4!4??? __CC+>CC.[[!4!4e!;<. Q++ ,.:)EII)$)(arrm__main__)'r)abc collectionsr4r TYPE_CHECKING _typeshedTypeVarrrDictDTOptionalCallabler*r,Hashablerr rkMappingr^r?r-LiteralrgABCr r2r6rarAGenericdequermr&doctesttestmodrVrrrs6r   U]]4U]]4 ZZB ..R0 1 ..R0 1U]]4u~~.EMM# MM"b& NN5;;r2v&' NN5==R().0mm-. O/UZZB'O/d((B'((Vo.^BF+o.ds#Bs#peU]]1%{'8'8';eP zGOOr