x
/
stacktach-simport
Code Issues Proposed changes

properly supports importing classes now

This commit is contained in:
Sandy Walsh 2014-05-26 15:54:49 +00:00
parent 04a43c8819
commit e811203152
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
simport/__init__.py
View File

@ -72,7 +72,7 @@ def _get_module(target):
sys.path.append(path)
if not class_or_function:
raise MissingMethodOrFunction("No Method or Function specified")
raise MissingMethodOrFunction("No Method or Function specified in '%s'" % target)
__import__(module)
@ -87,4 +87,6 @@ def load(target):
return getattr(sys.modules[module], function)
class_object = getattr(sys.modules[module], klass)
return getattr(class_object, function)
if function:
return getattr(class_object, function)
return class_object

7
tests/test_simport.py
View File

@ -71,3 +71,10 @@ class TestSimport(unittest.TestCase):
self.assertEqual(external.externalmodule,
sys.modules['external.externalmodule'])
self.assertEqual(old, external.externalmodule)
self.assertEqual(method, external.externalmodule.Blah.method_b)
def test_import_class(self):
klass = simport.load("tests/external|"
"external.externalmodule:Blah")
import external.externalmodule
self.assertEqual(klass, external.externalmodule.Blah)