Adds a close method to modules that needs to be implemented somehow

This lets you put code on the end of the lifecycle of a Module object.

2 files changed, 5 insertions, 0 deletions

diff --git a/base/base.py b/base/base.py
index 836779b..1010af8 100644
--- a/base/base.py
+++ b/base/base.py
@@ -84,6 +84,7 @@ def shell():
     
         # Go up a module in our stack
         elif keyword in ('up', 'cd..', 'cd ..'):
+            shell.stack[-1].close()
             shell.stack = shell.stack[:-1]
             shell.prompt = shell.prompt[:-1]
     
@@ -96,6 +97,8 @@ def shell():
     
         # Flush our module stack
         elif keyword == 'cd':
+            for obj in shell.stack:
+                obj.close()
             shell.stack = []
             shell.prompt = prompt
     
diff --git a/base/module.py b/base/module.py
index d527ea0..58a4bc3 100644
--- a/base/module.py
+++ b/base/module.py
@@ -1,4 +1,6 @@
 
 class Module(object):
     """ Our parent class for all command modules """
+    def close(self):
+        raise NotImplementedError
     pass