#602: Little cleanup

This commit is contained in:
Vladimir Iakovlev
2017-03-13 13:53:43 +01:00
parent c2b597f22b
commit 9e8b4f594d
2 changed files with 13 additions and 11 deletions
+6 -2
View File
@@ -1,11 +1,13 @@
import re
from thefuck.utils import replace_command
from thefuck.specific.sudo import sudo_support
from thefuck.utils import replace_command, for_app
no_command = "Error: No such command"
no_website = "hostscli.errors.WebsiteImportError"
@sudo_support
@for_app('hostscli')
def match(command):
errors = [no_command, no_website]
for error in errors:
@@ -14,9 +16,11 @@ def match(command):
return False
@sudo_support
def get_new_command(command):
if no_website in command.stderr:
return ['hostscli websites']
misspelled_command = re.findall(
r'Error: No such command ".*"', command.stderr)[0]
commands = ['block', 'unblock', 'websites', 'block_all', 'unblock_all']