Merge branch 'master' of github.com:nvbn/thefuck into slow

thefuck/specific/git.py

@@ -1,37 +1,32 @@
-from functools import wraps
 import re
 from shlex import split
+from decorator import decorator
 from ..types import Command
-from ..utils import quote
+from ..utils import quote, is_app
 
 
-def git_support(fn):
+@decorator
+def git_support(fn, command, settings):
     """Resolves git aliases and supports testing for both git and hub."""
-    @wraps(fn)
-    def wrapper(command, settings):
-        # supports GitHub's `hub` command
-        # which is recommended to be used with `alias git=hub`
-        # but at this point, shell aliases have already been resolved
-        is_git_cmd = command.script.startswith(('git', 'hub'))
+    # supports GitHub's `hub` command
+    # which is recommended to be used with `alias git=hub`
+    # but at this point, shell aliases have already been resolved
+    if not is_app(command, 'git', 'hub'):
+        return False
 
-        if not is_git_cmd:
-            return False
+    # perform git aliases expansion
+    if 'trace: alias expansion:' in command.stderr:
+        search = re.search("trace: alias expansion: ([^ ]*) => ([^\n]*)",
+                           command.stderr)
+        alias = search.group(1)
 
-        # perform git aliases expansion
-        if 'trace: alias expansion:' in command.stderr:
-            search = re.search("trace: alias expansion: ([^ ]*) => ([^\n]*)",
-                               command.stderr)
-            alias = search.group(1)
+        # by default git quotes everything, for example:
+        #     'commit' '--amend'
+        # which is surprising and does not allow to easily test for
+        # eg. 'git commit'
+        expansion = ' '.join(map(quote, split(search.group(2))))
+        new_script = command.script.replace(alias, expansion)
 
-            # by default git quotes everything, for example:
-            #     'commit' '--amend'
-            # which is surprising and does not allow to easily test for
-            # eg. 'git commit'
-            expansion = ' '.join(map(quote, split(search.group(2))))
-            new_script = command.script.replace(alias, expansion)
+        command = Command._replace(command, script=new_script)
 
-            command = Command._replace(command, script=new_script)
-
-        return fn(command, settings)
-
-    return wrapper
+    return fn(command, settings)

thefuck/specific/sudo.py

@@ -1,24 +1,22 @@
-from functools import wraps
 import six
+from decorator import decorator
 from ..types import Command
 
 
-def sudo_support(fn):
+@decorator
+def sudo_support(fn, command, settings):
     """Removes sudo before calling fn and adds it after."""
-    @wraps(fn)
-    def wrapper(command, settings):
-        if not command.script.startswith('sudo '):
-            return fn(command, settings)
+    if not command.script.startswith('sudo '):
+        return fn(command, settings)
 
-        result = fn(Command(command.script[5:],
-                            command.stdout,
-                            command.stderr),
-                    settings)
+    result = fn(Command(command.script[5:],
+                        command.stdout,
+                        command.stderr),
+                settings)
 
-        if result and isinstance(result, six.string_types):
-            return u'sudo {}'.format(result)
-        elif isinstance(result, list):
-            return [u'sudo {}'.format(x) for x in result]
-        else:
-            return result
-    return wrapper
+    if result and isinstance(result, six.string_types):
+        return u'sudo {}'.format(result)
+    elif isinstance(result, list):
+        return [u'sudo {}'.format(x) for x in result]
+    else:
+        return result