summaryrefslogtreecommitdiff
path: root/lib/Crypto/Util/wrapper.py
blob: 1090fc7382213dd3278e48cce5234f85e9b306e1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#
#   wrapper.py: Small class to wrap an object, instantiated from a class
#       or generated by a module.
#
# ===================================================================
# The contents of this file are dedicated to the public domain.  To
# the extent that dedication to the public domain is not available,
# everyone is granted a worldwide, perpetual, royalty-free,
# non-exclusive license to exercise all rights associated with the
# contents of this file for any purpose whatsoever.
# No rights are reserved.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
# ===================================================================
#

__all__ = [ 'Wrapper' ]

class Wrapper:
    '''
    Wrapper for an object, instantiated from a class
    or from a call to a new() function in a module.
    '''
    def __init__(self, wrapped, *args):
        """
        wrapped is either a class or a module with a new() function.
        """
        if hasattr(wrapped, 'new'):
            self._wrapped = wrapped.new(*args)
        else:
            self._wrapped = wrapped(*args)
    
    def __getattr__(self, name):
        try:
            return getattr(getattr(self,'_wrapped'),name)
        except AttributeError:
            if hasattr(self, name):
                return getattr(self,name)
            raise