mirror of
https://github.com/nvbn/thefuck.git
synced 2025-03-29 05:58:51 +00:00
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
import re
|
|
patterns = [
|
|
r'WARNING: REMOTE HOST IDENTIFICATION HAS CHANGED!',
|
|
r'WARNING: POSSIBLE DNS SPOOFING DETECTED!',
|
|
r"Warning: the \S+ host key for '([^']+)' differs from the key for the IP address '([^']+)'",
|
|
]
|
|
offending_pattern = re.compile(
|
|
r'(?:Offending (?:key for IP|\S+ key)|Matching host key) in ([^:]+):(\d+)',
|
|
re.MULTILINE)
|
|
|
|
commands = ['ssh', 'scp']
|
|
|
|
|
|
def match(command, settings):
|
|
if not command.script:
|
|
return False
|
|
if not command.script.split()[0] in commands:
|
|
return False
|
|
if not any([re.findall(pattern, command.stderr) for pattern in patterns]):
|
|
return False
|
|
return True
|
|
|
|
|
|
def remove_offending_keys(command, settings):
|
|
offending = offending_pattern.findall(command.stderr)
|
|
print offending
|
|
for filepath, lineno in offending:
|
|
with open(filepath, 'r') as fh:
|
|
lines = fh.readlines()
|
|
del lines[int(lineno) - 1]
|
|
print lines
|
|
with open(filepath, 'w') as fh:
|
|
fh.writelines(lines)
|
|
|
|
|
|
def get_new_command(command, settings):
|
|
remove_offending_keys(command, settings)
|
|
return command.script
|