diff options
author | 2015-08-23 15:26:27 +0300 | |
---|---|---|
committer | 2015-08-23 15:26:27 +0300 | |
commit | 449198c97d2794b44b461093265637522b7a64b4 (patch) | |
tree | 36eb68c83b907474f507f5e8cde72856e29414bb /src/console/zmq/sugar/attrsettr.py | |
parent | b156bb3ae7382481ed5f15e42d87cd3012870da5 (diff) | |
parent | 8384612b8493a4a896e91e3bb9d5d25689a87c12 (diff) |
rpc integration
Diffstat (limited to 'src/console/zmq/sugar/attrsettr.py')
-rwxr-xr-x | src/console/zmq/sugar/attrsettr.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/src/console/zmq/sugar/attrsettr.py b/src/console/zmq/sugar/attrsettr.py new file mode 100755 index 00000000..4bbd36d6 --- /dev/null +++ b/src/console/zmq/sugar/attrsettr.py @@ -0,0 +1,52 @@ +# coding: utf-8 +"""Mixin for mapping set/getattr to self.set/get""" + +# Copyright (C) PyZMQ Developers +# Distributed under the terms of the Modified BSD License. + + +from . import constants + +class AttributeSetter(object): + + def __setattr__(self, key, value): + """set zmq options by attribute""" + + # regular setattr only allowed for class-defined attributes + for obj in [self] + self.__class__.mro(): + if key in obj.__dict__: + object.__setattr__(self, key, value) + return + + upper_key = key.upper() + try: + opt = getattr(constants, upper_key) + except AttributeError: + raise AttributeError("%s has no such option: %s" % ( + self.__class__.__name__, upper_key) + ) + else: + self._set_attr_opt(upper_key, opt, value) + + def _set_attr_opt(self, name, opt, value): + """override if setattr should do something other than call self.set""" + self.set(opt, value) + + def __getattr__(self, key): + """get zmq options by attribute""" + upper_key = key.upper() + try: + opt = getattr(constants, upper_key) + except AttributeError: + raise AttributeError("%s has no such option: %s" % ( + self.__class__.__name__, upper_key) + ) + else: + return self._get_attr_opt(upper_key, opt) + + def _get_attr_opt(self, name, opt): + """override if getattr should do something other than call self.get""" + return self.get(opt) + + +__all__ = ['AttributeSetter'] |