aboutsummaryrefslogtreecommitdiff
blob: 3a73d5b430a7ce16bdbb74f089e83ca7626e0750 (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
73
74
# Copyright (C) 2005, 2006 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""A version of inspect that includes what 'copy' needs.

Importing the python standard module 'copy' is far more expensive than it
needs to be, because copy imports 'inspect' which imports 'tokenize'.
And 'copy' only needs 2 small functions out of 'inspect', but has to
load all of 'tokenize', which makes it horribly slow.

This module is designed to use tricky hacks in import rules, to avoid this
overhead.
"""


####
# These are the only 2 functions that 'copy' needs from 'inspect'
# As you can see, they are quite trivial, and don't justify the
# 40ms spent to import 'inspect' because it is importing 'tokenize'
# These are copied verbatim from the python standard library.

# ----------------------------------------------------------- class helpers
def _searchbases(cls, accum):
    # Simulate the "classic class" search order.
    if cls in accum:
        return
    accum.append(cls)
    for base in cls.__bases__:
        _searchbases(base, accum)


def getmro(cls):
    "Return tuple of base classes (including cls) in method resolution order."
    if hasattr(cls, "__mro__"):
        return cls.__mro__
    else:
        result = []
        _searchbases(cls, result)
        return tuple(result)


def inject_copy():
    """Import the 'copy' module with a hacked 'inspect' module"""
    # We don't actually care about 'getmro' but we need to pass
    # something in the list so that we get the direct module,
    # rather than getting the base module
    import sys

    # Don't hack around if 'inspect' already exists
    if 'inspect' in sys.modules:
        import copy
        return

    mod = __import__('snakeoil.fix_copy',
                     globals(), locals(), ['getmro'])

    sys.modules['inspect'] = mod
    try:
        import copy
    finally:
        del sys.modules['inspect']