Преглед изворни кода

Wrapped main behavior in a main() function.

Jonathan Reem пре 12 година
родитељ
комит
77728fd5f7
1 измењених фајлова са 4 додато и 1 уклоњено
  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()