mirror of
https://github.com/nvbn/thefuck.git
synced 2025-03-23 02:58:31 +00:00
27 lines
916 B
Python
27 lines
916 B
Python
from itertools import dropwhile, takewhile, islice
|
|
import re
|
|
import subprocess
|
|
from thefuck.utils import sudo_support, replace_command
|
|
|
|
|
|
@sudo_support
|
|
def match(command, settings):
|
|
return command.script.startswith('docker') \
|
|
and 'is not a docker command' in command.stderr
|
|
|
|
|
|
def get_docker_commands():
|
|
proc = subprocess.Popen('docker', stdout=subprocess.PIPE)
|
|
lines = [line.decode('utf-8') for line in proc.stdout.readlines()]
|
|
lines = dropwhile(lambda line: not line.startswith('Commands:'), lines)
|
|
lines = islice(lines, 1, None)
|
|
lines = list(takewhile(lambda line: line != '\n', lines))
|
|
return [line.strip().split(' ')[0] for line in lines]
|
|
|
|
|
|
@sudo_support
|
|
def get_new_command(command, settings):
|
|
wrong_command = re.findall(
|
|
r"docker: '(\w+)' is not a docker command.", command.stderr)[0]
|
|
return replace_command(command, wrong_command, get_docker_commands())
|