MirroredRepos/thefuck
1
0
Fork 0
mirror of https://github.com/nvbn/thefuck.git synced 2025-10-30 22:54:14 +00:00
Code Issues Releases Wiki Activity

sudo sh execute for && in commands - preventing double sudo

Browse Source
This commit is contained in:
makalaaneesh
2015-12-23 14:35:47 +05:30
parent c3b1ba7637
commit f249098336
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/rules/test_sudo.py
View File

@@ -25,6 +25,7 @@ def test_not_match():
@pytest.mark.parametrize('before, after', [
('ls', 'sudo ls'),
('echo a > b', 'sudo sh -c "echo a > b"'),
('echo "a" >> b', 'sudo sh -c "echo \\"a\\" >> b"')])
('echo "a" >> b', 'sudo sh -c "echo \\"a\\" >> b"'),
('mkdir && touch a', 'sudo sh -c "mkdir && touch a"')])
def test_get_new_command(before, after):
assert get_new_command(Command(before)) == after

6
thefuck/rules/sudo.py
View File

@@ -21,7 +21,7 @@ patterns = ['permission denied',
def match(command):
if command.script_parts and command.script_parts[0] == 'sudo':
if command.script_parts and '&&' not in command.script_parts and command.script_parts[0] == 'sudo':
return False
for pattern in patterns:
@@ -32,7 +32,9 @@ def match(command):
def get_new_command(command):
if '>' in command.script:
if '&&' in command.script:
return u'sudo sh -c "{}"'.format(" ".join([part for part in command.script_parts if part != "sudo"]))
elif '>' in command.script:
return u'sudo sh -c "{}"'.format(command.script.replace('"', '\\"'))
else:
return u'sudo {}'.format(command.script)