1
0
mirror of https://github.com/nvbn/thefuck.git synced 2025-02-23 21:39:32 +00:00
thefuck/thefuck/rules/hostscli.py
2017-03-13 13:53:43 +01:00

28 lines
753 B
Python

import re
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:
if error in command.stderr:
return True
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']
return replace_command(command, misspelled_command, commands)