Wrapped main behavior in a main() function.

reem · reem · commit 77728fd5f7a8 · 2013-10-01T20:41:08.000-04:00
diff --git a/command.py b/command.py
@@ -18,7 +18,7 @@ def undo(self):
         os.rename(self.dest, self.src)
 
 
-if __name__ == "__main__":
+def main():
     command_stack = []
 
     # commands are just pushed into the command stack
@@ -32,3 +32,6 @@ def undo(self):
     # and can also be undone at will
     for cmd in reversed(command_stack):
         cmd.undo()
+
+if __name__ == "__main__":
+    main()
