From 03bd5cb77c54033857810bb17562859eefa41221 Mon Sep 17 00:00:00 2001 From: Simon Busch Date: Wed, 9 Apr 2014 13:20:33 +0200 Subject: [PATCH] Add possibility to import templates from packages Does not have any unit tests yet. Signed-off-by: Simon Busch --- dbusmock/mockobject.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/dbusmock/mockobject.py b/dbusmock/mockobject.py index e4f130f..389df70 100644 --- a/dbusmock/mockobject.py +++ b/dbusmock/mockobject.py @@ -46,6 +46,9 @@ def load_module(name): exec(f.read(), mod.__dict__, mod.__dict__) return mod + if '.' in name: + return importlib.import_module(name) + return importlib.import_module('dbusmock.templates.' + name)