Forráskód Böngészése

Wrapped main behavior in a main() function.

Jonathan Reem 12 éve
szülő
commit
77728fd5f7
1 módosított fájl, 4 hozzáadás és 1 törlés
  1. 4 1
      command.py

+ 4 - 1
command.py

@@ -18,7 +18,7 @@ class MoveFileCommand(object):
         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 @@ if __name__ == "__main__":
     # and can also be undone at will
     for cmd in reversed(command_stack):
         cmd.undo()
+
+if __name__ == "__main__":
+    main()