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
|
# Copyright 2016 IBM Corp.
#
# All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
"""General utilities around wrappers."""
import importlib
import os
import sys
import pypowervm.wrappers.entry_wrapper as ewrap
_wrappers_pkg = ('pypowervm', 'wrappers')
_imports = None
_this_module = sys.modules[__name__]
def _get_imports():
"""Imports all modules in pypowervm.wrappers and returns them as a dict.
The returned dict is of the form:
{ module_name: <module object>, ... }
"""
global _imports
if _imports is None:
_modnames = [fname.rsplit('.', 1)[0]
for fname in os.listdir(os.path.join(*_wrappers_pkg))
if not fname.startswith('_')
and fname.endswith('.py')]
_imports = {
modname: importlib.import_module(
'.'.join(_wrappers_pkg) + '.' + modname)
for modname in _modnames}
return _imports
def wrapper_class_iter():
"""Iterator over all Wrapper subclasses defined in pypowervm.wrappers.
Each yield is the Wrapper subclass itself.
"""
for klass in (cls for imp in _get_imports().values()
for cls in vars(imp).values()):
try:
if issubclass(klass, ewrap.Wrapper):
yield klass
except TypeError:
# issubclass can't handle the things that aren't classes
pass
|
