Merge pull request #339 from mcarton/fix-338

Fix #338

thefuck/main.py

@@ -74,10 +74,10 @@ def get_command(settings, args):
             return types.Command(script, None, None)
 
 
-def run_command(command, settings):
+def run_command(old_cmd, command, settings):
     """Runs command from rule for passed command."""
     if command.side_effect:
-        command.side_effect(command, settings)
+        command.side_effect(old_cmd, command.script, settings)
     shells.put_to_history(command.script)
     print(command.script)
 
@@ -100,7 +100,7 @@ def fix_command():
         corrected_commands = get_corrected_commands(command, user_dir, settings)
         selected_command = select_command(corrected_commands, settings)
         if selected_command:
-            run_command(selected_command, settings)
+            run_command(command, selected_command, settings)
 
 
 def print_alias(entry_point=True):

thefuck/rules/dirty_untar.py

@@ -35,7 +35,7 @@ def get_new_command(command, settings):
                  .format(dir=_tar_file(command.script)[1], cmd=command.script)
 
 
-def side_effect(command, settings):
-    with tarfile.TarFile(_tar_file(command.script)[0]) as archive:
+def side_effect(old_cmd, command, settings):
+    with tarfile.TarFile(_tar_file(old_cmd.script)[0]) as archive:
         for file in archive.getnames():
             os.remove(file)

thefuck/rules/dirty_unzip.py

@@ -30,8 +30,8 @@ def get_new_command(command, settings):
     return '{} -d {}'.format(command.script, _zip_file(command)[:-4])
 
 
-def side_effect(command, settings):
-    with zipfile.ZipFile(_zip_file(command), 'r') as archive:
+def side_effect(old_cmd, command, settings):
+    with zipfile.ZipFile(_zip_file(old_cmd), 'r') as archive:
         for file in archive.namelist():
             os.remove(file)
 

thefuck/rules/ssh_known_hosts.py

@@ -26,8 +26,8 @@ def get_new_command(command, settings):
     return command.script
 
 
-def side_effect(command, settings):
-    offending = offending_pattern.findall(command.stderr)
+def side_effect(old_cmd, command, settings):
+    offending = offending_pattern.findall(old_cmd.stderr)
     for filepath, lineno in offending:
         with open(filepath, 'r') as fh:
             lines = fh.readlines()