summaryrefslogtreecommitdiff
path: root/repoze/bfg/mapply.py
blob: a27faf3eefed5691a9150205bc20a85674dbe30d (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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
##############################################################################
#
# Copyright (c) 2002 Zope Corporation and Contributors. All Rights Reserved.
#
# This software is subject to the provisions of the Zope Public License,
# Version 2.1 (ZPL).  A copy of the ZPL should accompany this distribution.
# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
# FOR A PARTICULAR PURPOSE
#
##############################################################################
"""Provide an apply-like facility that works with any mapping object
"""

def mapply(object,
           positional=(),
           keyword={},
           maybe=True,
           ):

    if hasattr(object,'__bases__'): # the object is a class
        raise TypeError('Cannot publish class %s' % object)

    f = object
    im = False

    if hasattr(f, 'im_func'):
        im = True
    elif not hasattr(f, 'func_defaults'):
        if hasattr(f, '__call__'):
            f = f.__call__
            if hasattr(f, 'im_func'):
                im = True
            elif not hasattr(f, 'func_defaults') and maybe:
                return object
        elif maybe:
            return object

    if im:
        f = f.im_func
        c = f.func_code
        defaults = f.func_defaults
        names = c.co_varnames[1:c.co_argcount]
    else:
        defaults = f.func_defaults
        c = f.func_code
        names = c.co_varnames[:c.co_argcount]

    nargs = len(names)
    if positional:
        positional = list(positional)
        if len(positional) > nargs:
            raise TypeError('too many arguments')
        args = positional
    else:
        args = []

    get = keyword.get
    nrequired = len(names) - (len(defaults or ()))
    for index in range(len(args), len(names)):
        name = names[index]
        v = get(name, args)
        if v is args:
            if index < nrequired:
                raise TypeError('Argument %s was omitted' % name)
            else:
                v = defaults[index-nrequired]
        args.append(v)

    args = tuple(args)
    return object(*args)