mirror of
				https://github.com/nvbn/thefuck.git
				synced 2025-11-04 09:02:08 +00:00 
			
		
		
		
	Compare commits
	
		
			71 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						 | 
					b80f3ea6e4 | ||
| 
						 | 
					99d9c9aff7 | ||
| 
						 | 
					4be60c78fa | ||
| 
						 | 
					847b9e2cec | ||
| 
						 | 
					c463fea8a0 | ||
| 
						 | 
					7652884df6 | ||
| 
						 | 
					e1ca120eb8 | ||
| 
						 | 
					d3295e6a4e | ||
| 
						 | 
					bc2c87e8fe | ||
| 
						 | 
					a2ac15da56 | ||
| 
						 | 
					4cf631fa47 | ||
| 
						 | 
					cbf6507e1e | ||
| 
						 | 
					49ed98c8a4 | ||
| 
						 | 
					52bf3907a1 | ||
| 
						 | 
					077de17d6c | ||
| 
						 | 
					419878f526 | ||
| 
						 | 
					f610cf2256 | ||
| 
						 | 
					2f04a953ba | ||
| 
						 | 
					698451f65d | ||
| 
						 | 
					b7cb407637 | ||
| 
						 | 
					bb6b9a638c | ||
| 
						 | 
					0009fb0588 | ||
| 
						 | 
					a9d3456e29 | ||
| 
						 | 
					1e28671934 | ||
| 
						 | 
					3134a60e27 | ||
| 
						 | 
					03dd7eda04 | ||
| 
						 | 
					d12a8bcdd8 | ||
| 
						 | 
					58069f0a3e | ||
| 
						 | 
					d0e02bc20c | ||
| 
						 | 
					e554238996 | ||
| 
						 | 
					fa465620ba | ||
| 
						 | 
					294ba07ce1 | ||
| 
						 | 
					e21befbcc4 | ||
| 
						 | 
					a454438939 | ||
| 
						 | 
					1fa7827f1a | ||
| 
						 | 
					48ec853436 | ||
| 
						 | 
					5a8d889dc0 | ||
| 
						 | 
					1f96faef2c | ||
| 
						 | 
					0235c0654d | ||
| 
						 | 
					f286033f82 | ||
| 
						 | 
					473f5e6a33 | ||
| 
						 | 
					f1cce413b3 | ||
| 
						 | 
					ee2b208adf | ||
| 
						 | 
					a20bf6fa23 | ||
| 
						 | 
					da050f0db3 | ||
| 
						 | 
					f5e9124327 | ||
| 
						 | 
					1f38e0a932 | ||
| 
						 | 
					380827d1d9 | ||
| 
						 | 
					54b5cd6122 | ||
| 
						 | 
					9611264210 | ||
| 
						 | 
					24ce459f2c | ||
| 
						 | 
					07b9aba0d0 | ||
| 
						 | 
					bb42780ca5 | ||
| 
						 | 
					af2bfe7c58 | ||
| 
						 | 
					157e3e95fc | ||
| 
						 | 
					776ff4e3db | ||
| 
						 | 
					5de020bccd | ||
| 
						 | 
					0272e8a801 | ||
| 
						 | 
					2e652112ff | ||
| 
						 | 
					12eab10028 | ||
| 
						 | 
					61eab83789 | ||
| 
						 | 
					d3d1f99232 | ||
| 
						 | 
					ca67080bd9 | ||
| 
						 | 
					54c408a6b5 | ||
| 
						 | 
					20b6c4c160 | ||
| 
						 | 
					0553d57ec1 | ||
| 
						 | 
					e046d55de8 | ||
| 
						 | 
					69a9516477 | ||
| 
						 | 
					c788dfbc14 | ||
| 
						 | 
					b4b599df80 | ||
| 
						 | 
					93b6a623e1 | 
							
								
								
									
										68
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										68
									
								
								README.md
									
									
									
									
									
								
							@@ -88,8 +88,8 @@ Reading package lists... Done
 | 
			
		||||
 | 
			
		||||
## Requirements
 | 
			
		||||
 | 
			
		||||
- python (2.7+ or 3.3+)
 | 
			
		||||
- pip
 | 
			
		||||
- python
 | 
			
		||||
- python-dev
 | 
			
		||||
 | 
			
		||||
## Installation
 | 
			
		||||
@@ -100,39 +100,17 @@ Install `The Fuck` with `pip`:
 | 
			
		||||
sudo pip install thefuck
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
If it fails try to use `easy_install`:
 | 
			
		||||
 | 
			
		||||
```bash
 | 
			
		||||
sudo easy_install thefuck
 | 
			
		||||
```
 | 
			
		||||
[Or using an OS package manager (OS X, Ubuntu, Arch).](https://github.com/nvbn/thefuck/wiki/Installation)
 | 
			
		||||
 | 
			
		||||
And add to `.bashrc` or `.zshrc` or `.bash_profile`(for OSX):
 | 
			
		||||
 | 
			
		||||
```bash
 | 
			
		||||
alias fuck='$(thefuck $(fc -ln -1))'
 | 
			
		||||
# You can use whatever you want as an alias, like for mondays:
 | 
			
		||||
alias fuck='eval $(thefuck $(fc -ln -1))'
 | 
			
		||||
# You can use whatever you want as an alias, like for Mondays:
 | 
			
		||||
alias FUCK='fuck'
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
Or in `config.fish`:
 | 
			
		||||
 | 
			
		||||
```fish
 | 
			
		||||
function fuck
 | 
			
		||||
    eval (thefuck $history[1])
 | 
			
		||||
end
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
Or in your Powershell `$PROFILE` on Windows:
 | 
			
		||||
 | 
			
		||||
```powershell
 | 
			
		||||
function fuck { 
 | 
			
		||||
    $fuck = $(thefuck (get-history -count 1).commandline)
 | 
			
		||||
    if($fuck.startswith("echo")) { 
 | 
			
		||||
        $fuck.substring(5) 
 | 
			
		||||
    } 
 | 
			
		||||
    else { iex "$fuck" } 
 | 
			
		||||
}
 | 
			
		||||
```
 | 
			
		||||
[On in your shell config (Bash, Zsh, Fish, Powershell).](https://github.com/nvbn/thefuck/wiki/Shell-aliases)
 | 
			
		||||
 | 
			
		||||
Changes will be available only in a new shell session.
 | 
			
		||||
 | 
			
		||||
@@ -148,21 +126,28 @@ sudo pip install thefuck --upgrade
 | 
			
		||||
The Fuck tries to match rule for the previous command, create new command
 | 
			
		||||
using matched rule and run it. Rules enabled by default:
 | 
			
		||||
 | 
			
		||||
* `brew_unknown_command` – fixes wrong brew commands, for example `brew docto/brew doctor`;
 | 
			
		||||
* `cd_parent` – changes `cd..` to `cd ..`;
 | 
			
		||||
* `cd_mkdir` – creates directories before cd'ing into them;
 | 
			
		||||
* `cp_omitting_directory` – adds `-a` when you `cp` directory;
 | 
			
		||||
* `fix_alt_space` – replaces Alt+Space with Space character;
 | 
			
		||||
* `git_no_command` – fixes wrong git commands like `git brnch`;
 | 
			
		||||
* `git_push` – adds `--set-upstream origin $branch` to previous failed `git push`;
 | 
			
		||||
* `has_exists_script` – prepends `./` when script/binary exists;
 | 
			
		||||
* `lein_not_task` – fixes wrong `lein` tasks like `lein rpl`;
 | 
			
		||||
* `mkdir_p` – adds `-p` when you trying to create directory without parent;
 | 
			
		||||
* `no_command` – fixes wrong console commands, for example `vom/vim`;
 | 
			
		||||
* `pip_unknown_command` – fixes wrong pip commands, for example `pip instatl/pip install`;
 | 
			
		||||
* `python_command` – prepends `python` when you trying to run not executable/without `./` python script;
 | 
			
		||||
* `sl_ls` – changes `sl` to `ls`;
 | 
			
		||||
* `rm_dir` – adds `-rf` when you trying to remove directory;
 | 
			
		||||
* `ssh_known_hosts` – removes host from `known_hosts` on warning;
 | 
			
		||||
* `sudo` – prepends `sudo` to previous command if it failed because of permissions;
 | 
			
		||||
* `switch_layout` – switches command from your local layout to en.
 | 
			
		||||
 | 
			
		||||
Bundled, but not enabled by default:
 | 
			
		||||
 | 
			
		||||
* `ls_lah` – adds -lah to ls;
 | 
			
		||||
* `rm_root` – adds `--no-preserve-root` to `rm -rf /` command.
 | 
			
		||||
 | 
			
		||||
## Creating your own rules
 | 
			
		||||
@@ -174,7 +159,7 @@ and `get_new_command(command: Command, settings: Settings) -> str`.
 | 
			
		||||
 | 
			
		||||
`Command` has three attributes: `script`, `stdout` and `stderr`.
 | 
			
		||||
 | 
			
		||||
`Settings` is `~/.thefuck/settings.py`.
 | 
			
		||||
`Settings` is a special object filled with `~/.thefuck/settings.py` and values from env, [more](#settings).
 | 
			
		||||
 | 
			
		||||
Simple example of the rule for running script with `sudo`:
 | 
			
		||||
 | 
			
		||||
@@ -195,11 +180,36 @@ def get_new_command(command, settings):
 | 
			
		||||
 | 
			
		||||
The Fuck has a few settings parameters, they can be changed in `~/.thefuck/settings.py`:
 | 
			
		||||
 | 
			
		||||
* `rules` – list of enabled rules, by default all;
 | 
			
		||||
* `rules` – list of enabled rules, by default `thefuck.conf.DEFAULT_RULES`;
 | 
			
		||||
* `require_confirmation` – require confirmation before running new command, by default `False`;
 | 
			
		||||
* `wait_command` – max amount of time in seconds for getting previous command output;
 | 
			
		||||
* `no_colors` – disable colored output.
 | 
			
		||||
 | 
			
		||||
Example of `settings.py`:
 | 
			
		||||
 | 
			
		||||
```python
 | 
			
		||||
rules = ['sudo', 'no_command']
 | 
			
		||||
require_confirmation = True
 | 
			
		||||
wait_command = 10
 | 
			
		||||
no_colors = False
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
Or via environment variables:
 | 
			
		||||
 | 
			
		||||
* `THEFUCK_RULES` – list of enabled rules, like `DEFAULT_RULES:rm_root` or `sudo:no_command`;
 | 
			
		||||
* `THEFUCK_REQUIRE_CONFIRMATION` – require confirmation before running new command, `true/false`;
 | 
			
		||||
* `THEFUCK_WAIT_COMMAND` – max amount of time in seconds for getting previous command output;
 | 
			
		||||
* `THEFUCK_NO_COLORS` – disable colored output, `true/false`.
 | 
			
		||||
 
 | 
			
		||||
For example:
 | 
			
		||||
 | 
			
		||||
```bash
 | 
			
		||||
export THEFUCK_RULES='sudo:no_command'
 | 
			
		||||
export THEFUCK_REQUIRE_CONFIRMATION='true'
 | 
			
		||||
export THEFUCK_WAIT_COMMAND=10
 | 
			
		||||
export THEFUCK_NO_COLORS='false'
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## Developing
 | 
			
		||||
 | 
			
		||||
Install `The Fuck` for development:
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										2
									
								
								setup.py
									
									
									
									
									
								
							
							
						
						
									
										2
									
								
								setup.py
									
									
									
									
									
								
							@@ -1,7 +1,7 @@
 | 
			
		||||
from setuptools import setup, find_packages
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
VERSION = '1.27'
 | 
			
		||||
VERSION = '1.31'
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
setup(name='thefuck',
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										0
									
								
								tests/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										0
									
								
								tests/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
								
								
									
										49
									
								
								tests/rules/test_brew_install.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										49
									
								
								tests/rules/test_brew_install.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,49 @@
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules.brew_install import match, get_new_command
 | 
			
		||||
from thefuck.rules.brew_install import brew_formulas
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
def brew_no_available_formula():
 | 
			
		||||
    return '''Error: No available formula for elsticsearch '''
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
def brew_install_no_argument():
 | 
			
		||||
    return '''This command requires a formula argument'''
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
def brew_already_installed():
 | 
			
		||||
    return '''Warning: git-2.3.5 already installed'''
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _is_not_okay_to_test():
 | 
			
		||||
    if 'elasticsearch' not in brew_formulas:
 | 
			
		||||
        return True
 | 
			
		||||
    return False
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.mark.skipif(_is_not_okay_to_test(),
 | 
			
		||||
                    reason='No need to run if there\'s no formula')
 | 
			
		||||
def test_match(brew_no_available_formula, brew_already_installed,
 | 
			
		||||
               brew_install_no_argument):
 | 
			
		||||
    assert match(Command('brew install elsticsearch',
 | 
			
		||||
                         stderr=brew_no_available_formula), None)
 | 
			
		||||
    assert not match(Command('brew install git',
 | 
			
		||||
                             stderr=brew_already_installed), None)
 | 
			
		||||
    assert not match(Command('brew install', stderr=brew_install_no_argument),
 | 
			
		||||
                     None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.mark.skipif(_is_not_okay_to_test(),
 | 
			
		||||
                    reason='No need to run if there\'s no formula')
 | 
			
		||||
def test_get_new_command(brew_no_available_formula):
 | 
			
		||||
    assert get_new_command(Command('brew install elsticsearch',
 | 
			
		||||
                                   stderr=brew_no_available_formula), None)\
 | 
			
		||||
        == 'brew install elasticsearch'
 | 
			
		||||
 | 
			
		||||
    assert get_new_command(Command('brew install aa',
 | 
			
		||||
                                   stderr=brew_no_available_formula),
 | 
			
		||||
                           None) != 'brew install aha'
 | 
			
		||||
							
								
								
									
										28
									
								
								tests/rules/test_brew_unknown_command.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										28
									
								
								tests/rules/test_brew_unknown_command.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,28 @@
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules.brew_unknown_command import match, get_new_command
 | 
			
		||||
from thefuck.rules.brew_unknown_command import brew_commands
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
def brew_unknown_cmd():
 | 
			
		||||
    return '''Error: Unknown command: inst'''
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
def brew_unknown_cmd_instaa():
 | 
			
		||||
    return '''Error: Unknown command: instaa'''
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match(brew_unknown_cmd):
 | 
			
		||||
    assert match(Command('brew inst', stderr=brew_unknown_cmd), None)
 | 
			
		||||
    for command in brew_commands:
 | 
			
		||||
        assert not match(Command('brew ' + command), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command(brew_unknown_cmd, brew_unknown_cmd_instaa):
 | 
			
		||||
    assert get_new_command(Command('brew inst', stderr=brew_unknown_cmd), None)\
 | 
			
		||||
        == 'brew list'
 | 
			
		||||
 | 
			
		||||
    assert get_new_command(Command('brew instaa', stderr=brew_unknown_cmd_instaa),
 | 
			
		||||
                           None) == 'brew install'
 | 
			
		||||
							
								
								
									
										25
									
								
								tests/rules/test_cd_mkdir.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								tests/rules/test_cd_mkdir.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,25 @@
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules.cd_mkdir import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.mark.parametrize('command', [
 | 
			
		||||
    Command(script='cd foo', stderr='cd: foo: No such file or directory'),
 | 
			
		||||
    Command(script='cd foo/bar/baz',
 | 
			
		||||
            stderr='cd: foo: No such file or directory'),
 | 
			
		||||
    Command(script='cd foo/bar/baz', stderr='cd: can\'t cd to foo/bar/baz')])
 | 
			
		||||
def test_match(command):
 | 
			
		||||
    assert match(command, None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.mark.parametrize('command', [
 | 
			
		||||
    Command(script='cd foo', stderr=''), Command()])
 | 
			
		||||
def test_not_match(command):
 | 
			
		||||
    assert not match(command, None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.mark.parametrize('command, new_command', [
 | 
			
		||||
    (Command('cd foo'), 'mkdir -p foo && cd foo'),
 | 
			
		||||
    (Command('cd foo/bar/baz'), 'mkdir -p foo/bar/baz && cd foo/bar/baz')])
 | 
			
		||||
def test_get_new_command(command, new_command):
 | 
			
		||||
    assert get_new_command(command, None) == new_command
 | 
			
		||||
@@ -1,12 +1,12 @@
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
from thefuck.rules.cd_parent import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    assert match(Command('cd..', '', 'cd..: command not found'), None)
 | 
			
		||||
    assert not match(Command('', '', ''), None)
 | 
			
		||||
    assert match(Command('cd..', stderr='cd..: command not found'), None)
 | 
			
		||||
    assert not match(Command(), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
    assert get_new_command(
 | 
			
		||||
        Command('cd..', '', ''), None) == 'cd ..'
 | 
			
		||||
        Command('cd..'), None) == 'cd ..'
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										53
									
								
								tests/rules/test_composer_not_command.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								tests/rules/test_composer_not_command.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,53 @@
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules.composer_not_command import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
def composer_not_command():
 | 
			
		||||
    return """
 | 
			
		||||
 | 
			
		||||
                                    
 | 
			
		||||
  [InvalidArgumentException]        
 | 
			
		||||
  Command "udpate" is not defined.  
 | 
			
		||||
  Did you mean this?                
 | 
			
		||||
      update
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
"""
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
def composer_not_command_one_of_this():
 | 
			
		||||
    return """
 | 
			
		||||
                            
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
  [InvalidArgumentException]       
 | 
			
		||||
  Command "pdate" is not defined.  
 | 
			
		||||
  Did you mean one of these?       
 | 
			
		||||
      selfupdate                   
 | 
			
		||||
      self-update                  
 | 
			
		||||
      update                       
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
"""
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match(composer_not_command, composer_not_command_one_of_this):
 | 
			
		||||
    assert match(Command('composer udpate',
 | 
			
		||||
                         stderr=composer_not_command), None)
 | 
			
		||||
    assert match(Command('composer pdate',
 | 
			
		||||
                         stderr=composer_not_command_one_of_this), None)
 | 
			
		||||
    assert not match(Command('ls update', stderr=composer_not_command),
 | 
			
		||||
                     None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command(composer_not_command, composer_not_command_one_of_this):
 | 
			
		||||
    assert get_new_command(Command('composer udpate',
 | 
			
		||||
                                   stderr=composer_not_command), None) \
 | 
			
		||||
           == 'composer update'
 | 
			
		||||
    assert get_new_command(
 | 
			
		||||
        Command('composer pdate', stderr=composer_not_command_one_of_this),
 | 
			
		||||
        None) == 'composer selfupdate'
 | 
			
		||||
							
								
								
									
										22
									
								
								tests/rules/test_fix_alt_space.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								tests/rules/test_fix_alt_space.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,22 @@
 | 
			
		||||
# -*- encoding: utf-8 -*-
 | 
			
		||||
 | 
			
		||||
from thefuck.rules.fix_alt_space import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    """The character before 'grep' is Alt+Space, which happens frequently
 | 
			
		||||
    on the Mac when typing the pipe character (Alt+7), and holding the Alt
 | 
			
		||||
    key pressed for longer than necessary.
 | 
			
		||||
 | 
			
		||||
    """
 | 
			
		||||
    assert match(Command(u'ps -ef | grep foo',
 | 
			
		||||
                         stderr=u'-bash:  grep: command not found'), None)
 | 
			
		||||
    assert not match(Command('ps -ef | grep foo'), None)
 | 
			
		||||
    assert not match(Command(), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
    """ Replace the Alt+Space character by a simple space """
 | 
			
		||||
    assert get_new_command(Command(u'ps -ef | grep foo'), None)\
 | 
			
		||||
           == 'ps -ef | grep foo'
 | 
			
		||||
@@ -1,6 +1,6 @@
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
from thefuck.rules.git_not_command import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
@@ -31,14 +31,14 @@ def git_command():
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match(git_not_command, git_command, git_not_command_one_of_this):
 | 
			
		||||
    assert match(Command('git brnch', '', git_not_command), None)
 | 
			
		||||
    assert match(Command('git st', '', git_not_command_one_of_this), None)
 | 
			
		||||
    assert not match(Command('ls brnch', '', git_not_command), None)
 | 
			
		||||
    assert not match(Command('git branch', '', git_command), None)
 | 
			
		||||
    assert match(Command('git brnch', stderr=git_not_command), None)
 | 
			
		||||
    assert match(Command('git st', stderr=git_not_command_one_of_this), None)
 | 
			
		||||
    assert not match(Command('ls brnch', stderr=git_not_command), None)
 | 
			
		||||
    assert not match(Command('git branch', stderr=git_command), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command(git_not_command, git_not_command_one_of_this):
 | 
			
		||||
    assert get_new_command(Command('git brnch', '', git_not_command), None)\
 | 
			
		||||
    assert get_new_command(Command('git brnch', stderr=git_not_command), None)\
 | 
			
		||||
        == 'git branch'
 | 
			
		||||
    assert get_new_command(
 | 
			
		||||
        Command('git st', '', git_not_command_one_of_this), None) == 'git status'
 | 
			
		||||
    assert get_new_command(Command('git st', stderr=git_not_command_one_of_this),
 | 
			
		||||
                           None) == 'git status'
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,6 @@
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
from thefuck.rules.git_push import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
@@ -14,11 +14,11 @@ To push the current branch and set the remote as upstream, use
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match(stderr):
 | 
			
		||||
    assert match(Command('git push master', '', stderr), None)
 | 
			
		||||
    assert not match(Command('git push master', '', ''), None)
 | 
			
		||||
    assert not match(Command('ls', '', stderr), None)
 | 
			
		||||
    assert match(Command('git push master', stderr=stderr), None)
 | 
			
		||||
    assert not match(Command('git push master'), None)
 | 
			
		||||
    assert not match(Command('ls', stderr=stderr), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command(stderr):
 | 
			
		||||
    assert get_new_command(Command('', '', stderr), None)\
 | 
			
		||||
    assert get_new_command(Command(stderr=stderr), None)\
 | 
			
		||||
        == "git push --set-upstream origin master"
 | 
			
		||||
 
 | 
			
		||||
@@ -1,5 +1,5 @@
 | 
			
		||||
from mock import Mock, patch
 | 
			
		||||
from thefuck.rules. has_exists_script import match, get_new_command
 | 
			
		||||
from thefuck.rules.has_exists_script import match, get_new_command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										13
									
								
								tests/rules/test_ls_lah.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								tests/rules/test_ls_lah.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,13 @@
 | 
			
		||||
from mock import patch, Mock
 | 
			
		||||
from thefuck.rules.ls_lah import match, get_new_command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    assert match(Mock(script='ls file.py'), None)
 | 
			
		||||
    assert match(Mock(script='ls /opt'), None)
 | 
			
		||||
    assert not match(Mock(script='ls -lah /opt'), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
    assert get_new_command(Mock(script='ls file.py'), None) == 'ls -lah file.py'
 | 
			
		||||
    assert get_new_command(Mock(script='ls'), None) == 'ls -lah'
 | 
			
		||||
@@ -1,13 +1,22 @@
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules.mkdir_p import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    assert match(Command('mkdir foo/bar/baz', '', 'mkdir: foo/bar: No such file or directory'), None)
 | 
			
		||||
    assert not match(Command('mkdir foo/bar/baz', '', ''), None)
 | 
			
		||||
    assert not match(Command('mkdir foo/bar/baz', '', 'foo bar baz'), None)
 | 
			
		||||
    assert not match(Command('', '', ''), None)
 | 
			
		||||
    assert match(Command('mkdir foo/bar/baz',
 | 
			
		||||
                         stderr='mkdir: foo/bar: No such file or directory'),
 | 
			
		||||
                 None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.mark.parametrize('command', [
 | 
			
		||||
    Command('mkdir foo/bar/baz'),
 | 
			
		||||
    Command('mkdir foo/bar/baz', stderr='foo bar baz'),
 | 
			
		||||
    Command()])
 | 
			
		||||
def test_not_match(command):
 | 
			
		||||
    assert not match(command, None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
    assert get_new_command(Command('mkdir foo/bar/baz', '', ''), None) == 'mkdir -p foo/bar/baz'
 | 
			
		||||
    assert get_new_command(Command('mkdir foo/bar/baz'), None)\
 | 
			
		||||
           == 'mkdir -p foo/bar/baz'
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										25
									
								
								tests/rules/test_pip_unknown_command.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								tests/rules/test_pip_unknown_command.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,25 @@
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules.pip_unknown_command import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
def pip_unknown_cmd():
 | 
			
		||||
    return '''ERROR: unknown command "instatl" - maybe you meant "install"'''
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
def pip_unknown_cmd_without_recommend():
 | 
			
		||||
    return '''ERROR: unknown command "i"'''
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match(pip_unknown_cmd, pip_unknown_cmd_without_recommend):
 | 
			
		||||
    assert match(Command('pip instatl', stderr=pip_unknown_cmd), None)
 | 
			
		||||
    assert not match(Command('pip i',
 | 
			
		||||
                             stderr=pip_unknown_cmd_without_recommend),
 | 
			
		||||
                     None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command(pip_unknown_cmd):
 | 
			
		||||
    assert get_new_command(Command('pip instatl', stderr=pip_unknown_cmd),
 | 
			
		||||
                           None) == 'pip install'
 | 
			
		||||
@@ -1,9 +1,12 @@
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
from thefuck.rules.python_command import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    assert match(Command('temp.py', '', 'Permission denied'), None)
 | 
			
		||||
    assert not match(Command('', '', ''), None)
 | 
			
		||||
    assert match(Command('temp.py', stderr='Permission denied'), None)
 | 
			
		||||
    assert not match(Command(), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
    assert get_new_command(Command('./test_sudo.py', '', ''), None) == 'python ./test_sudo.py'
 | 
			
		||||
    assert get_new_command(Command('./test_sudo.py'), None)\
 | 
			
		||||
           == 'python ./test_sudo.py'
 | 
			
		||||
 
 | 
			
		||||
@@ -1,12 +1,20 @@
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules.rm_dir import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    assert match(Command('rm foo', '', 'rm: foo: is a directory'), None)
 | 
			
		||||
    assert not match(Command('rm foo', '', ''), None)
 | 
			
		||||
    assert not match(Command('rm foo', '', 'foo bar baz'), None)
 | 
			
		||||
    assert not match(Command('', '', ''), None)
 | 
			
		||||
@pytest.mark.parametrize('command', [
 | 
			
		||||
    Command('rm foo', stderr='rm: foo: is a directory'),
 | 
			
		||||
    Command('rm foo', stderr='rm: foo: Is a directory')])
 | 
			
		||||
def test_match(command):
 | 
			
		||||
    assert match(command, None)
 | 
			
		||||
    assert match(command, None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.mark.parametrize('command', [
 | 
			
		||||
    Command('rm foo'), Command('rm foo'), Command()])
 | 
			
		||||
def test_not_match(command):
 | 
			
		||||
    assert not match(command, None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
 
 | 
			
		||||
@@ -1,18 +1,21 @@
 | 
			
		||||
from mock import Mock
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules.rm_root import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    assert match(Mock(script='rm -rf /',
 | 
			
		||||
                      stderr='add --no-preserve-root'), None)
 | 
			
		||||
    assert not match(Mock(script='ls',
 | 
			
		||||
                          stderr='add --no-preserve-root'), None)
 | 
			
		||||
    assert not match(Mock(script='rm --no-preserve-root /',
 | 
			
		||||
                          stderr='add --no-preserve-root'), None)
 | 
			
		||||
    assert not match(Mock(script='rm -rf /',
 | 
			
		||||
                          stderr=''), None)
 | 
			
		||||
    assert match(Command(script='rm -rf /',
 | 
			
		||||
                         stderr='add --no-preserve-root'), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.mark.parametrize('command', [
 | 
			
		||||
    Command(script='ls', stderr='add --no-preserve-root'),
 | 
			
		||||
    Command(script='rm --no-preserve-root /', stderr='add --no-preserve-root'),
 | 
			
		||||
    Command(script='rm -rf /', stderr='')])
 | 
			
		||||
def test_not_match(command):
 | 
			
		||||
    assert not match(command, None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
    assert get_new_command(Mock(script='rm -rf /'), None) \
 | 
			
		||||
        == 'rm -rf / --no-preserve-root'
 | 
			
		||||
    assert get_new_command(Command(script='rm -rf /'), None) \
 | 
			
		||||
           == 'rm -rf / --no-preserve-root'
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										12
									
								
								tests/rules/test_sl_ls.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								tests/rules/test_sl_ls.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
			
		||||
 | 
			
		||||
from thefuck.rules.sl_ls import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    assert match(Command('sl'), None)
 | 
			
		||||
    assert not match(Command('ls'), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
    assert get_new_command(Command('sl'), None) == 'ls'
 | 
			
		||||
@@ -1,8 +1,9 @@
 | 
			
		||||
import os
 | 
			
		||||
import pytest
 | 
			
		||||
from mock import Mock
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
from thefuck.rules.ssh_known_hosts import match, get_new_command, remove_offending_keys
 | 
			
		||||
from thefuck.rules.ssh_known_hosts import match, get_new_command,\
 | 
			
		||||
    remove_offending_keys
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.fixture
 | 
			
		||||
@@ -43,18 +44,18 @@ Host key verification failed.""".format(path, '98.765.432.321')
 | 
			
		||||
 | 
			
		||||
def test_match(ssh_error):
 | 
			
		||||
    errormsg, _, _, _ = ssh_error
 | 
			
		||||
    assert match(Command('ssh', '', errormsg), None)
 | 
			
		||||
    assert match(Command('ssh', '', errormsg), None)
 | 
			
		||||
    assert match(Command('scp something something', '', errormsg), None)
 | 
			
		||||
    assert match(Command('scp something something', '', errormsg), None)
 | 
			
		||||
    assert not match(Command('', '', errormsg), None)
 | 
			
		||||
    assert not match(Command('notssh', '', errormsg), None)
 | 
			
		||||
    assert not match(Command('ssh', '', ''), None)
 | 
			
		||||
    assert match(Command('ssh', stderr=errormsg), None)
 | 
			
		||||
    assert match(Command('ssh', stderr=errormsg), None)
 | 
			
		||||
    assert match(Command('scp something something', stderr=errormsg), None)
 | 
			
		||||
    assert match(Command('scp something something', stderr=errormsg), None)
 | 
			
		||||
    assert not match(Command(stderr=errormsg), None)
 | 
			
		||||
    assert not match(Command('notssh', stderr=errormsg), None)
 | 
			
		||||
    assert not match(Command('ssh'), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_remove_offending_keys(ssh_error):
 | 
			
		||||
    errormsg, path, reset, known_hosts = ssh_error
 | 
			
		||||
    command = Command('ssh user@host', '', errormsg)
 | 
			
		||||
    command = Command('ssh user@host', stderr=errormsg)
 | 
			
		||||
    remove_offending_keys(command, None)
 | 
			
		||||
    expected = ['123.234.567.890 asdjkasjdakjsd\n', '111.222.333.444 qwepoiwqepoiss\n']
 | 
			
		||||
    assert known_hosts(path) == expected
 | 
			
		||||
@@ -65,5 +66,5 @@ def test_get_new_command(ssh_error, monkeypatch):
 | 
			
		||||
 | 
			
		||||
    method = Mock()
 | 
			
		||||
    monkeypatch.setattr('thefuck.rules.ssh_known_hosts.remove_offending_keys', method)
 | 
			
		||||
    assert get_new_command(Command('ssh user@host', '', errormsg), None) == 'ssh user@host'
 | 
			
		||||
    assert get_new_command(Command('ssh user@host', stderr=errormsg), None) == 'ssh user@host'
 | 
			
		||||
    assert method.call_count
 | 
			
		||||
 
 | 
			
		||||
@@ -1,13 +1,21 @@
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules.sudo import match, get_new_command
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    assert match(Command('', '', 'Permission denied'), None)
 | 
			
		||||
    assert match(Command('', '', 'permission denied'), None)
 | 
			
		||||
    assert match(Command('', '', "npm ERR! Error: EACCES, unlink"), None)
 | 
			
		||||
    assert not match(Command('', '', ''), None)
 | 
			
		||||
@pytest.mark.parametrize('stderr, stdout', [
 | 
			
		||||
    ('Permission denied', ''),
 | 
			
		||||
    ('permission denied', ''),
 | 
			
		||||
    ("npm ERR! Error: EACCES, unlink", ''),
 | 
			
		||||
    ('requested operation requires superuser privilege', ''),
 | 
			
		||||
    ('', "error: [Errno 13] Permission denied: '/usr/local/lib/python2.7/dist-packages/ipaddr.py'")])
 | 
			
		||||
def test_match(stderr, stdout):
 | 
			
		||||
    assert match(Command(stderr=stderr, stdout=stdout), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_not_match():
 | 
			
		||||
    assert not match(Command(), None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
    assert get_new_command(Command('ls', '', ''), None) == 'sudo ls'
 | 
			
		||||
    assert get_new_command(Command('ls'), None) == 'sudo ls'
 | 
			
		||||
 
 | 
			
		||||
@@ -1,25 +1,27 @@
 | 
			
		||||
# -*- encoding: utf-8 -*-
 | 
			
		||||
 | 
			
		||||
from mock import Mock
 | 
			
		||||
import pytest
 | 
			
		||||
from thefuck.rules import switch_lang
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_match():
 | 
			
		||||
    assert switch_lang.match(Mock(stderr='command not found: фзе-пуе',
 | 
			
		||||
                                  script=u'фзе-пуе'), None)
 | 
			
		||||
    assert switch_lang.match(Mock(stderr='command not found: λσ',
 | 
			
		||||
                                  script=u'λσ'), None)
 | 
			
		||||
 | 
			
		||||
    assert not switch_lang.match(Mock(stderr='command not found: pat-get',
 | 
			
		||||
                                      script=u'pat-get'), None)
 | 
			
		||||
    assert not switch_lang.match(Mock(stderr='command not found: ls',
 | 
			
		||||
                                      script=u'ls'), None)
 | 
			
		||||
    assert not switch_lang.match(Mock(stderr='some info',
 | 
			
		||||
                                      script=u'фзе-пуе'), None)
 | 
			
		||||
@pytest.mark.parametrize('command', [
 | 
			
		||||
    Command(stderr='command not found: фзе-пуе', script=u'фзе-пуе'),
 | 
			
		||||
    Command(stderr='command not found: λσ', script=u'λσ')])
 | 
			
		||||
def test_match(command):
 | 
			
		||||
    assert switch_lang.match(command, None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_new_command():
 | 
			
		||||
    assert switch_lang.get_new_command(
 | 
			
		||||
        Mock(script=u'фзе-пуе штыефдд мшь'), None) == 'apt-get install vim'
 | 
			
		||||
    assert switch_lang.get_new_command(
 | 
			
		||||
        Mock(script=u'λσ -λα'), None) == 'ls -la'
 | 
			
		||||
@pytest.mark.parametrize('command', [
 | 
			
		||||
    Command(stderr='command not found: pat-get', script=u'pat-get'),
 | 
			
		||||
    Command(stderr='command not found: ls', script=u'ls'),
 | 
			
		||||
    Command(stderr='some info', script=u'фзе-пуе')])
 | 
			
		||||
def test_not_match(command):
 | 
			
		||||
    assert not switch_lang.match(command, None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@pytest.mark.parametrize('command, new_command', [
 | 
			
		||||
    (Command(u'фзе-пуе штыефдд мшь'), 'apt-get install vim'),
 | 
			
		||||
    (Command(u'λσ -λα'), 'ls -la')])
 | 
			
		||||
def test_get_new_command(command, new_command):
 | 
			
		||||
    assert switch_lang.get_new_command(command, None) == new_command
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										89
									
								
								tests/test_conf.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										89
									
								
								tests/test_conf.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,89 @@
 | 
			
		||||
import six
 | 
			
		||||
from mock import patch, Mock
 | 
			
		||||
from thefuck import conf
 | 
			
		||||
from tests.utils import Rule
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_default():
 | 
			
		||||
    assert Rule('test', enabled_by_default=True) in conf.DEFAULT_RULES
 | 
			
		||||
    assert Rule('test', enabled_by_default=False) not in conf.DEFAULT_RULES
 | 
			
		||||
    assert Rule('test', enabled_by_default=False) in (conf.DEFAULT_RULES + ['test'])
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_settings_defaults():
 | 
			
		||||
    with patch('thefuck.conf.load_source', return_value=object()), \
 | 
			
		||||
         patch('thefuck.conf.os.environ', new_callable=lambda: {}):
 | 
			
		||||
        for key, val in conf.DEFAULT_SETTINGS.items():
 | 
			
		||||
            assert getattr(conf.get_settings(Mock()), key) == val
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_settings_from_file():
 | 
			
		||||
    with patch('thefuck.conf.load_source', return_value=Mock(rules=['test'],
 | 
			
		||||
                                                             wait_command=10,
 | 
			
		||||
                                                             require_confirmation=True,
 | 
			
		||||
                                                             no_colors=True)), \
 | 
			
		||||
         patch('thefuck.conf.os.environ', new_callable=lambda: {}):
 | 
			
		||||
        settings = conf.get_settings(Mock())
 | 
			
		||||
        assert settings.rules == ['test']
 | 
			
		||||
        assert settings.wait_command == 10
 | 
			
		||||
        assert settings.require_confirmation is True
 | 
			
		||||
        assert settings.no_colors is True
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_settings_from_file_with_DEFAULT():
 | 
			
		||||
    with patch('thefuck.conf.load_source', return_value=Mock(rules=conf.DEFAULT_RULES + ['test'],
 | 
			
		||||
                                                             wait_command=10,
 | 
			
		||||
                                                             require_confirmation=True,
 | 
			
		||||
                                                             no_colors=True)), \
 | 
			
		||||
         patch('thefuck.conf.os.environ', new_callable=lambda: {}):
 | 
			
		||||
        settings = conf.get_settings(Mock())
 | 
			
		||||
        assert settings.rules == conf.DEFAULT_RULES + ['test']
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_settings_from_env():
 | 
			
		||||
    with patch('thefuck.conf.load_source', return_value=Mock(rules=['test'],
 | 
			
		||||
                                                             wait_command=10)), \
 | 
			
		||||
         patch('thefuck.conf.os.environ',
 | 
			
		||||
               new_callable=lambda: {'THEFUCK_RULES': 'bash:lisp',
 | 
			
		||||
                                     'THEFUCK_WAIT_COMMAND': '55',
 | 
			
		||||
                                     'THEFUCK_REQUIRE_CONFIRMATION': 'true',
 | 
			
		||||
                                     'THEFUCK_NO_COLORS': 'false'}):
 | 
			
		||||
        settings = conf.get_settings(Mock())
 | 
			
		||||
        assert settings.rules == ['bash', 'lisp']
 | 
			
		||||
        assert settings.wait_command == 55
 | 
			
		||||
        assert settings.require_confirmation is True
 | 
			
		||||
        assert settings.no_colors is False
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_settings_from_env_with_DEFAULT():
 | 
			
		||||
    with patch('thefuck.conf.load_source', return_value=Mock()), \
 | 
			
		||||
         patch('thefuck.conf.os.environ', new_callable=lambda: {'THEFUCK_RULES': 'DEFAULT_RULES:bash:lisp'}):
 | 
			
		||||
        settings = conf.get_settings(Mock())
 | 
			
		||||
        assert settings.rules == conf.DEFAULT_RULES + ['bash', 'lisp']
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_initialize_settings_file_ignore_if_exists():
 | 
			
		||||
    settings_path_mock = Mock(is_file=Mock(return_value=True), open=Mock())
 | 
			
		||||
    user_dir_mock = Mock(joinpath=Mock(return_value=settings_path_mock))
 | 
			
		||||
    conf.initialize_settings_file(user_dir_mock)
 | 
			
		||||
    assert settings_path_mock.is_file.call_count == 1
 | 
			
		||||
    assert not settings_path_mock.open.called
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_initialize_settings_file_create_if_exists_not():
 | 
			
		||||
    settings_file = six.StringIO()
 | 
			
		||||
    settings_path_mock = Mock(
 | 
			
		||||
        is_file=Mock(return_value=False),
 | 
			
		||||
        open=Mock(return_value=Mock(
 | 
			
		||||
            __exit__=lambda *args: None, __enter__=lambda *args: settings_file
 | 
			
		||||
        )),
 | 
			
		||||
    )
 | 
			
		||||
    user_dir_mock = Mock(joinpath=Mock(return_value=settings_path_mock))
 | 
			
		||||
    conf.initialize_settings_file(user_dir_mock)
 | 
			
		||||
    settings_file_contents = settings_file.getvalue()
 | 
			
		||||
    assert settings_path_mock.is_file.call_count == 1
 | 
			
		||||
    assert settings_path_mock.open.call_count == 1
 | 
			
		||||
    assert conf.SETTINGS_HEADER in settings_file_contents
 | 
			
		||||
    for setting in conf.DEFAULT_SETTINGS.items():
 | 
			
		||||
        assert '# {} = {}\n'.format(*setting) in settings_file_contents
 | 
			
		||||
    settings_file.close()
 | 
			
		||||
@@ -1,25 +1,8 @@
 | 
			
		||||
from subprocess import PIPE
 | 
			
		||||
from pathlib import PosixPath, Path
 | 
			
		||||
from mock import patch, Mock
 | 
			
		||||
from thefuck import main
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_settings():
 | 
			
		||||
    with patch('thefuck.main.load_source', return_value=Mock(rules=['bash'])):
 | 
			
		||||
        assert main.get_settings(Path('/')).rules == ['bash']
 | 
			
		||||
    with patch('thefuck.main.load_source', return_value=Mock(spec=[])):
 | 
			
		||||
        assert main.get_settings(Path('/')).rules is None
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_is_rule_enabled():
 | 
			
		||||
    assert main.is_rule_enabled(Mock(rules=None),
 | 
			
		||||
                                main.Rule('bash', None, None, True))
 | 
			
		||||
    assert not main.is_rule_enabled(Mock(rules=None),
 | 
			
		||||
                                    main.Rule('bash', None, None, False))
 | 
			
		||||
    assert main.is_rule_enabled(Mock(rules=['bash']),
 | 
			
		||||
                                main.Rule('bash', None, None, True))
 | 
			
		||||
    assert not main.is_rule_enabled(Mock(rules=['bash']),
 | 
			
		||||
                                    main.Rule('lisp', None, None, True))
 | 
			
		||||
from thefuck import main, conf, types
 | 
			
		||||
from tests.utils import Rule, Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_load_rule():
 | 
			
		||||
@@ -31,7 +14,7 @@ def test_load_rule():
 | 
			
		||||
                   get_new_command=get_new_command,
 | 
			
		||||
                   enabled_by_default=True)) as load_source:
 | 
			
		||||
        assert main.load_rule(Path('/rules/bash.py')) \
 | 
			
		||||
               == main.Rule('bash', match, get_new_command, True)
 | 
			
		||||
               == Rule('bash', match, get_new_command)
 | 
			
		||||
        load_source.assert_called_once_with('bash', '/rules/bash.py')
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@@ -43,14 +26,16 @@ def test_get_rules():
 | 
			
		||||
        glob.return_value = [PosixPath('bash.py'), PosixPath('lisp.py')]
 | 
			
		||||
        assert list(main.get_rules(
 | 
			
		||||
            Path('~'),
 | 
			
		||||
            Mock(rules=None))) == [main.Rule('bash', 'bash', 'bash', True),
 | 
			
		||||
                                   main.Rule('lisp', 'lisp', 'lisp', True),
 | 
			
		||||
                                   main.Rule('bash', 'bash', 'bash', True),
 | 
			
		||||
                                   main.Rule('lisp', 'lisp', 'lisp', True)]
 | 
			
		||||
            Mock(rules=conf.DEFAULT_RULES))) \
 | 
			
		||||
               == [Rule('bash', 'bash', 'bash'),
 | 
			
		||||
                   Rule('lisp', 'lisp', 'lisp'),
 | 
			
		||||
                   Rule('bash', 'bash', 'bash'),
 | 
			
		||||
                   Rule('lisp', 'lisp', 'lisp')]
 | 
			
		||||
        assert list(main.get_rules(
 | 
			
		||||
            Path('~'),
 | 
			
		||||
            Mock(rules=['bash']))) == [main.Rule('bash', 'bash', 'bash', True),
 | 
			
		||||
                                       main.Rule('bash', 'bash', 'bash', True)]
 | 
			
		||||
            Mock(rules=types.RulesNamesList(['bash'])))) \
 | 
			
		||||
               == [Rule('bash', 'bash', 'bash'),
 | 
			
		||||
                   Rule('bash', 'bash', 'bash')]
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_command():
 | 
			
		||||
@@ -63,7 +48,7 @@ def test_get_command():
 | 
			
		||||
        Popen.return_value.stderr.read.return_value = b'stderr'
 | 
			
		||||
        assert main.get_command(Mock(), ['thefuck', 'apt-get',
 | 
			
		||||
                                         'search', 'vim']) \
 | 
			
		||||
               == main.Command('apt-get search vim', 'stdout', 'stderr')
 | 
			
		||||
               == Command('apt-get search vim', 'stdout', 'stderr')
 | 
			
		||||
        Popen.assert_called_once_with('apt-get search vim',
 | 
			
		||||
                                      shell=True,
 | 
			
		||||
                                      stdout=PIPE,
 | 
			
		||||
@@ -73,12 +58,12 @@ def test_get_command():
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_get_matched_rule(capsys):
 | 
			
		||||
    rules = [main.Rule('', lambda x, _: x.script == 'cd ..', None, True),
 | 
			
		||||
             main.Rule('', lambda *_: False, None, True),
 | 
			
		||||
             main.Rule('rule', Mock(side_effect=OSError('Denied')), None, True)]
 | 
			
		||||
    assert main.get_matched_rule(main.Command('ls', '', ''),
 | 
			
		||||
    rules = [Rule('', lambda x, _: x.script == 'cd ..'),
 | 
			
		||||
             Rule('', lambda *_: False),
 | 
			
		||||
             Rule('rule', Mock(side_effect=OSError('Denied')))]
 | 
			
		||||
    assert main.get_matched_rule(Command('ls'),
 | 
			
		||||
                                 rules, Mock(no_colors=True)) is None
 | 
			
		||||
    assert main.get_matched_rule(main.Command('cd ..', '', ''),
 | 
			
		||||
    assert main.get_matched_rule(Command('cd ..'),
 | 
			
		||||
                                 rules, Mock(no_colors=True)) == rules[0]
 | 
			
		||||
    assert capsys.readouterr()[1].split('\n')[0] \
 | 
			
		||||
           == '[WARN] Rule rule:'
 | 
			
		||||
@@ -86,11 +71,11 @@ def test_get_matched_rule(capsys):
 | 
			
		||||
 | 
			
		||||
def test_run_rule(capsys):
 | 
			
		||||
    with patch('thefuck.main.confirm', return_value=True):
 | 
			
		||||
        main.run_rule(main.Rule('', None, lambda *_: 'new-command', True),
 | 
			
		||||
        main.run_rule(Rule(get_new_command=lambda *_: 'new-command'),
 | 
			
		||||
                      None, None)
 | 
			
		||||
        assert capsys.readouterr() == ('new-command\n', '')
 | 
			
		||||
    with patch('thefuck.main.confirm', return_value=False):
 | 
			
		||||
        main.run_rule(main.Rule('', None, lambda *_: 'new-command', True),
 | 
			
		||||
        main.run_rule(Rule(get_new_command=lambda *_: 'new-command'),
 | 
			
		||||
                      None, None)
 | 
			
		||||
        assert capsys.readouterr() == ('', '')
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										15
									
								
								tests/test_types.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								tests/test_types.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,15 @@
 | 
			
		||||
from thefuck.types import Rule, RulesNamesList, Settings
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_rules_names_list():
 | 
			
		||||
    assert RulesNamesList(['bash', 'lisp']) == ['bash', 'lisp']
 | 
			
		||||
    assert RulesNamesList(['bash', 'lisp']) == RulesNamesList(['bash', 'lisp'])
 | 
			
		||||
    assert Rule('lisp', None, None, False) in RulesNamesList(['lisp'])
 | 
			
		||||
    assert Rule('bash', None, None, False) not in RulesNamesList(['lisp'])
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_update_settings():
 | 
			
		||||
    settings = Settings({'key': 'val'})
 | 
			
		||||
    new_settings = settings.update(key='new-val')
 | 
			
		||||
    assert new_settings.key == 'new-val'
 | 
			
		||||
    assert settings.key == 'val'
 | 
			
		||||
@@ -1,17 +1,26 @@
 | 
			
		||||
from mock import Mock
 | 
			
		||||
from thefuck.utils import sudo_support
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
from thefuck.utils import sudo_support, wrap_settings
 | 
			
		||||
from thefuck.types import Settings
 | 
			
		||||
from tests.utils import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_wrap_settings():
 | 
			
		||||
    fn = lambda _, settings: settings
 | 
			
		||||
    assert wrap_settings({'key': 'val'})(fn)(None, Settings({})) \
 | 
			
		||||
           == {'key': 'val'}
 | 
			
		||||
    assert wrap_settings({'key': 'new-val'})(fn)(
 | 
			
		||||
        None, Settings({'key': 'val'})) == {'key': 'new-val'}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def test_sudo_support():
 | 
			
		||||
    fn = Mock(return_value=True, __name__='')
 | 
			
		||||
    assert sudo_support(fn)(Command('sudo ls', 'out', 'err'), None)
 | 
			
		||||
    fn.assert_called_once_with(Command('ls', 'out', 'err'), None)
 | 
			
		||||
    assert sudo_support(fn)(Command('sudo ls'), None)
 | 
			
		||||
    fn.assert_called_once_with(Command('ls'), None)
 | 
			
		||||
 | 
			
		||||
    fn.return_value = False
 | 
			
		||||
    assert not sudo_support(fn)(Command('sudo ls', 'out', 'err'), None)
 | 
			
		||||
    assert not sudo_support(fn)(Command('sudo ls'), None)
 | 
			
		||||
 | 
			
		||||
    fn.return_value = 'pwd'
 | 
			
		||||
    assert sudo_support(fn)(Command('sudo ls', 'out', 'err'), None) == 'sudo pwd'
 | 
			
		||||
    assert sudo_support(fn)(Command('sudo ls'), None) == 'sudo pwd'
 | 
			
		||||
 | 
			
		||||
    assert sudo_support(fn)(Command('ls', 'out', 'err'), None) == 'pwd'
 | 
			
		||||
    assert sudo_support(fn)(Command('ls'), None) == 'pwd'
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										11
									
								
								tests/utils.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								tests/utils.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,11 @@
 | 
			
		||||
from thefuck import types
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def Command(script='', stdout='', stderr=''):
 | 
			
		||||
    return types.Command(script, stdout, stderr)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def Rule(name='', match=lambda *_: True,
 | 
			
		||||
         get_new_command=lambda *_: '',
 | 
			
		||||
         enabled_by_default=True):
 | 
			
		||||
    return types.Rule(name, match, get_new_command, enabled_by_default)
 | 
			
		||||
							
								
								
									
										116
									
								
								thefuck/conf.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										116
									
								
								thefuck/conf.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,116 @@
 | 
			
		||||
from copy import copy
 | 
			
		||||
from imp import load_source
 | 
			
		||||
import os
 | 
			
		||||
import sys
 | 
			
		||||
from six import text_type
 | 
			
		||||
from . import logs, types
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class _DefaultRulesNames(types.RulesNamesList):
 | 
			
		||||
    def __add__(self, items):
 | 
			
		||||
        return _DefaultRulesNames(list(self) + items)
 | 
			
		||||
 | 
			
		||||
    def __contains__(self, item):
 | 
			
		||||
        return item.enabled_by_default or \
 | 
			
		||||
               super(_DefaultRulesNames, self).__contains__(item)
 | 
			
		||||
 | 
			
		||||
    def __eq__(self, other):
 | 
			
		||||
        if isinstance(other, _DefaultRulesNames):
 | 
			
		||||
            return super(_DefaultRulesNames, self).__eq__(other)
 | 
			
		||||
        else:
 | 
			
		||||
            return False
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
DEFAULT_RULES = _DefaultRulesNames([])
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
DEFAULT_SETTINGS = {'rules': DEFAULT_RULES,
 | 
			
		||||
                    'wait_command': 3,
 | 
			
		||||
                    'require_confirmation': False,
 | 
			
		||||
                    'no_colors': False}
 | 
			
		||||
 | 
			
		||||
ENV_TO_ATTR = {'THEFUCK_RULES': 'rules',
 | 
			
		||||
               'THEFUCK_WAIT_COMMAND': 'wait_command',
 | 
			
		||||
               'THEFUCK_REQUIRE_CONFIRMATION': 'require_confirmation',
 | 
			
		||||
               'THEFUCK_NO_COLORS': 'no_colors'}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
SETTINGS_HEADER = u"""# ~/.thefuck/settings.py: The Fuck settings file
 | 
			
		||||
#
 | 
			
		||||
# The rules are defined as in the example bellow:
 | 
			
		||||
#
 | 
			
		||||
# rules = ['cd_parent', 'git_push', 'python_command', 'sudo']
 | 
			
		||||
#
 | 
			
		||||
# The default values are as follows. Uncomment and change to fit your needs.
 | 
			
		||||
# See https://github.com/nvbn/thefuck#settings for more information.
 | 
			
		||||
#
 | 
			
		||||
 | 
			
		||||
"""
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _settings_from_file(user_dir):
 | 
			
		||||
    """Loads settings from file."""
 | 
			
		||||
    settings = load_source('settings',
 | 
			
		||||
                           text_type(user_dir.joinpath('settings.py')))
 | 
			
		||||
    return {key: getattr(settings, key)
 | 
			
		||||
            for key in DEFAULT_SETTINGS.keys()
 | 
			
		||||
            if hasattr(settings, key)}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _rules_from_env(val):
 | 
			
		||||
    """Transforms rules list from env-string to python."""
 | 
			
		||||
    val = val.split(':')
 | 
			
		||||
    if 'DEFAULT_RULES' in val:
 | 
			
		||||
        val = DEFAULT_RULES + [rule for rule in val if rule != 'DEFAULT_RULES']
 | 
			
		||||
    return val
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _val_from_env(env, attr):
 | 
			
		||||
    """Transforms env-strings to python."""
 | 
			
		||||
    val = os.environ[env]
 | 
			
		||||
    if attr == 'rules':
 | 
			
		||||
        val = _rules_from_env(val)
 | 
			
		||||
    elif attr == 'wait_command':
 | 
			
		||||
        val = int(val)
 | 
			
		||||
    elif attr in ('require_confirmation', 'no_colors'):
 | 
			
		||||
        val = val.lower() == 'true'
 | 
			
		||||
    return val
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _settings_from_env():
 | 
			
		||||
    """Loads settings from env."""
 | 
			
		||||
    return {attr: _val_from_env(env, attr)
 | 
			
		||||
            for env, attr in ENV_TO_ATTR.items()
 | 
			
		||||
            if env in os.environ}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_settings(user_dir):
 | 
			
		||||
    """Returns settings filled with values from `settings.py` and env."""
 | 
			
		||||
    conf = copy(DEFAULT_SETTINGS)
 | 
			
		||||
    try:
 | 
			
		||||
        conf.update(_settings_from_file(user_dir))
 | 
			
		||||
    except Exception:
 | 
			
		||||
        logs.exception("Can't load settings from file",
 | 
			
		||||
                       sys.exc_info(),
 | 
			
		||||
                       types.Settings(conf))
 | 
			
		||||
 | 
			
		||||
    try:
 | 
			
		||||
        conf.update(_settings_from_env())
 | 
			
		||||
    except Exception:
 | 
			
		||||
        logs.exception("Can't load settings from env",
 | 
			
		||||
                       sys.exc_info(),
 | 
			
		||||
                       types.Settings(conf))
 | 
			
		||||
 | 
			
		||||
    if not isinstance(conf['rules'], types.RulesNamesList):
 | 
			
		||||
        conf['rules'] = types.RulesNamesList(conf['rules'])
 | 
			
		||||
 | 
			
		||||
    return types.Settings(conf)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def initialize_settings_file(user_dir):
 | 
			
		||||
    settings_path = user_dir.joinpath('settings.py')
 | 
			
		||||
    if not settings_path.is_file():
 | 
			
		||||
        with settings_path.open(mode='w') as settings_file:
 | 
			
		||||
            settings_file.write(SETTINGS_HEADER)
 | 
			
		||||
            for setting in DEFAULT_SETTINGS.items():
 | 
			
		||||
                settings_file.write(u'# {} = {}\n'.format(*setting))
 | 
			
		||||
@@ -11,17 +11,21 @@ def color(color_, settings):
 | 
			
		||||
        return color_
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def rule_failed(rule, exc_info, settings):
 | 
			
		||||
def exception(title, exc_info, settings):
 | 
			
		||||
    sys.stderr.write(
 | 
			
		||||
        u'{warn}[WARN] Rule {name}:{reset}\n{trace}'
 | 
			
		||||
        u'{warn}[WARN] {title}:{reset}\n{trace}'
 | 
			
		||||
        u'{warn}----------------------------{reset}\n\n'.format(
 | 
			
		||||
            warn=color(colorama.Back.RED + colorama.Fore.WHITE
 | 
			
		||||
                       + colorama.Style.BRIGHT, settings),
 | 
			
		||||
            reset=color(colorama.Style.RESET_ALL, settings),
 | 
			
		||||
            name=rule.name,
 | 
			
		||||
            title=title,
 | 
			
		||||
            trace=''.join(format_exception(*exc_info))))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def rule_failed(rule, exc_info, settings):
 | 
			
		||||
    exception('Rule {}'.format(rule.name), exc_info, settings)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def show_command(new_command, settings):
 | 
			
		||||
    sys.stderr.write('{bold}{command}{reset}\n'.format(
 | 
			
		||||
        command=new_command,
 | 
			
		||||
 
 | 
			
		||||
@@ -1,4 +1,3 @@
 | 
			
		||||
from collections import namedtuple
 | 
			
		||||
from imp import load_source
 | 
			
		||||
from pathlib import Path
 | 
			
		||||
from os.path import expanduser
 | 
			
		||||
@@ -7,12 +6,7 @@ import os
 | 
			
		||||
import sys
 | 
			
		||||
from psutil import Process, TimeoutExpired
 | 
			
		||||
import colorama
 | 
			
		||||
from thefuck import logs
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
Command = namedtuple('Command', ('script', 'stdout', 'stderr'))
 | 
			
		||||
Rule = namedtuple('Rule', ('name', 'match', 'get_new_command',
 | 
			
		||||
                           'enabled_by_default'))
 | 
			
		||||
from . import logs, conf, types
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def setup_user_dir():
 | 
			
		||||
@@ -21,52 +15,28 @@ def setup_user_dir():
 | 
			
		||||
    rules_dir = user_dir.joinpath('rules')
 | 
			
		||||
    if not rules_dir.is_dir():
 | 
			
		||||
        rules_dir.mkdir(parents=True)
 | 
			
		||||
    user_dir.joinpath('settings.py').touch()
 | 
			
		||||
    conf.initialize_settings_file(user_dir)
 | 
			
		||||
    return user_dir
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_settings(user_dir):
 | 
			
		||||
    """Returns prepared settings module."""
 | 
			
		||||
    settings = load_source('settings',
 | 
			
		||||
                           str(user_dir.joinpath('settings.py')))
 | 
			
		||||
    settings.__dict__.setdefault('rules', None)
 | 
			
		||||
    settings.__dict__.setdefault('wait_command', 3)
 | 
			
		||||
    settings.__dict__.setdefault('require_confirmation', False)
 | 
			
		||||
    settings.__dict__.setdefault('no_colors', False)
 | 
			
		||||
    return settings
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def is_rule_enabled(settings, rule):
 | 
			
		||||
    """Returns `True` when rule mentioned in `rules` or `rules`
 | 
			
		||||
    isn't defined.
 | 
			
		||||
 | 
			
		||||
    """
 | 
			
		||||
    if settings.rules is None and rule.enabled_by_default:
 | 
			
		||||
        return True
 | 
			
		||||
    elif settings.rules and rule.name in settings.rules:
 | 
			
		||||
        return True
 | 
			
		||||
    else:
 | 
			
		||||
        return False
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def load_rule(rule):
 | 
			
		||||
    """Imports rule module and returns it."""
 | 
			
		||||
    rule_module = load_source(rule.name[:-3], str(rule))
 | 
			
		||||
    return Rule(rule.name[:-3], rule_module.match,
 | 
			
		||||
                rule_module.get_new_command,
 | 
			
		||||
                getattr(rule_module, 'enabled_by_default', True))
 | 
			
		||||
    return types.Rule(rule.name[:-3], rule_module.match,
 | 
			
		||||
                      rule_module.get_new_command,
 | 
			
		||||
                      getattr(rule_module, 'enabled_by_default', True))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_rules(user_dir, settings):
 | 
			
		||||
    """Returns all enabled rules."""
 | 
			
		||||
    bundled = Path(__file__).parent\
 | 
			
		||||
                            .joinpath('rules')\
 | 
			
		||||
                            .glob('*.py')
 | 
			
		||||
    bundled = Path(__file__).parent \
 | 
			
		||||
        .joinpath('rules') \
 | 
			
		||||
        .glob('*.py')
 | 
			
		||||
    user = user_dir.joinpath('rules').glob('*.py')
 | 
			
		||||
    for rule in sorted(list(bundled)) + list(user):
 | 
			
		||||
        if rule.name != '__init__.py':
 | 
			
		||||
            loaded_rule = load_rule(rule)
 | 
			
		||||
            if is_rule_enabled(settings, loaded_rule):
 | 
			
		||||
            if loaded_rule in settings.rules:
 | 
			
		||||
                yield loaded_rule
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@@ -101,8 +71,8 @@ def get_command(settings, args):
 | 
			
		||||
    result = Popen(script, shell=True, stdout=PIPE, stderr=PIPE,
 | 
			
		||||
                   env=dict(os.environ, LANG='C'))
 | 
			
		||||
    if wait_output(settings, result):
 | 
			
		||||
        return Command(script, result.stdout.read().decode('utf-8'),
 | 
			
		||||
                       result.stderr.read().decode('utf-8'))
 | 
			
		||||
        return types.Command(script, result.stdout.read().decode('utf-8'),
 | 
			
		||||
                             result.stderr.read().decode('utf-8'))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_matched_rule(command, rules, settings):
 | 
			
		||||
@@ -145,7 +115,7 @@ def is_second_run(command):
 | 
			
		||||
def main():
 | 
			
		||||
    colorama.init()
 | 
			
		||||
    user_dir = setup_user_dir()
 | 
			
		||||
    settings = get_settings(user_dir)
 | 
			
		||||
    settings = conf.get_settings(user_dir)
 | 
			
		||||
 | 
			
		||||
    command = get_command(settings, sys.argv)
 | 
			
		||||
    if command:
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										23
									
								
								thefuck/rules/apt_get.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										23
									
								
								thefuck/rules/apt_get.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,23 @@
 | 
			
		||||
try:
 | 
			
		||||
    import CommandNotFound
 | 
			
		||||
except ImportError:
 | 
			
		||||
    enabled_by_default = False
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    if 'not found' in command.stderr:
 | 
			
		||||
        try:
 | 
			
		||||
            c = CommandNotFound.CommandNotFound()
 | 
			
		||||
            pkgs = c.getPackages(command.script.split(" ")[0])
 | 
			
		||||
            name, _ = pkgs[0]
 | 
			
		||||
            return True
 | 
			
		||||
        except IndexError:
 | 
			
		||||
            # IndexError is thrown when no matching package is found
 | 
			
		||||
            return False
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_new_command(command, settings):
 | 
			
		||||
    c = CommandNotFound.CommandNotFound()
 | 
			
		||||
    pkgs = c.getPackages(command.script.split(" ")[0])
 | 
			
		||||
    name, _ = pkgs[0]
 | 
			
		||||
    return "sudo apt-get install {} && {}".format(name, command.script)
 | 
			
		||||
							
								
								
									
										43
									
								
								thefuck/rules/brew_install.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										43
									
								
								thefuck/rules/brew_install.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,43 @@
 | 
			
		||||
import difflib
 | 
			
		||||
import os
 | 
			
		||||
import re
 | 
			
		||||
from subprocess import check_output
 | 
			
		||||
 | 
			
		||||
import thefuck.logs
 | 
			
		||||
 | 
			
		||||
# Formulars are base on each local system's status
 | 
			
		||||
brew_formulas = []
 | 
			
		||||
try:
 | 
			
		||||
    brew_path_prefix = check_output(['brew', '--prefix']).strip()
 | 
			
		||||
    brew_formula_path = brew_path_prefix + '/Library/Formula'
 | 
			
		||||
 | 
			
		||||
    for file_name in os.listdir(brew_formula_path):
 | 
			
		||||
        if file_name.endswith('.rb'):
 | 
			
		||||
            brew_formulas.append(file_name.replace('.rb', ''))
 | 
			
		||||
except:
 | 
			
		||||
    pass
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _get_similar_formulars(formula_name):
 | 
			
		||||
    return difflib.get_close_matches(formula_name, brew_formulas, 1, 0.85)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    is_proper_command = ('brew install' in command.script and
 | 
			
		||||
                         'No available formula' in command.stderr)
 | 
			
		||||
 | 
			
		||||
    has_possible_formulas = False
 | 
			
		||||
    if is_proper_command:
 | 
			
		||||
        formula = re.findall(r'Error: No available formula for ([a-z]+)',
 | 
			
		||||
                             command.stderr)[0]
 | 
			
		||||
        has_possible_formulas = len(_get_similar_formulars(formula)) > 0
 | 
			
		||||
 | 
			
		||||
    return has_possible_formulas
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_new_command(command, settings):
 | 
			
		||||
    not_exist_formula = re.findall(r'Error: No available formula for ([a-z]+)',
 | 
			
		||||
                                   command.stderr)[0]
 | 
			
		||||
    exist_formula = _get_similar_formulars(not_exist_formula)[0]
 | 
			
		||||
 | 
			
		||||
    return command.script.replace(not_exist_formula, exist_formula, 1)
 | 
			
		||||
							
								
								
									
										33
									
								
								thefuck/rules/brew_unknown_command.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										33
									
								
								thefuck/rules/brew_unknown_command.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,33 @@
 | 
			
		||||
import difflib
 | 
			
		||||
import re
 | 
			
		||||
import thefuck.logs
 | 
			
		||||
 | 
			
		||||
# This commands are based on Homebrew 0.9.5
 | 
			
		||||
brew_commands = ['info', 'home', 'options', 'install', 'uninstall', 'search',
 | 
			
		||||
                 'list', 'update', 'upgrade', 'pin', 'unpin', 'doctor',
 | 
			
		||||
                 'create', 'edit']
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _get_similar_commands(command):
 | 
			
		||||
    return difflib.get_close_matches(command, brew_commands)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    is_proper_command = ('brew' in command.script and
 | 
			
		||||
                         'Unknown command' in command.stderr)
 | 
			
		||||
 | 
			
		||||
    has_possible_commands = False
 | 
			
		||||
    if is_proper_command:
 | 
			
		||||
        broken_cmd = re.findall(r'Error: Unknown command: ([a-z]+)',
 | 
			
		||||
                                command.stderr)[0]
 | 
			
		||||
        has_possible_commands = len(_get_similar_commands(broken_cmd)) > 0
 | 
			
		||||
 | 
			
		||||
    return has_possible_commands
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_new_command(command, settings):
 | 
			
		||||
    broken_cmd = re.findall(r'Error: Unknown command: ([a-z]+)',
 | 
			
		||||
                            command.stderr)[0]
 | 
			
		||||
    new_cmd = _get_similar_commands(broken_cmd)[0]
 | 
			
		||||
 | 
			
		||||
    return command.script.replace(broken_cmd, new_cmd, 1)
 | 
			
		||||
							
								
								
									
										14
									
								
								thefuck/rules/cd_mkdir.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								thefuck/rules/cd_mkdir.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
			
		||||
import re
 | 
			
		||||
from thefuck.utils import sudo_support
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@sudo_support
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    return (command.script.startswith('cd ')
 | 
			
		||||
        and ('no such file or directory' in command.stderr.lower()
 | 
			
		||||
            or 'cd: can\'t cd to' in command.stderr.lower()))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@sudo_support
 | 
			
		||||
def get_new_command(command, settings):
 | 
			
		||||
    return re.sub(r'^cd (.*)', 'mkdir -p \\1 && cd \\1', command.script)
 | 
			
		||||
							
								
								
									
										15
									
								
								thefuck/rules/composer_not_command.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								thefuck/rules/composer_not_command.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,15 @@
 | 
			
		||||
import re
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    return ('composer' in command.script
 | 
			
		||||
            and ('did you mean this?' in command.stderr.lower()
 | 
			
		||||
                 or 'did you mean one of these?' in command.stderr.lower()))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_new_command(command, settings):
 | 
			
		||||
    broken_cmd = re.findall(r"Command \"([^']*)\" is not defined", command.stderr)[0]
 | 
			
		||||
    new_cmd = re.findall(r'Did you mean this\?[^\n]*\n\s*([^\n]*)', command.stderr)
 | 
			
		||||
    if not new_cmd:
 | 
			
		||||
        new_cmd = re.findall(r'Did you mean one of these\?[^\n]*\n\s*([^\n]*)', command.stderr)
 | 
			
		||||
    return command.script.replace(broken_cmd, new_cmd[0].strip(), 1)
 | 
			
		||||
							
								
								
									
										15
									
								
								thefuck/rules/fix_alt_space.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								thefuck/rules/fix_alt_space.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,15 @@
 | 
			
		||||
# -*- encoding: utf-8 -*-
 | 
			
		||||
 | 
			
		||||
import re
 | 
			
		||||
from thefuck.utils import sudo_support
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@sudo_support
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    return ('command not found' in command.stderr.lower()
 | 
			
		||||
            and u' ' in command.script)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@sudo_support
 | 
			
		||||
def get_new_command(command, settings):
 | 
			
		||||
    return re.sub(u' ', ' ', command.script)
 | 
			
		||||
							
								
								
									
										11
									
								
								thefuck/rules/ls_lah.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								thefuck/rules/ls_lah.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,11 @@
 | 
			
		||||
enabled_by_default = False
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    return 'ls' in command.script and not ('ls -' in command.script)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_new_command(command, settings):
 | 
			
		||||
    command = command.script.split(' ')
 | 
			
		||||
    command[0] = 'ls -lah'
 | 
			
		||||
    return ' '.join(command)
 | 
			
		||||
							
								
								
									
										15
									
								
								thefuck/rules/pip_unknown_command.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								thefuck/rules/pip_unknown_command.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,15 @@
 | 
			
		||||
import re
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    return ('pip' in command.script and
 | 
			
		||||
            'unknown command' in command.stderr and
 | 
			
		||||
            'maybe you meant' in command.stderr)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_new_command(command, settings):
 | 
			
		||||
    broken_cmd = re.findall(r'ERROR: unknown command \"([a-z]+)\"',
 | 
			
		||||
                            command.stderr)[0]
 | 
			
		||||
    new_cmd = re.findall(r'maybe you meant \"([a-z]+)\"', command.stderr)[0]
 | 
			
		||||
 | 
			
		||||
    return command.script.replace(broken_cmd, new_cmd, 1)
 | 
			
		||||
@@ -5,7 +5,7 @@ from thefuck.utils import sudo_support
 | 
			
		||||
@sudo_support
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    return ('rm' in command.script
 | 
			
		||||
            and 'is a directory' in command.stderr)
 | 
			
		||||
            and 'is a directory' in command.stderr.lower())
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@sudo_support
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										14
									
								
								thefuck/rules/sl_ls.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								thefuck/rules/sl_ls.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
			
		||||
"""
 | 
			
		||||
This happens way too often
 | 
			
		||||
 | 
			
		||||
When typing really fast cause I'm a 1337 H4X0R,
 | 
			
		||||
I often fuck up 'ls' and type 'sl'. No more!
 | 
			
		||||
"""
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    return command.script == 'sl'
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_new_command(command, settings):
 | 
			
		||||
    return 'ls'
 | 
			
		||||
@@ -6,12 +6,15 @@ patterns = ['permission denied',
 | 
			
		||||
            'Operation not permitted',
 | 
			
		||||
            'root privilege',
 | 
			
		||||
            'This command has to be run under the root user.',
 | 
			
		||||
            'You need to be root to perform this command.']
 | 
			
		||||
            'This operation requires root.',
 | 
			
		||||
            'You need to be root to perform this command.',
 | 
			
		||||
            'requested operation requires superuser privilege']
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def match(command, settings):
 | 
			
		||||
    for pattern in patterns:
 | 
			
		||||
        if pattern.lower() in command.stderr.lower():
 | 
			
		||||
        if pattern.lower() in command.stderr.lower()\
 | 
			
		||||
                or pattern.lower() in command.stdout.lower():
 | 
			
		||||
            return True
 | 
			
		||||
    return False
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										26
									
								
								thefuck/types.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										26
									
								
								thefuck/types.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,26 @@
 | 
			
		||||
from collections import namedtuple
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
Command = namedtuple('Command', ('script', 'stdout', 'stderr'))
 | 
			
		||||
 | 
			
		||||
Rule = namedtuple('Rule', ('name', 'match', 'get_new_command',
 | 
			
		||||
                           'enabled_by_default'))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class RulesNamesList(list):
 | 
			
		||||
    """Wrapper a top of list for string rules names."""
 | 
			
		||||
 | 
			
		||||
    def __contains__(self, item):
 | 
			
		||||
        return super(RulesNamesList, self).__contains__(item.name)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Settings(dict):
 | 
			
		||||
 | 
			
		||||
    def __getattr__(self, item):
 | 
			
		||||
        return self.get(item)
 | 
			
		||||
 | 
			
		||||
    def update(self, **kwargs):
 | 
			
		||||
        """Returns new settings with new values from `kwargs`."""
 | 
			
		||||
        conf = dict(self)
 | 
			
		||||
        conf.update(kwargs)
 | 
			
		||||
        return Settings(conf)
 | 
			
		||||
@@ -1,7 +1,7 @@
 | 
			
		||||
from functools import wraps
 | 
			
		||||
import os
 | 
			
		||||
import six
 | 
			
		||||
from thefuck.main import Command
 | 
			
		||||
from .types import Command
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def which(program):
 | 
			
		||||
@@ -37,10 +37,7 @@ def wrap_settings(params):
 | 
			
		||||
    def decorator(fn):
 | 
			
		||||
        @wraps(fn)
 | 
			
		||||
        def wrapper(command, settings):
 | 
			
		||||
            for key, val in params.items():
 | 
			
		||||
                if not hasattr(settings, key):
 | 
			
		||||
                    setattr(settings, key, val)
 | 
			
		||||
            return fn(command, settings)
 | 
			
		||||
            return fn(command, settings.update(**params))
 | 
			
		||||
        return wrapper
 | 
			
		||||
    return decorator
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user