From aecaf493b4e9a6b2da88a8dd7c14ad7c9e039ad3 Mon Sep 17 00:00:00 2001 From: "Yaakov M. Nemoy" Date: Sun, 5 Oct 2008 02:12:32 -0400 Subject: 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. --- base/base.py | 3 +++ base/module.py | 2 ++ 2 files changed, 5 insertions(+) (limited to 'base') 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 -- cgit