MirroredRepos/thefuck
1
0
Fork 0
mirror of https://github.com/nvbn/thefuck.git synced 2025-02-21 12:28:41 +00:00
Code Issues Releases Wiki Activity

#205 Fix import in cd_correction

Browse Source
This commit is contained in:
nvbn 2015-05-19 15:46:23 +03:00
parent dc53f58b2a
commit 6590da623f
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
thefuck/rules/cd_correction.py
View File

@ -4,22 +4,25 @@ __author__ = "mmussomele"
"""Attempts to spellcheck and correct failed cd commands""" """Attempts to spellcheck and correct failed cd commands"""
import os import os
import cd_mkdir
from difflib import get_close_matches from difflib import get_close_matches
from thefuck.utils import sudo_support from thefuck.utils import sudo_support
from thefuck.rules import cd_mkdir
MAX_ALLOWED_DIFF = 0.6 MAX_ALLOWED_DIFF = 0.6
def _get_sub_dirs(parent): def _get_sub_dirs(parent):
"""Returns a list of the child directories of the given parent directory""" """Returns a list of the child directories of the given parent directory"""
return [child for child in os.listdir(parent) if os.path.isdir(os.path.join(parent, child))] return [child for child in os.listdir(parent) if os.path.isdir(os.path.join(parent, child))]
@sudo_support @sudo_support
def match(command, settings): def match(command, settings):
"""Match function copied from cd_mkdir.py""" """Match function copied from cd_mkdir.py"""
return (command.script.startswith('cd ') return (command.script.startswith('cd ')
and ('no such file or directory' in command.stderr.lower() and ('no such file or directory' in command.stderr.lower()
or 'cd: can\'t cd to' in command.stderr.lower())) or 'cd: can\'t cd to' in command.stderr.lower()))
@sudo_support @sudo_support
def get_new_command(command, settings): def get_new_command(command, settings):
@ -44,6 +47,7 @@ def get_new_command(command, settings):
cwd = os.path.join(cwd, best_matches[0]) cwd = os.path.join(cwd, best_matches[0])
else: else:
return cd_mkdir.get_new_command(command, settings) return cd_mkdir.get_new_command(command, settings)
return "cd {0}".format(cwd) return "cd {0}".format(cwd)
enabled_by_default = True enabled_by_default = True