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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
# Copyright 2009 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
# $Id$
__all__ = ['lazy_import']
import sys
import types
from portage.proxy.objectproxy import ObjectProxy
class _LazyImport(ObjectProxy):
__slots__ = ('_scope', '_alias', '_name', '_target')
def __init__(self, scope, alias, name):
ObjectProxy.__init__(self)
object.__setattr__(self, '_scope', scope)
object.__setattr__(self, '_alias', alias)
object.__setattr__(self, '_name', name)
def _get_target(self):
try:
return object.__getattribute__(self, '_target')
except AttributeError:
pass
name = object.__getattribute__(self, '_name')
__import__(name)
target = sys.modules[name]
object.__setattr__(self, '_target', target)
object.__getattribute__(self, '_scope')[
object.__getattribute__(self, '_alias')] = target
return target
class _LazyImportFrom(_LazyImport):
__slots__ = ()
def _get_target(self):
try:
return object.__getattribute__(self, '_target')
except AttributeError:
pass
name = object.__getattribute__(self, '_name')
components = name.split('.')
parent_name = '.'.join(components[:-1])
__import__(parent_name)
target = getattr(sys.modules[parent_name], components[-1])
object.__setattr__(self, '_target', target)
object.__getattribute__(self, '_scope')[
object.__getattribute__(self, '_alias')] = target
return target
def lazyimport(scope, *args):
"""
Create a proxy in the given scope in order to performa a lazy import.
Syntax Result
foo import foo
foo:bar,baz from foo import bar, baz
foo:bar@baz from foo import bar as baz
@param scope: the scope in which to place the import, typically globals()
@type myfilename: dict
@param args: module names to import
@type args: strings
"""
modules = sys.modules
for s in args:
parts = s.split(':', 1)
if len(parts) == 1:
name = s
if not name or not isinstance(name, basestring):
raise ValueError(name)
components = name.split('.')
parent_scope = scope
for i in xrange(len(components)):
alias = components[i]
if i < len(components) - 1:
parent_name = ".".join(components[:i+1])
__import__(parent_name)
mod = modules.get(parent_name)
if not isinstance(mod, types.ModuleType):
# raise an exception
__import__(name)
parent_scope[alias] = mod
parent_scope = mod.__dict__
continue
already_imported = modules.get(name)
if already_imported is not None:
parent_scope[alias] = already_imported
else:
parent_scope[alias] = \
_LazyImport(parent_scope, alias, name)
else:
name, fromlist = parts
already_imported = modules.get(name)
fromlist = fromlist.split(',')
for s in fromlist:
alias = s.split('@', 1)
if len(alias) == 1:
alias = alias[0]
orig = alias
else:
orig, alias = alias
if already_imported is not None:
try:
scope[alias] = getattr(already_imported, orig)
except AttributeError:
raise ImportError('cannot import name %s' % orig)
else:
scope[alias] = _LazyImportFrom(scope, alias,
name + '.' + orig)
|