FilesExpand file tree

 master

/

command.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

36 lines (27 loc) · 983 Bytes

 master

/

command.py

Copy path

Top

File metadata and controls

• Code
• Blame

36 lines (27 loc) · 983 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

import os

class MoveFileCommand(object):

def __init__(self, src, dest):

self.src = src

self.dest = dest

def execute(self):

self()

def __call__(self):

print('renaming {} to {}'.format(self.src, self.dest))

os.rename(self.src, self.dest)

def undo(self):

print('renaming {} to {}'.format(self.dest, self.src))

os.rename(self.dest, self.src)

if __name__ == "__main__":

undo_stack = []

ren1 = MoveFileCommand('foo.txt', 'bar.txt')

ren2 = MoveFileCommand('bar.txt', 'baz.txt')

# commands are just pushed into the command stack

for cmd in ren1, ren2:

undo_stack.append(cmd)

# they can be executed later on will

for cmd in undo_stack:

cmd.execute() # foo.txt is now renamed to baz.txt

# and can also be undone on will

for cmd in undo_stack:

undo_stack.pop().undo() # Now it's bar.txt

undo_stack.pop().undo() # and back to foo.txt