add extras module with extras.attrs.AttribDictable

Author: Someguy123

privex/helpers/__init__.py

@@ -102,6 +102,11 @@
 except ImportError:
     log.debug('privex.helpers __init__ failed to import "setuppy.bump", not loading packaging helpers')
 
+try:
+    from privex.helpers.extras import *
+except ImportError:
+    log.debug('privex.helpers __init__ failed to import "extras", cannot load any extras.* modules...')
+
 
 def _setup_logging(level=logging.WARNING):
     """

privex/helpers/extras/__init__.py

@@ -0,0 +1,24 @@
+"""
+Various helper functions/classes which depend on a certain package being installed.
+
+This constructor file attempts to load each extras module individually, each wrapped with a try/catch
+for :class:`ImportError` so that one unavailable package doesn't cause problems.
+
+"""
+import logging
+
+log = logging.getLogger(__name__)
+
+__all__ = ['HAS_ATTRS']
+HAS_ATTRS = False
+
+try:
+    from privex.helpers.extras.attrs import *
+    HAS_ATTRS = True
+    __all__ += ['AttribDictable']
+except ImportError:
+    log.debug('privex.helpers.extras __init__ failed to import "attrs", not loading attrs library helpers')
+    
+
+
+

privex/helpers/extras/attrs.py

@@ -0,0 +1,54 @@
+import attr
+
+__all__ = [
+    'AttribDictable'
+]
+
+
+@attr.s
+class AttribDictable:
+    """
+    A small mixin class for :py:func:`attr.s` classes, allowing them to behave like dictionaries, and
+    support casting into dictionaries using ``dict(x)``.
+    
+    Usage::
+        
+        >>> import attr
+        >>> from privex.helpers import AttribDictable
+        >>>
+        >>> @attr.s
+        >>> class Example(AttribDictable):
+        ...     hello = attr.ib(type=str)
+        ...     testing = attr.ib(type=bool, default=True)
+        ...
+        >>> x = Example(hello='world')
+        >>> x['hello']
+        'world'
+        >>> x['hello'] = 'lorem ipsum'
+        >>> dict(x)
+        {'hello': 'lorem ipsum', 'testing': True}
+        
+    
+    """
+    def get(self, key, default=None):
+        try:
+            return self[key]
+        except KeyError:
+            return default
+    
+    def __iter__(self):
+        """Handle casting via ``dict(myclass)``"""
+        for k, v in attr.asdict(self).items():
+            yield k, v
+    
+    def __getitem__(self, key):
+        """
+        When the instance is accessed like a dict, try returning the matching attribute.
+        If the attribute doesn't exist, raise :class:`KeyError`
+        """
+        if hasattr(self, key):
+            return getattr(self, key)
+        raise KeyError(key)
+    
+    def __setitem__(self, key, value):
+        return setattr(self, key, value)