[gedit-plugins] [commander] Fix rollback importer

[Jesse van den Kieboom <jessevdk src gnome org>]

commit 0d15d33f26cac2c1651ff0f3e79d7430a71c09d2
Author: Jesse van den Kieboom <jessevdk gnome org>
Date:   Sun Jul 13 13:14:23 2014 +0200

    [commander] Fix rollback importer

 plugins/commander/commander/commands/module.py     |    5 ++++-
 .../commander/commands/rollbackimporter.py         |    3 +++
 2 files changed, 7 insertions(+), 1 deletions(-)
---
diff --git a/plugins/commander/commander/commands/module.py b/plugins/commander/commander/commands/module.py
index 11be1d3..e8dccba 100644
--- a/plugins/commander/commander/commands/module.py
+++ b/plugins/commander/commander/commands/module.py
@@ -125,12 +125,15 @@ class Module(method.Method):
             sys.path.insert(0, self._dirname)
 
             self._rollback.monitor()
-            self._rollback.cancel()
+
             self.mod = importlib.import_module(self.real_name)
+            self._rollback.cancel()
 
             if not utils.is_commander_module(self.mod):
                 raise Exception('Module is not a commander module...')
 
+            self._rollback.insert_main(self.real_name)
+
             if '__default__' in self.mod.__dict__:
                 self.method = self.mod.__dict__['__default__']
             else:
diff --git a/plugins/commander/commander/commands/rollbackimporter.py 
b/plugins/commander/commander/commands/rollbackimporter.py
index 2e7698e..5fa298c 100644
--- a/plugins/commander/commander/commands/rollbackimporter.py
+++ b/plugins/commander/commander/commands/rollbackimporter.py
@@ -44,6 +44,9 @@ class RollbackImporter:
 
         return mod
 
+    def insert_main(self, mod):
+        self._new_modules.insert(0, mod)
+
     def uninstall(self):
         self.cancel()