From 4bf6e87c542c908121ffb7f8ebdd2f836a988899 Mon Sep 17 00:00:00 2001
From: "W. Trevor King" <wking@tremily.us>
Date: Fri, 24 Aug 2012 09:38:13 -0400
Subject: command:base: only cleanup UseInterface.storage_callback if it's
 there.

It's hard to see why it wouldn't be, but .setup_command handles the
case where it's missing, so we should be consistent here.
---
 libbe/command/base.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'libbe/command')

diff --git a/libbe/command/base.py b/libbe/command/base.py
index 40e4bf9..a5af1f0 100644
--- a/libbe/command/base.py
+++ b/libbe/command/base.py
@@ -583,5 +583,6 @@ class UserInterface (object):
         return self._user_id
 
     def cleanup(self):
-        self.storage_callbacks.cleanup()
+        if self.storage_callbacks is not None:
+            self.storage_callbacks.cleanup()
         self.io.cleanup()
-- 
cgit