mirror of
				https://github.com/nvbn/thefuck.git
				synced 2025-11-04 00:52:04 +00:00 
			
		
		
		
	Compare commits
	
		
			247 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						 | 
					4fb990742d | ||
| 
						 | 
					cf3dca6f51 | ||
| 
						 | 
					5187bada1b | ||
| 
						 | 
					0238569b71 | ||
| 
						 | 
					463b4fef2f | ||
| 
						 | 
					f90bac10ed | ||
| 
						 | 
					90014b2b05 | ||
| 
						 | 
					4276cacaf6 | ||
| 
						 | 
					b31aea3737 | ||
| 
						 | 
					fbfb4b5e41 | ||
| 
						 | 
					51c37bc5ab | ||
| 
						 | 
					5d0912fee8 | ||
| 
						 | 
					f6a4902074 | ||
| 
						 | 
					707d91200e | ||
| 
						 | 
					b3e09d68df | ||
| 
						 | 
					78769e4fbc | ||
| 
						 | 
					3e4c043ccc | ||
| 
						 | 
					934099fe9e | ||
| 
						 | 
					464f86eccf | ||
| 
						 | 
					891fbe7ed1 | ||
| 
						 | 
					5abab8bd1e | ||
| 
						 | 
					7ebc8a38af | ||
| 
						 | 
					f40b63f44b | ||
| 
						 | 
					4b4e7acc0f | ||
| 
						 | 
					a8587d3871 | ||
| 
						 | 
					370c58e679 | ||
| 
						 | 
					328e65179e | ||
| 
						 | 
					63bb4da8e1 | ||
| 
						 | 
					0b5a7a8e2d | ||
| 
						 | 
					5693bd49f7 | ||
| 
						 | 
					12f8d017b9 | ||
| 
						 | 
					c7071763a3 | ||
| 
						 | 
					27b5b9de6a | ||
| 
						 | 
					c0eae8b85c | ||
| 
						 | 
					fbf7b91005 | ||
| 
						 | 
					3d842fe6eb | ||
| 
						 | 
					17d359b43f | ||
| 
						 | 
					22503cdb94 | ||
| 
						 | 
					a8de919300 | ||
| 
						 | 
					fac18de242 | ||
| 
						 | 
					26fc18dfe4 | ||
| 
						 | 
					0fb5c9a228 | ||
| 
						 | 
					04a342bbc7 | ||
| 
						 | 
					669fbff6ce | ||
| 
						 | 
					e2542915e4 | ||
| 
						 | 
					1b08a7dcb6 | ||
| 
						 | 
					8d77a2d528 | ||
| 
						 | 
					18ea4272ab | ||
| 
						 | 
					a75c99eb12 | ||
| 
						 | 
					f3cdfbdbdb | ||
| 
						 | 
					30082bc9a9 | ||
| 
						 | 
					3822f62d90 | ||
| 
						 | 
					25cc98a21a | ||
| 
						 | 
					409d839e92 | ||
| 
						 | 
					51b5dd0460 | ||
| 
						 | 
					cb33c912e5 | ||
| 
						 | 
					470d66eeb2 | ||
| 
						 | 
					5552fd3dc9 | ||
| 
						 | 
					ab55c1cccb | ||
| 
						 | 
					7173e0dbad | ||
| 
						 | 
					369ea7ff46 | ||
| 
						 | 
					40fe604adc | ||
| 
						 | 
					ef504b6436 | ||
| 
						 | 
					b59e83cca9 | ||
| 
						 | 
					6d718a38dc | ||
| 
						 | 
					330f91f5dc | ||
| 
						 | 
					e3cc9c52e6 | ||
| 
						 | 
					01ce65047a | ||
| 
						 | 
					3203d57b36 | ||
| 
						 | 
					28ab6c62f8 | ||
| 
						 | 
					360e4673eb | ||
| 
						 | 
					c6aead735b | ||
| 
						 | 
					af3e1a555f | ||
| 
						 | 
					897b847975 | ||
| 
						 | 
					a0949b1102 | ||
| 
						 | 
					2f1460120e | ||
| 
						 | 
					43ec397190 | ||
| 
						 | 
					eb537bef81 | ||
| 
						 | 
					b033d3893b | ||
| 
						 | 
					633c4f8415 | ||
| 
						 | 
					5bfe0ac997 | ||
| 
						 | 
					e8a55220ad | ||
| 
						 | 
					ea306038f9 | ||
| 
						 | 
					6a88cc47b6 | ||
| 
						 | 
					96fe1e77b3 | ||
| 
						 | 
					c08d9125e4 | ||
| 
						 | 
					be682170e5 | ||
| 
						 | 
					5e981d9b01 | ||
| 
						 | 
					add499af7f | ||
| 
						 | 
					9c711734aa | ||
| 
						 | 
					ff7a433f39 | ||
| 
						 | 
					6bb7d79ddc | ||
| 
						 | 
					f6c013d033 | ||
| 
						 | 
					01cf199866 | ||
| 
						 | 
					3d41a3fb7c | ||
| 
						 | 
					f55fa35ebf | ||
| 
						 | 
					ce922758a4 | ||
| 
						 | 
					c47968a180 | ||
| 
						 | 
					581c97ec4b | ||
| 
						 | 
					0a53966f9b | ||
| 
						 | 
					ed4e7946d7 | ||
| 
						 | 
					2ed96b1d51 | ||
| 
						 | 
					79d94e2651 | ||
| 
						 | 
					c08182509d | ||
| 
						 | 
					1d2d907c60 | ||
| 
						 | 
					13996261be | ||
| 
						 | 
					afcd7fc67e | ||
| 
						 | 
					c0c7397057 | ||
| 
						 | 
					707743e7a7 | ||
| 
						 | 
					d8779dc4a6 | ||
| 
						 | 
					ba9214f7fc | ||
| 
						 | 
					660422806c | ||
| 
						 | 
					3c8978784b | ||
| 
						 | 
					995b373347 | ||
| 
						 | 
					dbe1a94c7d | ||
| 
						 | 
					15e13d7c1a | ||
| 
						 | 
					3194913965 | ||
| 
						 | 
					237f43ebdb | ||
| 
						 | 
					a5aadc6e90 | ||
| 
						 | 
					18ce062300 | ||
| 
						 | 
					73bc6c0184 | ||
| 
						 | 
					0296a4a46d | ||
| 
						 | 
					54a9769c10 | ||
| 
						 | 
					abc7238d14 | ||
| 
						 | 
					710a72ee8c | ||
| 
						 | 
					e09c6530e5 | ||
| 
						 | 
					b1da6a883a | ||
| 
						 | 
					a9e3b22fa4 | ||
| 
						 | 
					9debcdf676 | ||
| 
						 | 
					718cadb85a | ||
| 
						 | 
					910e6f4759 | ||
| 
						 | 
					d3146aa0ac | ||
| 
						 | 
					190e47ecdb | ||
| 
						 | 
					84a28d8c73 | ||
| 
						 | 
					551e35e3b6 | ||
| 
						 | 
					2bebfabf8d | ||
| 
						 | 
					675317b247 | ||
| 
						 | 
					6cf430cc23 | ||
| 
						 | 
					d088dac0f4 | ||
| 
						 | 
					c65fdd0f81 | ||
| 
						 | 
					e7d7b80c09 | ||
| 
						 | 
					f986df23d5 | ||
| 
						 | 
					cf5f18bf23 | ||
| 
						 | 
					44c06c483e | ||
| 
						 | 
					1f48d5e12a | ||
| 
						 | 
					2c3df1ad47 | ||
| 
						 | 
					5319871326 | ||
| 
						 | 
					d6ce5e1e62 | ||
| 
						 | 
					c42898dde3 | ||
| 
						 | 
					17b9104939 | ||
| 
						 | 
					02d9613618 | ||
| 
						 | 
					b63ce26853 | ||
| 
						 | 
					ce6855fd97 | ||
| 
						 | 
					051f5fcb89 | ||
| 
						 | 
					6590da623f | ||
| 
						 | 
					dc53f58b2a | ||
| 
						 | 
					961d4d5293 | ||
| 
						 | 
					1ffc9624ed | ||
| 
						 | 
					afcee5844b | ||
| 
						 | 
					881967f4c5 | ||
| 
						 | 
					3c673e0972 | ||
| 
						 | 
					8fdcff776a | ||
| 
						 | 
					1b5c935f30 | ||
| 
						 | 
					8d256390a1 | ||
| 
						 | 
					51800afca8 | ||
| 
						 | 
					07831666db | ||
| 
						 | 
					252859e63a | ||
| 
						 | 
					a54c97f624 | ||
| 
						 | 
					9ef346468c | ||
| 
						 | 
					f04c4396eb | ||
| 
						 | 
					9ade21bf0a | ||
| 
						 | 
					179839c32f | ||
| 
						 | 
					3d0d4be4a9 | ||
| 
						 | 
					d854320acc | ||
| 
						 | 
					bb4b42d2f1 | ||
| 
						 | 
					6539c853b4 | ||
| 
						 | 
					5f2b2433b1 | ||
| 
						 | 
					d41b1d48d2 | ||
| 
						 | 
					bbdac1884a | ||
| 
						 | 
					d5bd57fb49 | ||
| 
						 | 
					fc8f1b1136 | ||
| 
						 | 
					d7c67ad09d | ||
| 
						 | 
					73939836d4 | ||
| 
						 | 
					744f17d905 | ||
| 
						 | 
					08a2065119 | ||
| 
						 | 
					5504aa44a1 | ||
| 
						 | 
					3c4f9d50a9 | ||
| 
						 | 
					371a4b0ad3 | ||
| 
						 | 
					9cf41f8e43 | ||
| 
						 | 
					d2e511fa2c | ||
| 
						 | 
					a1437db40a | ||
| 
						 | 
					239f91b670 | ||
| 
						 | 
					7b29b54ac7 | ||
| 
						 | 
					a83d75991b | ||
| 
						 | 
					14d14c5ac6 | ||
| 
						 | 
					65c624ad52 | ||
| 
						 | 
					a77db59da5 | ||
| 
						 | 
					8ac4dafe6d | ||
| 
						 | 
					779e29906e | ||
| 
						 | 
					e8de4ee7e8 | ||
| 
						 | 
					47a1faa881 | ||
| 
						 | 
					ab97b94faf | ||
| 
						 | 
					7489040f8f | ||
| 
						 | 
					484a53e314 | ||
| 
						 | 
					0fc7c00e8d | ||
| 
						 | 
					64318c09b7 | ||
| 
						 | 
					7e55041963 | ||
| 
						 | 
					5b6e17b5f1 | ||
| 
						 | 
					6cdc2c27fb | ||
| 
						 | 
					62c605d0ac | ||
| 
						 | 
					8930d01601 | ||
| 
						 | 
					c749615ad6 | ||
| 
						 | 
					f03d8c54b1 | ||
| 
						 | 
					20f1c76d27 | ||
| 
						 | 
					f477cd69c2 | ||
| 
						 | 
					690729d5a1 | ||
| 
						 | 
					f082ba829f | ||
| 
						 | 
					112e20d7c5 | ||
| 
						 | 
					95007220fb | ||
| 
						 | 
					56f636f3d8 | ||
| 
						 | 
					932a7c5db5 | ||
| 
						 | 
					1bed4d4e8d | ||
| 
						 | 
					e0bba379ff | ||
| 
						 | 
					045959ec47 | ||
| 
						 | 
					65aeea857e | ||
| 
						 | 
					793e883073 | ||
| 
						 | 
					a395ac568c | ||
| 
						 | 
					29e70e14a0 | ||
| 
						 | 
					0cdd23edcf | ||
| 
						 | 
					36d80859a4 | ||
| 
						 | 
					2b12b4bfce | ||
| 
						 | 
					91c1fe414a | ||
| 
						 | 
					f3d377114e | ||
| 
						 | 
					05f594b918 | ||
| 
						 | 
					5bf1424613 | ||
| 
						 | 
					fc3fcf028a | ||
| 
						 | 
					5864faadef | ||
| 
						 | 
					608d48e408 | ||
| 
						 | 
					9380eb1f56 | ||
| 
						 | 
					fb069b74d7 | ||
| 
						 | 
					6624ecb3b8 | ||
| 
						 | 
					a8ff2375c0 | ||
| 
						 | 
					80bfbec422 | ||
| 
						 | 
					3f2fe0d275 | ||
| 
						 | 
					fc364b99b9 | ||
| 
						 | 
					742f6f9c94 | ||
| 
						 | 
					cd1bee9cb0 | 
@@ -4,6 +4,10 @@ python:
 | 
				
			|||||||
  - "3.3"
 | 
					  - "3.3"
 | 
				
			||||||
  - "2.7"
 | 
					  - "2.7"
 | 
				
			||||||
install:
 | 
					install:
 | 
				
			||||||
  - python setup.py develop
 | 
					 | 
				
			||||||
  - pip install -r requirements.txt
 | 
					  - pip install -r requirements.txt
 | 
				
			||||||
script: py.test
 | 
					  - python setup.py develop
 | 
				
			||||||
 | 
					  - pip install coveralls
 | 
				
			||||||
 | 
					script:
 | 
				
			||||||
 | 
					  - export COVERAGE_PYTHON_VERSION=python-${TRAVIS_PYTHON_VERSION:0:1}
 | 
				
			||||||
 | 
					  - coverage run --source=thefuck,tests -m py.test -v
 | 
				
			||||||
 | 
					after_success: coveralls
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										96
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										96
									
								
								README.md
									
									
									
									
									
								
							@@ -1,10 +1,14 @@
 | 
				
			|||||||
# The Fuck [](https://travis-ci.org/nvbn/thefuck)
 | 
					# The Fuck [](https://travis-ci.org/nvbn/thefuck) [](https://coveralls.io/github/nvbn/thefuck?branch=master)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					**Aliases changed in 1.34.**
 | 
				
			||||||
 | 
					
 | 
				
			||||||
Magnificent app which corrects your previous console command,
 | 
					Magnificent app which corrects your previous console command,
 | 
				
			||||||
inspired by a [@liamosaur](https://twitter.com/liamosaur/)
 | 
					inspired by a [@liamosaur](https://twitter.com/liamosaur/)
 | 
				
			||||||
[tweet](https://twitter.com/liamosaur/status/506975850596536320).
 | 
					[tweet](https://twitter.com/liamosaur/status/506975850596536320).
 | 
				
			||||||
 | 
					
 | 
				
			||||||
Few examples:
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Few more examples:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
```bash
 | 
					```bash
 | 
				
			||||||
➜ apt-get install vim
 | 
					➜ apt-get install vim
 | 
				
			||||||
@@ -71,7 +75,7 @@ REPL-y 0.3.1
 | 
				
			|||||||
...
 | 
					...
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
If you are scared to blindly run changed command, there's `require_confirmation`
 | 
					If you are scared to blindly run the changed command, there is a `require_confirmation`
 | 
				
			||||||
[settings](#settings) option:
 | 
					[settings](#settings) option:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
```bash
 | 
					```bash
 | 
				
			||||||
@@ -102,14 +106,25 @@ sudo pip install thefuck
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
[Or using an OS package manager (OS X, Ubuntu, Arch).](https://github.com/nvbn/thefuck/wiki/Installation)
 | 
					[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):
 | 
					And add to the `.bashrc` or `.bash_profile`(for OSX):
 | 
				
			||||||
 | 
					
 | 
				
			||||||
```bash
 | 
					```bash
 | 
				
			||||||
alias fuck='eval $(thefuck $(fc -ln -1))'
 | 
					alias fuck='eval $(thefuck $(fc -ln -1)); history -r'
 | 
				
			||||||
# You can use whatever you want as an alias, like for Mondays:
 | 
					# You can use whatever you want as an alias, like for Mondays:
 | 
				
			||||||
alias FUCK='fuck'
 | 
					alias FUCK='fuck'
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Or in your `.zshrc`:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					```bash
 | 
				
			||||||
 | 
					alias fuck='eval $(thefuck $(fc -ln -1 | tail -n 1)); fc -R'
 | 
				
			||||||
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					If you are using `tcsh`:
 | 
				
			||||||
 | 
					```tcsh
 | 
				
			||||||
 | 
					alias fuck 'set fucked_cmd=`history -h 2 | head -n 1` && eval `thefuck ${fucked_cmd}`'
 | 
				
			||||||
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
Alternatively, you can redirect the output of `thefuck-alias`:
 | 
					Alternatively, you can redirect the output of `thefuck-alias`:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
```bash
 | 
					```bash
 | 
				
			||||||
@@ -119,6 +134,7 @@ thefuck-alias >> ~/.bashrc
 | 
				
			|||||||
[Or in your shell config (Bash, Zsh, Fish, Powershell).](https://github.com/nvbn/thefuck/wiki/Shell-aliases)
 | 
					[Or 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.
 | 
					Changes will be available only in a new shell session.
 | 
				
			||||||
 | 
					To make them available immediately, run `source ~/.bashrc` (or your shell config file like `.zshrc`).
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
## Update
 | 
					## Update
 | 
				
			||||||
@@ -129,34 +145,70 @@ sudo pip install thefuck --upgrade
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
## How it works
 | 
					## How it works
 | 
				
			||||||
 | 
					
 | 
				
			||||||
The Fuck tries to match rule for the previous command, create new command
 | 
					The Fuck tries to match a rule for the previous command, creates a new command
 | 
				
			||||||
using matched rule and run it. Rules enabled by default:
 | 
					using the matched rule and runs it. Rules enabled by default are as follows:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
* `brew_unknown_command` – fixes wrong brew commands, for example `brew docto/brew doctor`;
 | 
					* `cargo` – runs `cargo build` instead of `cargo`;
 | 
				
			||||||
* `cd_parent` – changes `cd..` to `cd ..`;
 | 
					* `cargo_no_command` – fixes wrongs commands like `cargo buid`;
 | 
				
			||||||
 | 
					* `cd_correction` – spellchecks and correct failed cd commands;
 | 
				
			||||||
* `cd_mkdir` – creates directories before cd'ing into them;
 | 
					* `cd_mkdir` – creates directories before cd'ing into them;
 | 
				
			||||||
 | 
					* `cd_parent` – changes `cd..` to `cd ..`;
 | 
				
			||||||
 | 
					* `composer_not_command` – fixes composer command name;
 | 
				
			||||||
* `cp_omitting_directory` – adds `-a` when you `cp` directory;
 | 
					* `cp_omitting_directory` – adds `-a` when you `cp` directory;
 | 
				
			||||||
 | 
					* `cpp11` – add missing `-std=c++11` to `g++` or `clang++`;
 | 
				
			||||||
 | 
					* `django_south_ghost` – adds `--delete-ghost-migrations` to failed because ghosts django south migration;
 | 
				
			||||||
 | 
					* `django_south_merge` – adds `--merge` to inconsistent django south migration;
 | 
				
			||||||
 | 
					* `dry` – fix repetitions like "git git push";
 | 
				
			||||||
* `fix_alt_space` – replaces Alt+Space with Space character;
 | 
					* `fix_alt_space` – replaces Alt+Space with Space character;
 | 
				
			||||||
 | 
					* `git_add` – fix *"Did you forget to 'git add'?"*;
 | 
				
			||||||
 | 
					* `git_branch_delete` – changes `git branch -d` to `git branch -D`;
 | 
				
			||||||
 | 
					* `git_branch_list` – catches `git branch list` in place of `git branch` and removes created branch;
 | 
				
			||||||
 | 
					* `git_checkout` – creates the branch before checking-out;
 | 
				
			||||||
 | 
					* `git_diff_staged` – adds `--staged` to previous `git diff` with unexpected output;
 | 
				
			||||||
* `git_no_command` – fixes wrong git commands like `git brnch`;
 | 
					* `git_no_command` – fixes wrong git commands like `git brnch`;
 | 
				
			||||||
 | 
					* `git_pull` – sets upstream before executing previous `git pull`;
 | 
				
			||||||
* `git_push` – adds `--set-upstream origin $branch` to previous failed `git push`;
 | 
					* `git_push` – adds `--set-upstream origin $branch` to previous failed `git push`;
 | 
				
			||||||
 | 
					* `git_stash` – stashes you local modifications before rebasing or switching branch;
 | 
				
			||||||
 | 
					* `go_run` – appends `.go` extension when compiling/running Go programs
 | 
				
			||||||
 | 
					* `grep_recursive` – adds `-r` when you trying to grep directory;
 | 
				
			||||||
* `has_exists_script` – prepends `./` when script/binary exists;
 | 
					* `has_exists_script` – prepends `./` when script/binary exists;
 | 
				
			||||||
 | 
					* `history` – tries to replace command with most similar command from history;
 | 
				
			||||||
 | 
					* `java` – removes `.java` extension when running Java programs;
 | 
				
			||||||
 | 
					* `javac` – appends missing `.java` when compiling Java files;
 | 
				
			||||||
* `lein_not_task` – fixes wrong `lein` tasks like `lein rpl`;
 | 
					* `lein_not_task` – fixes wrong `lein` tasks like `lein rpl`;
 | 
				
			||||||
 | 
					* `ls_lah` – adds -lah to ls;
 | 
				
			||||||
 | 
					* `man` – change manual section;
 | 
				
			||||||
 | 
					* `man_no_space` – fixes man commands without spaces, for example `mandiff`;
 | 
				
			||||||
 | 
					* `mercurial` – fixes wrong `hg` commands;
 | 
				
			||||||
* `mkdir_p` – adds `-p` when you trying to create directory without parent;
 | 
					* `mkdir_p` – adds `-p` when you trying to create directory without parent;
 | 
				
			||||||
* `no_command` – fixes wrong console commands, for example `vom/vim`;
 | 
					* `no_command` – fixes wrong console commands, for example `vom/vim`;
 | 
				
			||||||
 | 
					* `no_such_file` – creates missing directories with `mv` and `cp` commands;
 | 
				
			||||||
 | 
					* `open` – prepends `http` to address passed to `open`;
 | 
				
			||||||
* `pip_unknown_command` – fixes wrong pip commands, for example `pip instatl/pip install`;
 | 
					* `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;
 | 
					* `python_command` – prepends `python` when you trying to run not executable/without `./` python script;
 | 
				
			||||||
* `sl_ls` – changes `sl` to `ls`;
 | 
					* `python_execute` – appends missing `.py` when executing Python files;
 | 
				
			||||||
 | 
					* `quotation_marks` – fixes uneven usage of `'` and `"` when containing args'
 | 
				
			||||||
* `rm_dir` – adds `-rf` when you trying to remove directory;
 | 
					* `rm_dir` – adds `-rf` when you trying to remove directory;
 | 
				
			||||||
 | 
					* `sed_unterminated_s` – adds missing '/' to `sed`'s `s` commands;
 | 
				
			||||||
 | 
					* `sl_ls` – changes `sl` to `ls`;
 | 
				
			||||||
* `ssh_known_hosts` – removes host from `known_hosts` on warning;
 | 
					* `ssh_known_hosts` – removes host from `known_hosts` on warning;
 | 
				
			||||||
* `sudo` – prepends `sudo` to previous command if it failed because of permissions;
 | 
					* `sudo` – prepends `sudo` to previous command if it failed because of permissions;
 | 
				
			||||||
* `switch_layout` – switches command from your local layout to en;
 | 
					* `switch_layout` – switches command from your local layout to en;
 | 
				
			||||||
 | 
					* `systemctl` – correctly orders parameters of confusing systemctl;
 | 
				
			||||||
 | 
					* `test.py` – runs `py.test` instead of `test.py`;
 | 
				
			||||||
 | 
					* `tmux` – fixes tmux commands;
 | 
				
			||||||
 | 
					* `whois` – fixes `whois` command.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Enabled by default only on specific platforms:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
* `apt_get` – installs app from apt if it not installed;
 | 
					* `apt_get` – installs app from apt if it not installed;
 | 
				
			||||||
* `brew_install` – fixes formula name for `brew install`;
 | 
					* `brew_install` – fixes formula name for `brew install`;
 | 
				
			||||||
* `composer_not_command` – fixes composer command name.
 | 
					* `brew_unknown_command` – fixes wrong brew commands, for example `brew docto/brew doctor`;
 | 
				
			||||||
 | 
					* `brew_upgrade` – appends `--all` to `brew upgrade` as per Homebrew's new behaviour
 | 
				
			||||||
 | 
					* `pacman` – installs app with `pacman` or `yaourt` if it is not installed.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
Bundled, but not enabled by default:
 | 
					Bundled, but not enabled by default:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
* `ls_lah` – adds -lah to ls;
 | 
					 | 
				
			||||||
* `rm_root` – adds `--no-preserve-root` to `rm -rf /` command.
 | 
					* `rm_root` – adds `--no-preserve-root` to `rm -rf /` command.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
## Creating your own rules
 | 
					## Creating your own rules
 | 
				
			||||||
@@ -167,7 +219,7 @@ in `~/.thefuck/rules`. Rule should contain two functions:
 | 
				
			|||||||
and `get_new_command(command: Command, settings: Settings) -> str`.
 | 
					and `get_new_command(command: Command, settings: Settings) -> str`.
 | 
				
			||||||
Also the rule can contain optional function
 | 
					Also the rule can contain optional function
 | 
				
			||||||
`side_effect(command: Command, settings: Settings) -> None` and
 | 
					`side_effect(command: Command, settings: Settings) -> None` and
 | 
				
			||||||
optional boolean `enabled_by_default`
 | 
					optional boolean `enabled_by_default`.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
`Command` has three attributes: `script`, `stdout` and `stderr`.
 | 
					`Command` has three attributes: `script`, `stdout` and `stderr`.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -189,6 +241,8 @@ enabled_by_default = True
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
def side_effect(command, settings):
 | 
					def side_effect(command, settings):
 | 
				
			||||||
    subprocess.call('chmod 777 .', shell=True)
 | 
					    subprocess.call('chmod 777 .', shell=True)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					priority = 1000  # Lower first
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
[More examples of rules](https://github.com/nvbn/thefuck/tree/master/thefuck/rules),
 | 
					[More examples of rules](https://github.com/nvbn/thefuck/tree/master/thefuck/rules),
 | 
				
			||||||
@@ -196,12 +250,14 @@ def side_effect(command, settings):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
## Settings
 | 
					## Settings
 | 
				
			||||||
 | 
					
 | 
				
			||||||
The Fuck has a few settings parameters, they can be changed in `~/.thefuck/settings.py`:
 | 
					The Fuck has a few settings parameters which can be changed in `~/.thefuck/settings.py`:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
* `rules` – list of enabled rules, by default `thefuck.conf.DEFAULT_RULES`;
 | 
					* `rules` – list of enabled rules, by default `thefuck.conf.DEFAULT_RULES`;
 | 
				
			||||||
* `require_confirmation` – require confirmation before running new command, by default `False`;
 | 
					* `require_confirmation` – requires confirmation before running new command, by default `True`;
 | 
				
			||||||
* `wait_command` – max amount of time in seconds for getting previous command output;
 | 
					* `wait_command` – max amount of time in seconds for getting previous command output;
 | 
				
			||||||
* `no_colors` – disable colored output.
 | 
					* `no_colors` – disable colored output;
 | 
				
			||||||
 | 
					* `priority` – dict with rules priorities, rule with lower `priority` will be matched first;
 | 
				
			||||||
 | 
					* `debug` – enabled debug output, by default `False`;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
Example of `settings.py`:
 | 
					Example of `settings.py`:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -210,6 +266,8 @@ rules = ['sudo', 'no_command']
 | 
				
			|||||||
require_confirmation = True
 | 
					require_confirmation = True
 | 
				
			||||||
wait_command = 10
 | 
					wait_command = 10
 | 
				
			||||||
no_colors = False
 | 
					no_colors = False
 | 
				
			||||||
 | 
					priority = {'sudo': 100, 'no_command': 9999}
 | 
				
			||||||
 | 
					debug = False
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
Or via environment variables:
 | 
					Or via environment variables:
 | 
				
			||||||
@@ -217,7 +275,10 @@ Or via environment variables:
 | 
				
			|||||||
* `THEFUCK_RULES` – list of enabled rules, like `DEFAULT_RULES:rm_root` or `sudo:no_command`;
 | 
					* `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_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_WAIT_COMMAND` – max amount of time in seconds for getting previous command output;
 | 
				
			||||||
* `THEFUCK_NO_COLORS` – disable colored output, `true/false`.
 | 
					* `THEFUCK_NO_COLORS` – disable colored output, `true/false`;
 | 
				
			||||||
 | 
					* `THEFUCK_PRIORITY` – priority of the rules, like `no_command=9999:apt_get=100`,
 | 
				
			||||||
 | 
					rule with lower `priority` will be matched first;
 | 
				
			||||||
 | 
					* `THEFUCK_DEBUG` – enables debug output, `true/false`.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
For example:
 | 
					For example:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -226,6 +287,7 @@ export THEFUCK_RULES='sudo:no_command'
 | 
				
			|||||||
export THEFUCK_REQUIRE_CONFIRMATION='true'
 | 
					export THEFUCK_REQUIRE_CONFIRMATION='true'
 | 
				
			||||||
export THEFUCK_WAIT_COMMAND=10
 | 
					export THEFUCK_WAIT_COMMAND=10
 | 
				
			||||||
export THEFUCK_NO_COLORS='false'
 | 
					export THEFUCK_NO_COLORS='false'
 | 
				
			||||||
 | 
					export THEFUCK_PRIORITY='no_command=9999:apt_get=100'
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
## Developing
 | 
					## Developing
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										
											BIN
										
									
								
								example.gif
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										
											BIN
										
									
								
								example.gif
									
									
									
									
									
										Normal file
									
								
							
										
											Binary file not shown.
										
									
								
							| 
		 After Width: | Height: | Size: 704 KiB  | 
@@ -28,4 +28,4 @@ call('git commit -am "Bump to {}"'.format(version), shell=True)
 | 
				
			|||||||
call('git tag {}'.format(version), shell=True)
 | 
					call('git tag {}'.format(version), shell=True)
 | 
				
			||||||
call('git push', shell=True)
 | 
					call('git push', shell=True)
 | 
				
			||||||
call('git push --tags', shell=True)
 | 
					call('git push --tags', shell=True)
 | 
				
			||||||
call('python setup.py sdist upload', shell=True)
 | 
					call('python setup.py sdist bdist_wheel upload', shell=True)
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,2 +1,5 @@
 | 
				
			|||||||
pytest
 | 
					pytest
 | 
				
			||||||
mock
 | 
					mock
 | 
				
			||||||
 | 
					pytest-mock
 | 
				
			||||||
 | 
					wheel
 | 
				
			||||||
 | 
					setuptools>=17.1
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										20
									
								
								setup.py
									
									
									
									
									
										
										
										Normal file → Executable file
									
								
							
							
						
						
									
										20
									
								
								setup.py
									
									
									
									
									
										
										
										Normal file → Executable file
									
								
							@@ -1,8 +1,20 @@
 | 
				
			|||||||
 | 
					#!/usr/bin/env python
 | 
				
			||||||
from setuptools import setup, find_packages
 | 
					from setuptools import setup, find_packages
 | 
				
			||||||
 | 
					import sys
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					if sys.version_info < (2, 7):
 | 
				
			||||||
 | 
					    print('thefuck requires Python version 2.7 or later' +
 | 
				
			||||||
 | 
					          ' ({}.{} detected).'.format(*sys.version_info[:2]))
 | 
				
			||||||
 | 
					    sys.exit(-1)
 | 
				
			||||||
 | 
					elif (3, 0) < sys.version_info < (3, 3):
 | 
				
			||||||
 | 
					    print('thefuck requires Python version 3.3 or later' +
 | 
				
			||||||
 | 
					          ' ({}.{} detected).'.format(*sys.version_info[:2]))
 | 
				
			||||||
 | 
					    sys.exit(-1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
VERSION = '1.34'
 | 
					VERSION = '2.0'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					install_requires = ['psutil', 'colorama', 'six']
 | 
				
			||||||
 | 
					extras_require = {':python_version<"3.4"': ['pathlib']}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
setup(name='thefuck',
 | 
					setup(name='thefuck',
 | 
				
			||||||
      version=VERSION,
 | 
					      version=VERSION,
 | 
				
			||||||
@@ -15,6 +27,8 @@ setup(name='thefuck',
 | 
				
			|||||||
                                      'tests', 'release']),
 | 
					                                      'tests', 'release']),
 | 
				
			||||||
      include_package_data=True,
 | 
					      include_package_data=True,
 | 
				
			||||||
      zip_safe=False,
 | 
					      zip_safe=False,
 | 
				
			||||||
      install_requires=['pathlib', 'psutil', 'colorama', 'six'],
 | 
					      install_requires=install_requires,
 | 
				
			||||||
 | 
					      extras_require=extras_require,
 | 
				
			||||||
      entry_points={'console_scripts': [
 | 
					      entry_points={'console_scripts': [
 | 
				
			||||||
          'thefuck = thefuck.main:main', 'thefuck-alias = thefuck.main:alias']})
 | 
					          'thefuck = thefuck.main:main',
 | 
				
			||||||
 | 
					          'thefuck-alias = thefuck.shells:app_alias']})
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										6
									
								
								tests/conftest.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								tests/conftest.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,6 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def no_memoize(monkeypatch):
 | 
				
			||||||
 | 
					    monkeypatch.setattr('thefuck.utils.memoize.disabled', True)
 | 
				
			||||||
							
								
								
									
										0
									
								
								tests/rules/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										0
									
								
								tests/rules/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
								
								
									
										6
									
								
								tests/rules/conftest.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								tests/rules/conftest.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,6 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture(autouse=True)
 | 
				
			||||||
 | 
					def generic_shell(monkeypatch):
 | 
				
			||||||
 | 
					    monkeypatch.setattr('thefuck.shells.and_', lambda *x: ' && '.join(x))
 | 
				
			||||||
							
								
								
									
										59
									
								
								tests/rules/test_apt_get.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										59
									
								
								tests/rules/test_apt_get.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,59 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from mock import Mock, patch
 | 
				
			||||||
 | 
					from thefuck.rules import apt_get
 | 
				
			||||||
 | 
					from thefuck.rules.apt_get import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# python-commandnotfound is available in ubuntu 14.04+
 | 
				
			||||||
 | 
					@pytest.mark.skipif(not getattr(apt_get, 'enabled_by_default', True),
 | 
				
			||||||
 | 
					                    reason='Skip if python-commandnotfound is not available')
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='vim', stderr='vim: command not found')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, return_value', [
 | 
				
			||||||
 | 
					    (Command(script='vim', stderr='vim: command not found'),
 | 
				
			||||||
 | 
					     [('vim', 'main'), ('vim-tiny', 'main')])])
 | 
				
			||||||
 | 
					@patch('thefuck.rules.apt_get.CommandNotFound', create=True)
 | 
				
			||||||
 | 
					@patch.multiple(apt_get, create=True, apt_get='apt_get')
 | 
				
			||||||
 | 
					def test_match_mocked(cmdnf_mock, command, return_value):
 | 
				
			||||||
 | 
					    get_packages = Mock(return_value=return_value)
 | 
				
			||||||
 | 
					    cmdnf_mock.CommandNotFound.return_value = Mock(getPackages=get_packages)
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					    assert cmdnf_mock.CommandNotFound.called
 | 
				
			||||||
 | 
					    assert get_packages.called
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='vim', stderr=''), Command()])
 | 
				
			||||||
 | 
					def test_not_match(command):
 | 
				
			||||||
 | 
					    assert not match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# python-commandnotfound is available in ubuntu 14.04+
 | 
				
			||||||
 | 
					@pytest.mark.skipif(not getattr(apt_get, 'enabled_by_default', True),
 | 
				
			||||||
 | 
					                    reason='Skip if python-commandnotfound is not available')
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('vim'), 'sudo apt-get install vim && vim'),
 | 
				
			||||||
 | 
					    (Command('convert'), 'sudo apt-get install imagemagick && convert')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command, return_value', [
 | 
				
			||||||
 | 
					    (Command('vim'), 'sudo apt-get install vim && vim',
 | 
				
			||||||
 | 
					     [('vim', 'main'), ('vim-tiny', 'main')]),
 | 
				
			||||||
 | 
					    (Command('convert'), 'sudo apt-get install imagemagick && convert',
 | 
				
			||||||
 | 
					     [('imagemagick', 'main'),
 | 
				
			||||||
 | 
					      ('graphicsmagick-imagemagick-compat', 'universe')])])
 | 
				
			||||||
 | 
					@patch('thefuck.rules.apt_get.CommandNotFound', create=True)
 | 
				
			||||||
 | 
					@patch.multiple(apt_get, create=True, apt_get='apt_get')
 | 
				
			||||||
 | 
					def test_get_new_command_mocked(cmdnf_mock, command, new_command, return_value):
 | 
				
			||||||
 | 
					    get_packages = Mock(return_value=return_value)
 | 
				
			||||||
 | 
					    cmdnf_mock.CommandNotFound.return_value = Mock(getPackages=get_packages)
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
 | 
					    assert cmdnf_mock.CommandNotFound.called
 | 
				
			||||||
 | 
					    assert get_packages.called
 | 
				
			||||||
							
								
								
									
										15
									
								
								tests/rules/test_brew_upgrade.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								tests/rules/test_brew_upgrade.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,15 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.brew_upgrade import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='brew upgrade')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('brew upgrade'), 'brew upgrade --all')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										21
									
								
								tests/rules/test_cargo_no_command.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								tests/rules/test_cargo_no_command.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.cargo_no_command import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					no_such_subcommand = """No such subcommand
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        Did you mean `build`?
 | 
				
			||||||
 | 
					"""
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='cargo buid', stderr=no_such_subcommand)])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('cargo buid', stderr=no_such_subcommand), 'cargo build')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
@@ -1,14 +1,22 @@
 | 
				
			|||||||
from mock import Mock
 | 
					import pytest
 | 
				
			||||||
from thefuck.rules.cp_omitting_directory import match, get_new_command
 | 
					from thefuck.rules.cp_omitting_directory import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def test_match():
 | 
					@pytest.mark.parametrize('script, stderr', [
 | 
				
			||||||
    assert match(Mock(script='cp dir', stderr="cp: omitting directory 'dir'"),
 | 
					    ('cp dir', 'cp: dor: is a directory'),
 | 
				
			||||||
                 None)
 | 
					    ('cp dir', "cp: omitting directory 'dir'")])
 | 
				
			||||||
    assert not match(Mock(script='some dir',
 | 
					def test_match(script, stderr):
 | 
				
			||||||
                          stderr="cp: omitting directory 'dir'"), None)
 | 
					    assert match(Command(script, stderr=stderr), None)
 | 
				
			||||||
    assert not match(Mock(script='cp dir', stderr=""), None)
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('script, stderr', [
 | 
				
			||||||
 | 
					    ('some dir', 'cp: dor: is a directory'),
 | 
				
			||||||
 | 
					    ('some dir', "cp: omitting directory 'dir'"),
 | 
				
			||||||
 | 
					    ('cp dir', '')])
 | 
				
			||||||
 | 
					def test_not_match(script, stderr):
 | 
				
			||||||
 | 
					    assert not match(Command(script, stderr=stderr), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def test_get_new_command():
 | 
					def test_get_new_command():
 | 
				
			||||||
    assert get_new_command(Mock(script='cp dir'), None) == 'cp -a dir'
 | 
					    assert get_new_command(Command(script='cp dir'), None) == 'cp -a dir'
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										53
									
								
								tests/rules/test_django_south_ghost.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								tests/rules/test_django_south_ghost.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,53 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.django_south_ghost import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def stderr():
 | 
				
			||||||
 | 
					    return '''Traceback (most recent call last):
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../bin/python", line 42, in <module>
 | 
				
			||||||
 | 
					    exec(compile(__file__f.read(), __file__, "exec"))
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../app/manage.py", line 34, in <module>
 | 
				
			||||||
 | 
					    execute_from_command_line(sys.argv)
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../lib/django/core/management/__init__.py", line 443, in execute_from_command_line
 | 
				
			||||||
 | 
					    utility.execute()
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../lib/django/core/management/__init__.py", line 382, in execute
 | 
				
			||||||
 | 
					    self.fetch_command(subcommand).run_from_argv(self.argv)
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../lib/django/core/management/base.py", line 196, in run_from_argv
 | 
				
			||||||
 | 
					    self.execute(*args, **options.__dict__)
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../lib/django/core/management/base.py", line 232, in execute
 | 
				
			||||||
 | 
					    output = self.handle(*args, **options)
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../app/lib/south/management/commands/migrate.py", line 108, in handle
 | 
				
			||||||
 | 
					    ignore_ghosts = ignore_ghosts,
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../app/lib/south/migration/__init__.py", line 193, in migrate_app
 | 
				
			||||||
 | 
					    applied_all = check_migration_histories(applied_all, delete_ghosts, ignore_ghosts)
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../app/lib/south/migration/__init__.py", line 88, in check_migration_histories
 | 
				
			||||||
 | 
					    raise exceptions.GhostMigrations(ghosts)
 | 
				
			||||||
 | 
					south.exceptions.GhostMigrations: 
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					 ! These migrations are in the database but not on disk:
 | 
				
			||||||
 | 
					    <app1: 0033_auto__...>
 | 
				
			||||||
 | 
					    <app1: 0034_fill_...>
 | 
				
			||||||
 | 
					    <app1: 0035_rename_...>
 | 
				
			||||||
 | 
					    <app2: 0003_add_...>
 | 
				
			||||||
 | 
					    <app2: 0004_denormalize_...>
 | 
				
			||||||
 | 
					    <app1: 0033_auto....>
 | 
				
			||||||
 | 
					    <app1: 0034_fill...>
 | 
				
			||||||
 | 
					 ! I'm not trusting myself; either fix this yourself by fiddling
 | 
				
			||||||
 | 
					 ! with the south_migrationhistory table, or pass --delete-ghost-migrations
 | 
				
			||||||
 | 
					 ! to South to have it delete ALL of these records (this may not be good).
 | 
				
			||||||
 | 
					'''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match(stderr):
 | 
				
			||||||
 | 
					    assert match(Command('./manage.py migrate', stderr=stderr), None)
 | 
				
			||||||
 | 
					    assert match(Command('python manage.py migrate', stderr=stderr), None)
 | 
				
			||||||
 | 
					    assert not match(Command('./manage.py migrate'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('app migrate', stderr=stderr), None)
 | 
				
			||||||
 | 
					    assert not match(Command('./manage.py test', stderr=stderr), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command():
 | 
				
			||||||
 | 
					    assert get_new_command(Command('./manage.py migrate auth'), None)\
 | 
				
			||||||
 | 
					        == './manage.py migrate auth --delete-ghost-migrations'
 | 
				
			||||||
							
								
								
									
										43
									
								
								tests/rules/test_django_south_merge.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										43
									
								
								tests/rules/test_django_south_merge.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,43 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.django_south_merge import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def stderr():
 | 
				
			||||||
 | 
					    return '''Running migrations for app:
 | 
				
			||||||
 | 
					 ! Migration app:0003_auto... should not have been applied before app:0002_auto__add_field_query_due_date_ but was.
 | 
				
			||||||
 | 
					Traceback (most recent call last):
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../bin/python", line 42, in <module>
 | 
				
			||||||
 | 
					    exec(compile(__file__f.read(), __file__, "exec"))
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../app/manage.py", line 34, in <module>
 | 
				
			||||||
 | 
					    execute_from_command_line(sys.argv)
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../lib/django/core/management/__init__.py", line 443, in execute_from_command_line
 | 
				
			||||||
 | 
					    utility.execute()
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../lib/django/core/management/__init__.py", line 382, in execute
 | 
				
			||||||
 | 
					    self.fetch_command(subcommand).run_from_argv(self.argv)
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../lib/django/core/management/base.py", line 196, in run_from_argv
 | 
				
			||||||
 | 
					    self.execute(*args, **options.__dict__)
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../lib/django/core/management/base.py", line 232, in execute
 | 
				
			||||||
 | 
					    output = self.handle(*args, **options)
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../app/lib/south/management/commands/migrate.py", line 108, in handle
 | 
				
			||||||
 | 
					    ignore_ghosts = ignore_ghosts,
 | 
				
			||||||
 | 
					  File "/home/nvbn/work/.../app/lib/south/migration/__init__.py", line 207, in migrate_app
 | 
				
			||||||
 | 
					    raise exceptions.InconsistentMigrationHistory(problems)
 | 
				
			||||||
 | 
					south.exceptions.InconsistentMigrationHistory: Inconsistent migration history
 | 
				
			||||||
 | 
					The following options are available:
 | 
				
			||||||
 | 
					    --merge: will just attempt the migration ignoring any potential dependency conflicts.
 | 
				
			||||||
 | 
					'''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match(stderr):
 | 
				
			||||||
 | 
					    assert match(Command('./manage.py migrate', stderr=stderr), None)
 | 
				
			||||||
 | 
					    assert match(Command('python manage.py migrate', stderr=stderr), None)
 | 
				
			||||||
 | 
					    assert not match(Command('./manage.py migrate'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('app migrate', stderr=stderr), None)
 | 
				
			||||||
 | 
					    assert not match(Command('./manage.py test', stderr=stderr), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command():
 | 
				
			||||||
 | 
					    assert get_new_command(Command('./manage.py migrate auth'), None) \
 | 
				
			||||||
 | 
					           == './manage.py migrate auth --merge'
 | 
				
			||||||
							
								
								
									
										17
									
								
								tests/rules/test_dry.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								tests/rules/test_dry.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.dry import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='cd cd foo'),
 | 
				
			||||||
 | 
					    Command(script='git git push origin/master')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('cd cd foo'), 'cd foo'),
 | 
				
			||||||
 | 
					    (Command('git git push origin/master'), 'git push origin/master')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										39
									
								
								tests/rules/test_git_add.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										39
									
								
								tests/rules/test_git_add.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,39 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.git_add import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def did_not_match(target, did_you_forget=True):
 | 
				
			||||||
 | 
					    error = ("error: pathspec '{}' did not match any "
 | 
				
			||||||
 | 
					             "file(s) known to git.".format(target))
 | 
				
			||||||
 | 
					    if did_you_forget:
 | 
				
			||||||
 | 
					        error = ("{}\nDid you forget to 'git add'?'".format(error))
 | 
				
			||||||
 | 
					    return error
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='git submodule update unknown',
 | 
				
			||||||
 | 
					            stderr=did_not_match('unknown')),
 | 
				
			||||||
 | 
					    Command(script='git commit unknown',
 | 
				
			||||||
 | 
					            stderr=did_not_match('unknown'))])  # Older versions of Git
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='git submodule update known', stderr=('')),
 | 
				
			||||||
 | 
					    Command(script='git commit known', stderr=('')),
 | 
				
			||||||
 | 
					    Command(script='git commit unknown',  # Newer versions of Git
 | 
				
			||||||
 | 
					            stderr=did_not_match('unknown', False))])
 | 
				
			||||||
 | 
					def test_not_match(command):
 | 
				
			||||||
 | 
					    assert not match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('git submodule update unknown', stderr=did_not_match('unknown')),
 | 
				
			||||||
 | 
					     'git add -- unknown && git submodule update unknown'),
 | 
				
			||||||
 | 
					    (Command('git commit unknown', stderr=did_not_match('unknown')),  # Old Git
 | 
				
			||||||
 | 
					     'git add -- unknown && git commit unknown')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										22
									
								
								tests/rules/test_git_branch_delete.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								tests/rules/test_git_branch_delete.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,22 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.git_branch_delete import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def stderr():
 | 
				
			||||||
 | 
					    return '''error: The branch 'branch' is not fully merged.
 | 
				
			||||||
 | 
					If you are sure you want to delete it, run 'git branch -D branch'.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					'''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match(stderr):
 | 
				
			||||||
 | 
					    assert match(Command('git branch -d branch', stderr=stderr), None)
 | 
				
			||||||
 | 
					    assert not match(Command('git branch -d branch'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('ls', stderr=stderr), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command(stderr):
 | 
				
			||||||
 | 
					    assert get_new_command(Command('git branch -d branch', stderr=stderr), None)\
 | 
				
			||||||
 | 
					        == "git branch -D branch"
 | 
				
			||||||
							
								
								
									
										18
									
								
								tests/rules/test_git_branch_list.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								tests/rules/test_git_branch_list.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,18 @@
 | 
				
			|||||||
 | 
					from thefuck import shells
 | 
				
			||||||
 | 
					from thefuck.rules.git_branch_list import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match():
 | 
				
			||||||
 | 
					    assert match(Command('git branch list'), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_not_match():
 | 
				
			||||||
 | 
					    assert not match(Command(), None)
 | 
				
			||||||
 | 
					    assert not match(Command('git commit'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('git branch'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('git stash list'), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command():
 | 
				
			||||||
 | 
					    assert (get_new_command(Command('git branch list'), None) ==
 | 
				
			||||||
 | 
					            shells.and_('git branch --delete list', 'git branch'))
 | 
				
			||||||
							
								
								
									
										37
									
								
								tests/rules/test_git_checkout.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										37
									
								
								tests/rules/test_git_checkout.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,37 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.git_checkout import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def did_not_match(target, did_you_forget=False):
 | 
				
			||||||
 | 
					    error = ("error: pathspec '{}' did not match any "
 | 
				
			||||||
 | 
					             "file(s) known to git.".format(target))
 | 
				
			||||||
 | 
					    if did_you_forget:
 | 
				
			||||||
 | 
					        error = ("{}\nDid you forget to 'git add'?'".format(error))
 | 
				
			||||||
 | 
					    return error
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='git checkout unknown', stderr=did_not_match('unknown')),
 | 
				
			||||||
 | 
					    Command(script='git commit unknown', stderr=did_not_match('unknown'))])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='git submodule update unknown',
 | 
				
			||||||
 | 
					            stderr=did_not_match('unknown', True)),
 | 
				
			||||||
 | 
					    Command(script='git checkout known', stderr=('')),
 | 
				
			||||||
 | 
					    Command(script='git commit known', stderr=(''))])
 | 
				
			||||||
 | 
					def test_not_match(command):
 | 
				
			||||||
 | 
					    assert not match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command(script='git checkout unknown', stderr=did_not_match('unknown')),
 | 
				
			||||||
 | 
					     'git branch unknown && git checkout unknown'),
 | 
				
			||||||
 | 
					    (Command('git commit unknown', stderr=did_not_match('unknown')),
 | 
				
			||||||
 | 
					     'git branch unknown && git commit unknown')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										23
									
								
								tests/rules/test_git_diff_staged.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										23
									
								
								tests/rules/test_git_diff_staged.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,23 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.git_diff_staged import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [Command(script='git diff')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='git diff --staged'),
 | 
				
			||||||
 | 
					    Command(script='git tag'),
 | 
				
			||||||
 | 
					    Command(script='git branch'),
 | 
				
			||||||
 | 
					    Command(script='git log')])
 | 
				
			||||||
 | 
					def test_not_match(command):
 | 
				
			||||||
 | 
					    assert not match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('git diff'), 'git diff --staged')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
@@ -25,6 +25,16 @@ stats
 | 
				
			|||||||
"""
 | 
					"""
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def git_not_command_closest():
 | 
				
			||||||
 | 
					    return '''git: 'tags' is not a git command. See 'git --help'.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Did you mean one of these?
 | 
				
			||||||
 | 
						stage
 | 
				
			||||||
 | 
						tag
 | 
				
			||||||
 | 
					'''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@pytest.fixture
 | 
					@pytest.fixture
 | 
				
			||||||
def git_command():
 | 
					def git_command():
 | 
				
			||||||
    return "* master"
 | 
					    return "* master"
 | 
				
			||||||
@@ -37,8 +47,11 @@ def test_match(git_not_command, git_command, git_not_command_one_of_this):
 | 
				
			|||||||
    assert not match(Command('git branch', stderr=git_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):
 | 
					def test_get_new_command(git_not_command, git_not_command_one_of_this,
 | 
				
			||||||
    assert get_new_command(Command('git brnch', stderr=git_not_command), None)\
 | 
					                         git_not_command_closest):
 | 
				
			||||||
        == 'git branch'
 | 
					    assert get_new_command(Command('git brnch', stderr=git_not_command), None) \
 | 
				
			||||||
 | 
					           == 'git branch'
 | 
				
			||||||
    assert get_new_command(Command('git st', stderr=git_not_command_one_of_this),
 | 
					    assert get_new_command(Command('git st', stderr=git_not_command_one_of_this),
 | 
				
			||||||
                           None) == 'git status'
 | 
					                           None) == 'git status'
 | 
				
			||||||
 | 
					    assert get_new_command(Command('git tags', stderr=git_not_command_closest),
 | 
				
			||||||
 | 
					                           None) == 'git tag'
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										29
									
								
								tests/rules/test_git_pull.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								tests/rules/test_git_pull.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,29 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.git_pull import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def stderr():
 | 
				
			||||||
 | 
					    return '''There is no tracking information for the current branch.
 | 
				
			||||||
 | 
					Please specify which branch you want to merge with.
 | 
				
			||||||
 | 
					See git-pull(1) for details
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    git pull <remote> <branch>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					If you wish to set tracking information for this branch you can do so with:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    git branch --set-upstream-to=<remote>/<branch> master
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					'''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match(stderr):
 | 
				
			||||||
 | 
					    assert match(Command('git pull', stderr=stderr), None)
 | 
				
			||||||
 | 
					    assert not match(Command('git pull'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('ls', stderr=stderr), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command(stderr):
 | 
				
			||||||
 | 
					    assert get_new_command(Command('git pull', stderr=stderr), None) \
 | 
				
			||||||
 | 
					           == "git branch --set-upstream-to=origin/master master && git pull"
 | 
				
			||||||
							
								
								
									
										37
									
								
								tests/rules/test_git_stash.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										37
									
								
								tests/rules/test_git_stash.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,37 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.git_stash import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					cherry_pick_error = (
 | 
				
			||||||
 | 
					        'error: Your local changes would be overwritten by cherry-pick.\n'
 | 
				
			||||||
 | 
					        'hint: Commit your changes or stash them to proceed.\n'
 | 
				
			||||||
 | 
					        'fatal: cherry-pick failed')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					rebase_error = (
 | 
				
			||||||
 | 
					        'Cannot rebase: Your index contains uncommitted changes.\n'
 | 
				
			||||||
 | 
					        'Please commit or stash them.')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='git cherry-pick a1b2c3d', stderr=cherry_pick_error),
 | 
				
			||||||
 | 
					    Command(script='git rebase -i HEAD~7', stderr=rebase_error)])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='git cherry-pick a1b2c3d', stderr=('')),
 | 
				
			||||||
 | 
					    Command(script='git rebase -i HEAD~7', stderr=(''))])
 | 
				
			||||||
 | 
					def test_not_match(command):
 | 
				
			||||||
 | 
					    assert not match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command(script='git cherry-pick a1b2c3d', stderr=cherry_pick_error),
 | 
				
			||||||
 | 
					     'git stash && git cherry-pick a1b2c3d'),
 | 
				
			||||||
 | 
					    (Command('git rebase -i HEAD~7', stderr=rebase_error),
 | 
				
			||||||
 | 
					     'git stash && git rebase -i HEAD~7')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										17
									
								
								tests/rules/test_go_run.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								tests/rules/test_go_run.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.go_run import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='go run foo'), 
 | 
				
			||||||
 | 
					    Command(script='go run bar')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('go run foo'), 'go run foo.go'),
 | 
				
			||||||
 | 
					    (Command('go run bar'), 'go run bar.go')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										12
									
								
								tests/rules/test_grep_recursive.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								tests/rules/test_grep_recursive.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
				
			|||||||
 | 
					from thefuck.rules.grep_recursive import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match():
 | 
				
			||||||
 | 
					    assert match(Command('grep blah .', stderr='grep: .: Is a directory'), None)
 | 
				
			||||||
 | 
					    assert not match(Command(), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command():
 | 
				
			||||||
 | 
					    assert get_new_command(
 | 
				
			||||||
 | 
					        Command('grep blah .'), None) == 'grep -r blah .'
 | 
				
			||||||
							
								
								
									
										35
									
								
								tests/rules/test_history.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								tests/rules/test_history.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,35 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.history import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def history(mocker):
 | 
				
			||||||
 | 
					    return mocker.patch('thefuck.rules.history.get_history',
 | 
				
			||||||
 | 
					                        return_value=['ls cat', 'diff x', 'nocommand x'])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def callables(mocker):
 | 
				
			||||||
 | 
					    return mocker.patch('thefuck.rules.history.get_all_callables',
 | 
				
			||||||
 | 
					                        return_value=['diff', 'ls'])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.usefixtures('history', 'callables', 'no_memoize')
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('script', ['ls cet', 'daff x'])
 | 
				
			||||||
 | 
					def test_match(script):
 | 
				
			||||||
 | 
					    assert match(Command(script=script), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.usefixtures('history', 'callables', 'no_memoize')
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('script', ['apt-get', 'nocommand y'])
 | 
				
			||||||
 | 
					def test_not_match(script):
 | 
				
			||||||
 | 
					    assert not match(Command(script=script), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.usefixtures('history', 'callables', 'no_memoize')
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('script, result', [
 | 
				
			||||||
 | 
					    ('ls cet', 'ls cat'),
 | 
				
			||||||
 | 
					    ('daff x', 'diff x')])
 | 
				
			||||||
 | 
					def test_get_new_command(script, result):
 | 
				
			||||||
 | 
					    assert get_new_command(Command(script), None) == result
 | 
				
			||||||
							
								
								
									
										17
									
								
								tests/rules/test_java.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								tests/rules/test_java.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.java import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='java foo.java'),
 | 
				
			||||||
 | 
					    Command(script='java bar.java')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('java foo.java'), 'java foo'),
 | 
				
			||||||
 | 
					    (Command('java bar.java'), 'java bar')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										17
									
								
								tests/rules/test_javac.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								tests/rules/test_javac.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.javac import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='javac foo'),
 | 
				
			||||||
 | 
					    Command(script='javac bar')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('javac foo'), 'javac foo.java'),
 | 
				
			||||||
 | 
					    (Command('javac bar'), 'javac bar.java')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
@@ -3,9 +3,12 @@ from thefuck.rules.ls_lah import match, get_new_command
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def test_match():
 | 
					def test_match():
 | 
				
			||||||
 | 
					    assert match(Mock(script='ls'), None)
 | 
				
			||||||
    assert match(Mock(script='ls file.py'), None)
 | 
					    assert match(Mock(script='ls file.py'), None)
 | 
				
			||||||
    assert match(Mock(script='ls /opt'), None)
 | 
					    assert match(Mock(script='ls /opt'), None)
 | 
				
			||||||
    assert not match(Mock(script='ls -lah /opt'), None)
 | 
					    assert not match(Mock(script='ls -lah /opt'), None)
 | 
				
			||||||
 | 
					    assert not match(Mock(script='pacman -S binutils'), None)
 | 
				
			||||||
 | 
					    assert not match(Mock(script='lsof'), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def test_get_new_command():
 | 
					def test_get_new_command():
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										34
									
								
								tests/rules/test_man.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								tests/rules/test_man.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,34 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.man import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command('man read'),
 | 
				
			||||||
 | 
					    Command('man 2 read'),
 | 
				
			||||||
 | 
					    Command('man 3 read'),
 | 
				
			||||||
 | 
					    Command('man -s2 read'),
 | 
				
			||||||
 | 
					    Command('man -s3 read'),
 | 
				
			||||||
 | 
					    Command('man -s 2 read'),
 | 
				
			||||||
 | 
					    Command('man -s 3 read')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command('man'),
 | 
				
			||||||
 | 
					    Command('man ')])
 | 
				
			||||||
 | 
					def test_not_match(command):
 | 
				
			||||||
 | 
					    assert not match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('man read'), 'man 3 read'),
 | 
				
			||||||
 | 
					    (Command('man 2 read'), 'man 3 read'),
 | 
				
			||||||
 | 
					    (Command('man 3 read'), 'man 2 read'),
 | 
				
			||||||
 | 
					    (Command('man -s2 read'), 'man -s3 read'),
 | 
				
			||||||
 | 
					    (Command('man -s3 read'), 'man -s2 read'),
 | 
				
			||||||
 | 
					    (Command('man -s 2 read'), 'man -s 3 read'),
 | 
				
			||||||
 | 
					    (Command('man -s 3 read'), 'man -s 2 read')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										12
									
								
								tests/rules/test_man_no_space.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								tests/rules/test_man_no_space.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
				
			|||||||
 | 
					from thefuck.rules.man_no_space import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match():
 | 
				
			||||||
 | 
					    assert match(Command('mandiff', stderr='mandiff: command not found'), None)
 | 
				
			||||||
 | 
					    assert not match(Command(), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command():
 | 
				
			||||||
 | 
					    assert get_new_command(
 | 
				
			||||||
 | 
					        Command('mandiff'), None) == 'man diff'
 | 
				
			||||||
							
								
								
									
										134
									
								
								tests/rules/test_mercurial.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										134
									
								
								tests/rules/test_mercurial.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,134 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					from thefuck.rules.mercurial import (
 | 
				
			||||||
 | 
					    extract_possisiblities, match, get_new_command
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command('hg base', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'base'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of blame, phase, rebase?)'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg branchch', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'branchch'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of branch, branches?)'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg vert', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'vert'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of revert?)'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg lgo -r tip', stderr=(
 | 
				
			||||||
 | 
					        "hg: command 're' is ambiguous:"
 | 
				
			||||||
 | 
					        '\n(did you mean one of log?)'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg rerere', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'rerere'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of revert?)'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg re', stderr=(
 | 
				
			||||||
 | 
					        "hg: command 're' is ambiguous:"
 | 
				
			||||||
 | 
					        '\n    rebase recover remove rename resolve revert'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg re re', stderr=(
 | 
				
			||||||
 | 
					        "hg: command 're' is ambiguous:"
 | 
				
			||||||
 | 
					        '\n    rebase recover remove rename resolve revert'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command('hg', stderr=(
 | 
				
			||||||
 | 
					        '\nMercurial Distributed SCM\n\nbasic commands:'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg asdf', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'asdf'"
 | 
				
			||||||
 | 
					        '\nMercurial Distributed SCM\n\nbasic commands:'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg qwer', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'qwer'"
 | 
				
			||||||
 | 
					        '\nMercurial Distributed SCM\n\nbasic commands:'
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg me', stderr=(
 | 
				
			||||||
 | 
					        "\nabort: no repository found in './thefuck' (.hg not found)!"
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg reb', stderr=(
 | 
				
			||||||
 | 
					        "\nabort: no repository found in './thefuck' (.hg not found)!"
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					    Command('hg co', stderr=(
 | 
				
			||||||
 | 
					        "\nabort: no repository found in './thefuck' (.hg not found)!"
 | 
				
			||||||
 | 
					    )),
 | 
				
			||||||
 | 
					])
 | 
				
			||||||
 | 
					def test_not_match(command):
 | 
				
			||||||
 | 
					    assert not match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, possibilities', [
 | 
				
			||||||
 | 
					    (Command('hg base', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'base'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of blame, phase, rebase?)'
 | 
				
			||||||
 | 
					    )), ['blame', 'phase', 'rebase']),
 | 
				
			||||||
 | 
					    (Command('hg branchch', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'branchch'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of branch, branches?)'
 | 
				
			||||||
 | 
					    )), ['branch', 'branches']),
 | 
				
			||||||
 | 
					    (Command('hg vert', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'vert'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of revert?)'
 | 
				
			||||||
 | 
					    )), ['revert']),
 | 
				
			||||||
 | 
					    (Command('hg lgo -r tip', stderr=(
 | 
				
			||||||
 | 
					        "hg: command 're' is ambiguous:"
 | 
				
			||||||
 | 
					        '\n(did you mean one of log?)'
 | 
				
			||||||
 | 
					    )), ['log']),
 | 
				
			||||||
 | 
					    (Command('hg rerere', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'rerere'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of revert?)'
 | 
				
			||||||
 | 
					    )), ['revert']),
 | 
				
			||||||
 | 
					    (Command('hg re', stderr=(
 | 
				
			||||||
 | 
					        "hg: command 're' is ambiguous:"
 | 
				
			||||||
 | 
					        '\n    rebase recover remove rename resolve revert'
 | 
				
			||||||
 | 
					    )), ['rebase', 'recover', 'remove', 'rename', 'resolve', 'revert']),
 | 
				
			||||||
 | 
					    (Command('hg re re', stderr=(
 | 
				
			||||||
 | 
					        "hg: command 're' is ambiguous:"
 | 
				
			||||||
 | 
					        '\n    rebase recover remove rename resolve revert'
 | 
				
			||||||
 | 
					    )), ['rebase', 'recover', 'remove', 'rename', 'resolve', 'revert']),
 | 
				
			||||||
 | 
					])
 | 
				
			||||||
 | 
					def test_extract_possisiblities(command, possibilities):
 | 
				
			||||||
 | 
					    assert extract_possisiblities(command) == possibilities
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('hg base', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'base'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of blame, phase, rebase?)'
 | 
				
			||||||
 | 
					    )), 'hg rebase'),
 | 
				
			||||||
 | 
					    (Command('hg branchch', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'branchch'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of branch, branches?)'
 | 
				
			||||||
 | 
					    )), 'hg branch'),
 | 
				
			||||||
 | 
					    (Command('hg vert', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'vert'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of revert?)'
 | 
				
			||||||
 | 
					    )), 'hg revert'),
 | 
				
			||||||
 | 
					    (Command('hg lgo -r tip', stderr=(
 | 
				
			||||||
 | 
					        "hg: command 're' is ambiguous:"
 | 
				
			||||||
 | 
					        '\n(did you mean one of log?)'
 | 
				
			||||||
 | 
					    )), 'hg log -r tip'),
 | 
				
			||||||
 | 
					    (Command('hg rerere', stderr=(
 | 
				
			||||||
 | 
					        "hg: unknown command 'rerere'"
 | 
				
			||||||
 | 
					        '\n(did you mean one of revert?)'
 | 
				
			||||||
 | 
					    )), 'hg revert'),
 | 
				
			||||||
 | 
					    (Command('hg re', stderr=(
 | 
				
			||||||
 | 
					        "hg: command 're' is ambiguous:"
 | 
				
			||||||
 | 
					        '\n    rebase recover remove rename resolve revert'
 | 
				
			||||||
 | 
					    )), 'hg rebase'),
 | 
				
			||||||
 | 
					    (Command('hg re re', stderr=(
 | 
				
			||||||
 | 
					        "hg: command 're' is ambiguous:"
 | 
				
			||||||
 | 
					        '\n    rebase recover remove rename resolve revert'
 | 
				
			||||||
 | 
					    )), 'hg rebase re'),
 | 
				
			||||||
 | 
					])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
@@ -1,19 +1,42 @@
 | 
				
			|||||||
from mock import patch, Mock
 | 
					import pytest
 | 
				
			||||||
from thefuck.rules.no_command import match, get_new_command
 | 
					from thefuck.rules.no_command import match, get_new_command, get_all_callables
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def test_match():
 | 
					@pytest.fixture(autouse=True)
 | 
				
			||||||
    with patch('thefuck.rules.no_command._get_all_bins',
 | 
					def _safe(mocker):
 | 
				
			||||||
               return_value=['vim', 'apt-get']):
 | 
					    mocker.patch('thefuck.rules.no_command._safe', return_value=[])
 | 
				
			||||||
        assert match(Mock(stderr='vom: not found', script='vom file.py'), None)
 | 
					 | 
				
			||||||
        assert not match(Mock(stderr='qweqwe: not found', script='qweqwe'), None)
 | 
					 | 
				
			||||||
        assert not match(Mock(stderr='some text', script='vom file.py'), None)
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def test_get_new_command():
 | 
					@pytest.fixture(autouse=True)
 | 
				
			||||||
    with patch('thefuck.rules.no_command._get_all_bins',
 | 
					def get_aliases(mocker):
 | 
				
			||||||
               return_value=['vim', 'apt-get']):
 | 
					    mocker.patch('thefuck.rules.no_command.get_aliases',
 | 
				
			||||||
        assert get_new_command(
 | 
					                 return_value=['vim', 'apt-get', 'fsck', 'fuck'])
 | 
				
			||||||
            Mock(stderr='vom: not found',
 | 
					
 | 
				
			||||||
                 script='vom file.py'),
 | 
					
 | 
				
			||||||
            None) == 'vim file.py'
 | 
					@pytest.mark.usefixtures('no_memoize')
 | 
				
			||||||
 | 
					def test_get_all_callables(*args):
 | 
				
			||||||
 | 
					    all_callables = get_all_callables()
 | 
				
			||||||
 | 
					    assert 'vim' in all_callables
 | 
				
			||||||
 | 
					    assert 'fsck' in all_callables
 | 
				
			||||||
 | 
					    assert 'fuck' not in all_callables
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.usefixtures('no_memoize')
 | 
				
			||||||
 | 
					def test_match(*args):
 | 
				
			||||||
 | 
					    assert match(Command(stderr='vom: not found', script='vom file.py'), None)
 | 
				
			||||||
 | 
					    assert match(Command(stderr='fucck: not found', script='fucck'), None)
 | 
				
			||||||
 | 
					    assert not match(Command(stderr='qweqwe: not found', script='qweqwe'), None)
 | 
				
			||||||
 | 
					    assert not match(Command(stderr='some text', script='vom file.py'), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.usefixtures('no_memoize')
 | 
				
			||||||
 | 
					def test_get_new_command(*args):
 | 
				
			||||||
 | 
					    assert get_new_command(
 | 
				
			||||||
 | 
					        Command(stderr='vom: not found',
 | 
				
			||||||
 | 
					                script='vom file.py'),
 | 
				
			||||||
 | 
					        None) == 'vim file.py'
 | 
				
			||||||
 | 
					    assert get_new_command(
 | 
				
			||||||
 | 
					        Command(stderr='fucck: not found',
 | 
				
			||||||
 | 
					                script='fucck'),
 | 
				
			||||||
 | 
					        Command) == 'fsck'
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										19
									
								
								tests/rules/test_no_such_file.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								tests/rules/test_no_such_file.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,19 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.no_such_file import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='mv foo bar/foo', stderr="mv: cannot move 'foo' to 'bar/foo': No such file or directory"),
 | 
				
			||||||
 | 
					    Command(script='mv foo bar/', stderr="mv: cannot move 'foo' to 'bar/': No such file or directory"),
 | 
				
			||||||
 | 
					    ])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command(script='mv foo bar/foo', stderr="mv: cannot move 'foo' to 'bar/foo': No such file or directory"), 'mkdir -p bar && mv foo bar/foo'),
 | 
				
			||||||
 | 
					    (Command(script='mv foo bar/', stderr="mv: cannot move 'foo' to 'bar/': No such file or directory"), 'mkdir -p bar && mv foo bar/'),
 | 
				
			||||||
 | 
					    ])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										25
									
								
								tests/rules/test_open.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								tests/rules/test_open.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,25 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.open import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='open foo.com'),
 | 
				
			||||||
 | 
					    Command(script='open foo.ly'),
 | 
				
			||||||
 | 
					    Command(script='open foo.org'),
 | 
				
			||||||
 | 
					    Command(script='open foo.net'),
 | 
				
			||||||
 | 
					    Command(script='open foo.se'),
 | 
				
			||||||
 | 
					    Command(script='open foo.io')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('open foo.com'), 'open http://foo.com'),
 | 
				
			||||||
 | 
					    (Command('open foo.ly'), 'open http://foo.ly'),
 | 
				
			||||||
 | 
					    (Command('open foo.org'), 'open http://foo.org'),
 | 
				
			||||||
 | 
					    (Command('open foo.net'), 'open http://foo.net'),
 | 
				
			||||||
 | 
					    (Command('open foo.se'), 'open http://foo.se'),
 | 
				
			||||||
 | 
					    (Command('open foo.io'), 'open http://foo.io')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										69
									
								
								tests/rules/test_pacman.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										69
									
								
								tests/rules/test_pacman.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,69 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from mock import patch
 | 
				
			||||||
 | 
					from thefuck.rules import pacman
 | 
				
			||||||
 | 
					from thefuck.rules.pacman import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					pacman_cmd = getattr(pacman, 'pacman', 'pacman')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					PKGFILE_OUTPUT_CONVERT = '''
 | 
				
			||||||
 | 
					extra/imagemagick 6.9.1.0-1\t/usr/bin/convert
 | 
				
			||||||
 | 
					'''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					PKGFILE_OUTPUT_VIM = '''
 | 
				
			||||||
 | 
					extra/gvim 7.4.712-1        \t/usr/bin/vim
 | 
				
			||||||
 | 
					extra/gvim-python3 7.4.712-1\t/usr/bin/vim
 | 
				
			||||||
 | 
					extra/vim 7.4.712-1         \t/usr/bin/vim
 | 
				
			||||||
 | 
					extra/vim-minimal 7.4.712-1 \t/usr/bin/vim
 | 
				
			||||||
 | 
					extra/vim-python3 7.4.712-1 \t/usr/bin/vim
 | 
				
			||||||
 | 
					'''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.skipif(not getattr(pacman, 'enabled_by_default', True),
 | 
				
			||||||
 | 
					                    reason='Skip if pacman is not available')
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='vim', stderr='vim: command not found'),
 | 
				
			||||||
 | 
					    Command(script='sudo vim', stderr='sudo: vim: command not found')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, return_value', [
 | 
				
			||||||
 | 
					    (Command(script='vim', stderr='vim: command not found'), PKGFILE_OUTPUT_VIM),
 | 
				
			||||||
 | 
					    (Command(script='sudo vim', stderr='sudo: vim: command not found'), PKGFILE_OUTPUT_VIM)])
 | 
				
			||||||
 | 
					@patch('thefuck.rules.pacman.subprocess')
 | 
				
			||||||
 | 
					@patch.multiple(pacman, create=True, pacman=pacman_cmd)
 | 
				
			||||||
 | 
					def test_match_mocked(subp_mock, command, return_value):
 | 
				
			||||||
 | 
					    subp_mock.check_output.return_value = return_value
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='vim', stderr=''), Command(),
 | 
				
			||||||
 | 
					    Command(script='sudo vim', stderr=''), Command()])
 | 
				
			||||||
 | 
					def test_not_match(command):
 | 
				
			||||||
 | 
					    assert not match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.skipif(not getattr(pacman, 'enabled_by_default', True),
 | 
				
			||||||
 | 
					                    reason='Skip if pacman is not available')
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('vim'), '{} -S extra/gvim && vim'.format(pacman_cmd)),
 | 
				
			||||||
 | 
					    (Command('sudo vim'), '{} -S extra/gvim && sudo vim'.format(pacman_cmd)),
 | 
				
			||||||
 | 
					    (Command('convert'), '{} -S extra/imagemagick && convert'.format(pacman_cmd)),
 | 
				
			||||||
 | 
					    (Command('sudo convert'), '{} -S extra/imagemagick && sudo convert'.format(pacman_cmd))])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command, mocker):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command, return_value', [
 | 
				
			||||||
 | 
					    (Command('vim'), '{} -S extra/gvim && vim'.format(pacman_cmd), PKGFILE_OUTPUT_VIM),
 | 
				
			||||||
 | 
					    (Command('sudo vim'), '{} -S extra/gvim && sudo vim'.format(pacman_cmd), PKGFILE_OUTPUT_VIM),
 | 
				
			||||||
 | 
					    (Command('convert'), '{} -S extra/imagemagick && convert'.format(pacman_cmd), PKGFILE_OUTPUT_CONVERT),
 | 
				
			||||||
 | 
					    (Command('sudo convert'), '{} -S extra/imagemagick && sudo convert'.format(pacman_cmd), PKGFILE_OUTPUT_CONVERT)])
 | 
				
			||||||
 | 
					@patch('thefuck.rules.pacman.subprocess')
 | 
				
			||||||
 | 
					@patch.multiple(pacman, create=True, pacman=pacman_cmd)
 | 
				
			||||||
 | 
					def test_get_new_command_mocked(subp_mock, command, new_command, return_value):
 | 
				
			||||||
 | 
					    subp_mock.check_output.return_value = return_value
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										17
									
								
								tests/rules/test_python_execute.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								tests/rules/test_python_execute.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.python_execute import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='python foo'),
 | 
				
			||||||
 | 
					    Command(script='python bar')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('python foo'), 'python foo.py'),
 | 
				
			||||||
 | 
					    (Command('python bar'), 'python bar.py')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										19
									
								
								tests/rules/test_quotation_marks.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								tests/rules/test_quotation_marks.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,19 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.quotation_marks import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script="git commit -m \'My Message\""),
 | 
				
			||||||
 | 
					    Command(script="git commit -am \"Mismatched Quotation Marks\'"),
 | 
				
			||||||
 | 
					    Command(script="echo \"hello\'")])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command("git commit -m \'My Message\""), "git commit -m \"My Message\""),
 | 
				
			||||||
 | 
					    (Command("git commit -am \"Mismatched Quotation Marks\'"), "git commit -am \"Mismatched Quotation Marks\""),
 | 
				
			||||||
 | 
					    (Command("echo \"hello\'"), "echo \"hello\"")])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
							
								
								
									
										28
									
								
								tests/rules/test_sed_unterminated_s.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										28
									
								
								tests/rules/test_sed_unterminated_s.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,28 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.sed_unterminated_s import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def sed_unterminated_s():
 | 
				
			||||||
 | 
					    return "sed: -e expression #1, char 9: unterminated `s' command"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match(sed_unterminated_s):
 | 
				
			||||||
 | 
					    assert match(Command('sed -e s/foo/bar', stderr=sed_unterminated_s), None)
 | 
				
			||||||
 | 
					    assert match(Command('sed -es/foo/bar', stderr=sed_unterminated_s), None)
 | 
				
			||||||
 | 
					    assert match(Command('sed -e s/foo/bar -e s/baz/quz', stderr=sed_unterminated_s), None)
 | 
				
			||||||
 | 
					    assert not match(Command('sed -e s/foo/bar'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('sed -es/foo/bar'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('sed -e s/foo/bar -e s/baz/quz'), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command(sed_unterminated_s):
 | 
				
			||||||
 | 
					    assert get_new_command(Command('sed -e s/foo/bar', stderr=sed_unterminated_s), None) \
 | 
				
			||||||
 | 
					            == 'sed -e s/foo/bar/'
 | 
				
			||||||
 | 
					    assert get_new_command(Command('sed -es/foo/bar', stderr=sed_unterminated_s), None) \
 | 
				
			||||||
 | 
					            == 'sed -es/foo/bar/'
 | 
				
			||||||
 | 
					    assert get_new_command(Command(r"sed -e 's/\/foo/bar'", stderr=sed_unterminated_s), None) \
 | 
				
			||||||
 | 
					            == r"sed -e 's/\/foo/bar/'"
 | 
				
			||||||
 | 
					    assert get_new_command(Command(r"sed -e s/foo/bar -es/baz/quz", stderr=sed_unterminated_s), None) \
 | 
				
			||||||
 | 
					            == r"sed -e s/foo/bar/ -es/baz/quz/"
 | 
				
			||||||
@@ -8,6 +8,9 @@ from tests.utils import Command
 | 
				
			|||||||
    ('permission denied', ''),
 | 
					    ('permission denied', ''),
 | 
				
			||||||
    ("npm ERR! Error: EACCES, unlink", ''),
 | 
					    ("npm ERR! Error: EACCES, unlink", ''),
 | 
				
			||||||
    ('requested operation requires superuser privilege', ''),
 | 
					    ('requested operation requires superuser privilege', ''),
 | 
				
			||||||
 | 
					    ('need to be root', ''),
 | 
				
			||||||
 | 
					    ('need root', ''),
 | 
				
			||||||
 | 
					    ('must be root', ''),
 | 
				
			||||||
    ('', "error: [Errno 13] Permission denied: '/usr/local/lib/python2.7/dist-packages/ipaddr.py'")])
 | 
					    ('', "error: [Errno 13] Permission denied: '/usr/local/lib/python2.7/dist-packages/ipaddr.py'")])
 | 
				
			||||||
def test_match(stderr, stdout):
 | 
					def test_match(stderr, stdout):
 | 
				
			||||||
    assert match(Command(stderr=stderr, stdout=stdout), None)
 | 
					    assert match(Command(stderr=stderr, stdout=stdout), None)
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										18
									
								
								tests/rules/test_systemctl.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								tests/rules/test_systemctl.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,18 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.systemctl import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match():
 | 
				
			||||||
 | 
					    assert match(Command('systemctl nginx start', stderr='Unknown operation \'nginx\'.'), None)
 | 
				
			||||||
 | 
					    assert match(Command('sudo systemctl nginx start', stderr='Unknown operation \'nginx\'.'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('systemctl start nginx'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('systemctl start nginx'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('sudo systemctl nginx', stderr='Unknown operation \'nginx\'.'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('systemctl nginx', stderr='Unknown operation \'nginx\'.'), None)
 | 
				
			||||||
 | 
					    assert not match(Command('systemctl start wtf', stderr='Failed to start wtf.service: Unit wtf.service failed to load: No such file or directory.'), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command():
 | 
				
			||||||
 | 
					    assert get_new_command(Command('systemctl nginx start'), None) == "systemctl start nginx"
 | 
				
			||||||
 | 
					    assert get_new_command(Command('sudo systemctl nginx start'), None) == "sudo systemctl start nginx"
 | 
				
			||||||
							
								
								
									
										19
									
								
								tests/rules/test_tmux.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								tests/rules/test_tmux.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,19 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.tmux import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def tmux_ambiguous():
 | 
				
			||||||
 | 
					    return "ambiguous command: list, could be: " \
 | 
				
			||||||
 | 
					           "list-buffers, list-clients, list-commands, list-keys, " \
 | 
				
			||||||
 | 
					           "list-panes, list-sessions, list-windows"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_match(tmux_ambiguous):
 | 
				
			||||||
 | 
					    assert match(Command('tmux list', stderr=tmux_ambiguous), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_get_new_command(tmux_ambiguous):
 | 
				
			||||||
 | 
					    assert get_new_command(Command('tmux list', stderr=tmux_ambiguous), None)\
 | 
				
			||||||
 | 
					        == 'tmux list-keys'
 | 
				
			||||||
							
								
								
									
										23
									
								
								tests/rules/test_whois.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										23
									
								
								tests/rules/test_whois.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,23 @@
 | 
				
			|||||||
 | 
					import pytest
 | 
				
			||||||
 | 
					from thefuck.rules.whois import match, get_new_command
 | 
				
			||||||
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command', [
 | 
				
			||||||
 | 
					    Command(script='whois https://en.wikipedia.org/wiki/Main_Page'),
 | 
				
			||||||
 | 
					    Command(script='whois https://en.wikipedia.org/'),
 | 
				
			||||||
 | 
					    Command(script='whois en.wikipedia.org')])
 | 
				
			||||||
 | 
					def test_match(command):
 | 
				
			||||||
 | 
					    assert match(command, None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_not_match():
 | 
				
			||||||
 | 
					    assert not match(Command(script='whois'), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('command, new_command', [
 | 
				
			||||||
 | 
					    (Command('whois https://en.wikipedia.org/wiki/Main_Page'), 'whois en.wikipedia.org'),
 | 
				
			||||||
 | 
					    (Command('whois https://en.wikipedia.org/'), 'whois en.wikipedia.org'),
 | 
				
			||||||
 | 
					    (Command('whois en.wikipedia.org'), 'whois wikipedia.org')])
 | 
				
			||||||
 | 
					def test_get_new_command(command, new_command):
 | 
				
			||||||
 | 
					    assert get_new_command(command, None) == new_command
 | 
				
			||||||
@@ -14,10 +14,8 @@ def test_default(enabled, rules, result):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@pytest.fixture
 | 
					@pytest.fixture
 | 
				
			||||||
def load_source(monkeypatch):
 | 
					def load_source(mocker):
 | 
				
			||||||
    mock = Mock()
 | 
					    return mocker.patch('thefuck.conf.load_source')
 | 
				
			||||||
    monkeypatch.setattr('thefuck.conf.load_source', mock)
 | 
					 | 
				
			||||||
    return mock
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@pytest.fixture
 | 
					@pytest.fixture
 | 
				
			||||||
@@ -40,12 +38,14 @@ class TestSettingsFromFile(object):
 | 
				
			|||||||
        load_source.return_value = Mock(rules=['test'],
 | 
					        load_source.return_value = Mock(rules=['test'],
 | 
				
			||||||
                                        wait_command=10,
 | 
					                                        wait_command=10,
 | 
				
			||||||
                                        require_confirmation=True,
 | 
					                                        require_confirmation=True,
 | 
				
			||||||
                                        no_colors=True)
 | 
					                                        no_colors=True,
 | 
				
			||||||
 | 
					                                        priority={'vim': 100})
 | 
				
			||||||
        settings = conf.get_settings(Mock())
 | 
					        settings = conf.get_settings(Mock())
 | 
				
			||||||
        assert settings.rules == ['test']
 | 
					        assert settings.rules == ['test']
 | 
				
			||||||
        assert settings.wait_command == 10
 | 
					        assert settings.wait_command == 10
 | 
				
			||||||
        assert settings.require_confirmation is True
 | 
					        assert settings.require_confirmation is True
 | 
				
			||||||
        assert settings.no_colors is True
 | 
					        assert settings.no_colors is True
 | 
				
			||||||
 | 
					        assert settings.priority == {'vim': 100}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_from_file_with_DEFAULT(self, load_source):
 | 
					    def test_from_file_with_DEFAULT(self, load_source):
 | 
				
			||||||
        load_source.return_value = Mock(rules=conf.DEFAULT_RULES + ['test'],
 | 
					        load_source.return_value = Mock(rules=conf.DEFAULT_RULES + ['test'],
 | 
				
			||||||
@@ -62,12 +62,14 @@ class TestSettingsFromEnv(object):
 | 
				
			|||||||
        environ.update({'THEFUCK_RULES': 'bash:lisp',
 | 
					        environ.update({'THEFUCK_RULES': 'bash:lisp',
 | 
				
			||||||
                        'THEFUCK_WAIT_COMMAND': '55',
 | 
					                        'THEFUCK_WAIT_COMMAND': '55',
 | 
				
			||||||
                        'THEFUCK_REQUIRE_CONFIRMATION': 'true',
 | 
					                        'THEFUCK_REQUIRE_CONFIRMATION': 'true',
 | 
				
			||||||
                        'THEFUCK_NO_COLORS': 'false'})
 | 
					                        'THEFUCK_NO_COLORS': 'false',
 | 
				
			||||||
 | 
					                        'THEFUCK_PRIORITY': 'bash=10:lisp=wrong:vim=15'})
 | 
				
			||||||
        settings = conf.get_settings(Mock())
 | 
					        settings = conf.get_settings(Mock())
 | 
				
			||||||
        assert settings.rules == ['bash', 'lisp']
 | 
					        assert settings.rules == ['bash', 'lisp']
 | 
				
			||||||
        assert settings.wait_command == 55
 | 
					        assert settings.wait_command == 55
 | 
				
			||||||
        assert settings.require_confirmation is True
 | 
					        assert settings.require_confirmation is True
 | 
				
			||||||
        assert settings.no_colors is False
 | 
					        assert settings.no_colors is False
 | 
				
			||||||
 | 
					        assert settings.priority == {'bash': 10, 'vim': 15}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_from_env_with_DEFAULT(self, environ):
 | 
					    def test_from_env_with_DEFAULT(self, environ):
 | 
				
			||||||
        environ.update({'THEFUCK_RULES': 'DEFAULT_RULES:bash:lisp'})
 | 
					        environ.update({'THEFUCK_RULES': 'DEFAULT_RULES:bash:lisp'})
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,42 +0,0 @@
 | 
				
			|||||||
import pytest
 | 
					 | 
				
			||||||
from mock import Mock
 | 
					 | 
				
			||||||
from thefuck.history import History
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
class TestHistory(object):
 | 
					 | 
				
			||||||
    @pytest.fixture(autouse=True)
 | 
					 | 
				
			||||||
    def process(self, monkeypatch):
 | 
					 | 
				
			||||||
        Process = Mock()
 | 
					 | 
				
			||||||
        Process.return_value.parent.return_value.pid = 1
 | 
					 | 
				
			||||||
        monkeypatch.setattr('thefuck.history.Process', Process)
 | 
					 | 
				
			||||||
        return Process
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    @pytest.fixture(autouse=True)
 | 
					 | 
				
			||||||
    def db(self, monkeypatch):
 | 
					 | 
				
			||||||
        class DBMock(dict):
 | 
					 | 
				
			||||||
            def __init__(self):
 | 
					 | 
				
			||||||
                super(DBMock, self).__init__()
 | 
					 | 
				
			||||||
                self.sync = Mock()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            def __call__(self, *args, **kwargs):
 | 
					 | 
				
			||||||
                return self
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        db = DBMock()
 | 
					 | 
				
			||||||
        monkeypatch.setattr('thefuck.history.shelve.open', db)
 | 
					 | 
				
			||||||
        return db
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def test_set(self, db):
 | 
					 | 
				
			||||||
        history = History()
 | 
					 | 
				
			||||||
        history.update(last_command='ls',
 | 
					 | 
				
			||||||
                       last_fixed_command=None)
 | 
					 | 
				
			||||||
        assert db == {'1-last_command': 'ls',
 | 
					 | 
				
			||||||
                      '1-last_fixed_command': None}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def test_get(self, db):
 | 
					 | 
				
			||||||
        history = History()
 | 
					 | 
				
			||||||
        db['1-last_command'] = 'cd ..'
 | 
					 | 
				
			||||||
        assert history.last_command == 'cd ..'
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def test_get_without_value(self):
 | 
					 | 
				
			||||||
        history = History()
 | 
					 | 
				
			||||||
        assert history.last_command is None
 | 
					 | 
				
			||||||
@@ -5,3 +5,10 @@ from thefuck import logs
 | 
				
			|||||||
def test_color():
 | 
					def test_color():
 | 
				
			||||||
    assert logs.color('red', Mock(no_colors=False)) == 'red'
 | 
					    assert logs.color('red', Mock(no_colors=False)) == 'red'
 | 
				
			||||||
    assert logs.color('red', Mock(no_colors=True)) == ''
 | 
					    assert logs.color('red', Mock(no_colors=True)) == ''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_debug(capsys):
 | 
				
			||||||
 | 
					    logs.debug('test', Mock(no_colors=True, debug=True))
 | 
				
			||||||
 | 
					    assert capsys.readouterr() == ('', 'DEBUG: test\n')
 | 
				
			||||||
 | 
					    logs.debug('test', Mock(no_colors=True, debug=False))
 | 
				
			||||||
 | 
					    assert capsys.readouterr() == ('', '')
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -6,34 +6,55 @@ from thefuck import main, conf, types
 | 
				
			|||||||
from tests.utils import Rule, Command
 | 
					from tests.utils import Rule, Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def test_load_rule(monkeypatch):
 | 
					def test_load_rule(mocker):
 | 
				
			||||||
    match = object()
 | 
					    match = object()
 | 
				
			||||||
    get_new_command = object()
 | 
					    get_new_command = object()
 | 
				
			||||||
    load_source = Mock()
 | 
					    load_source = mocker.patch(
 | 
				
			||||||
    load_source.return_value = Mock(match=match,
 | 
					        'thefuck.main.load_source',
 | 
				
			||||||
                                    get_new_command=get_new_command,
 | 
					        return_value=Mock(match=match,
 | 
				
			||||||
                                    enabled_by_default=True)
 | 
					                          get_new_command=get_new_command,
 | 
				
			||||||
    monkeypatch.setattr('thefuck.main.load_source', load_source)
 | 
					                          enabled_by_default=True,
 | 
				
			||||||
 | 
					                          priority=900))
 | 
				
			||||||
    assert main.load_rule(Path('/rules/bash.py')) \
 | 
					    assert main.load_rule(Path('/rules/bash.py')) \
 | 
				
			||||||
           == Rule('bash', match, get_new_command)
 | 
					           == Rule('bash', match, get_new_command, priority=900)
 | 
				
			||||||
    load_source.assert_called_once_with('bash', '/rules/bash.py')
 | 
					    load_source.assert_called_once_with('bash', '/rules/bash.py')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@pytest.mark.parametrize('conf_rules, rules', [
 | 
					class TestGetRules(object):
 | 
				
			||||||
    (conf.DEFAULT_RULES, [Rule('bash', 'bash', 'bash'),
 | 
					    @pytest.fixture(autouse=True)
 | 
				
			||||||
                          Rule('lisp', 'lisp', 'lisp'),
 | 
					    def glob(self, mocker):
 | 
				
			||||||
                          Rule('bash', 'bash', 'bash'),
 | 
					        return mocker.patch('thefuck.main.Path.glob', return_value=[])
 | 
				
			||||||
                          Rule('lisp', 'lisp', 'lisp')]),
 | 
					
 | 
				
			||||||
    (types.RulesNamesList(['bash']), [Rule('bash', 'bash', 'bash'),
 | 
					    def _compare_names(self, rules, names):
 | 
				
			||||||
                                      Rule('bash', 'bash', 'bash')])])
 | 
					        return [r.name for r in rules] == names
 | 
				
			||||||
def test_get_rules(monkeypatch, conf_rules, rules):
 | 
					
 | 
				
			||||||
    monkeypatch.setattr(
 | 
					    @pytest.mark.parametrize('conf_rules, rules', [
 | 
				
			||||||
        'thefuck.main.Path.glob',
 | 
					        (conf.DEFAULT_RULES, ['bash', 'lisp', 'bash', 'lisp']),
 | 
				
			||||||
        lambda *_: [PosixPath('bash.py'), PosixPath('lisp.py')])
 | 
					        (types.RulesNamesList(['bash']), ['bash', 'bash'])])
 | 
				
			||||||
    monkeypatch.setattr('thefuck.main.load_source',
 | 
					    def test_get(self, monkeypatch, glob, conf_rules, rules):
 | 
				
			||||||
                        lambda x, _: Mock(match=x, get_new_command=x,
 | 
					        glob.return_value = [PosixPath('bash.py'), PosixPath('lisp.py')]
 | 
				
			||||||
                                          enabled_by_default=True))
 | 
					        monkeypatch.setattr('thefuck.main.load_source',
 | 
				
			||||||
    assert list(main.get_rules(Path('~'), Mock(rules=conf_rules))) == rules
 | 
					                            lambda x, _: Rule(x))
 | 
				
			||||||
 | 
					        assert self._compare_names(
 | 
				
			||||||
 | 
					            main.get_rules(Path('~'), Mock(rules=conf_rules, priority={})),
 | 
				
			||||||
 | 
					            rules)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    @pytest.mark.parametrize('priority, unordered, ordered', [
 | 
				
			||||||
 | 
					        ({},
 | 
				
			||||||
 | 
					         [Rule('bash', priority=100), Rule('python', priority=5)],
 | 
				
			||||||
 | 
					         ['python', 'bash']),
 | 
				
			||||||
 | 
					        ({},
 | 
				
			||||||
 | 
					         [Rule('lisp', priority=9999), Rule('c', priority=conf.DEFAULT_PRIORITY)],
 | 
				
			||||||
 | 
					         ['c', 'lisp']),
 | 
				
			||||||
 | 
					        ({'python': 9999},
 | 
				
			||||||
 | 
					         [Rule('bash', priority=100), Rule('python', priority=5)],
 | 
				
			||||||
 | 
					         ['bash', 'python'])])
 | 
				
			||||||
 | 
					    def test_ordered_by_priority(self, monkeypatch, priority, unordered, ordered):
 | 
				
			||||||
 | 
					        monkeypatch.setattr('thefuck.main._get_loaded_rules',
 | 
				
			||||||
 | 
					                            lambda *_: unordered)
 | 
				
			||||||
 | 
					        assert self._compare_names(
 | 
				
			||||||
 | 
					            main.get_rules(Path('~'), Mock(priority=priority)),
 | 
				
			||||||
 | 
					            ordered)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class TestGetCommand(object):
 | 
					class TestGetCommand(object):
 | 
				
			||||||
@@ -56,30 +77,23 @@ class TestGetCommand(object):
 | 
				
			|||||||
        monkeypatch.setattr('thefuck.shells.to_shell', lambda x: x)
 | 
					        monkeypatch.setattr('thefuck.shells.to_shell', lambda x: x)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_get_command_calls(self, Popen):
 | 
					    def test_get_command_calls(self, Popen):
 | 
				
			||||||
        assert main.get_command(Mock(), Mock(),
 | 
					        assert main.get_command(Mock(env={}),
 | 
				
			||||||
            ['thefuck', 'apt-get', 'search', 'vim']) \
 | 
					            ['thefuck', 'apt-get', 'search', 'vim']) \
 | 
				
			||||||
               == Command('apt-get search vim', 'stdout', 'stderr')
 | 
					               == Command('apt-get search vim', 'stdout', 'stderr')
 | 
				
			||||||
        Popen.assert_called_once_with('apt-get search vim',
 | 
					        Popen.assert_called_once_with('apt-get search vim',
 | 
				
			||||||
                                      shell=True,
 | 
					                                      shell=True,
 | 
				
			||||||
                                      stdout=PIPE,
 | 
					                                      stdout=PIPE,
 | 
				
			||||||
                                      stderr=PIPE,
 | 
					                                      stderr=PIPE,
 | 
				
			||||||
                                      env={'LANG': 'C'})
 | 
					                                      env={})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    @pytest.mark.parametrize('history, args, result', [
 | 
					    @pytest.mark.parametrize('args, result', [
 | 
				
			||||||
        (Mock(), [''], None),
 | 
					        (['thefuck', 'ls', '-la'], 'ls -la'),
 | 
				
			||||||
        (Mock(last_command='ls', last_fixed_command='ls -la'),
 | 
					        (['thefuck', 'ls'], 'ls')])
 | 
				
			||||||
         ['thefuck', 'fuck'], 'ls -la'),
 | 
					    def test_get_command_script(self, args, result):
 | 
				
			||||||
        (Mock(last_command='ls', last_fixed_command='ls -la'),
 | 
					 | 
				
			||||||
         ['thefuck', 'ls'], 'ls -la'),
 | 
					 | 
				
			||||||
        (Mock(last_command='ls', last_fixed_command=''),
 | 
					 | 
				
			||||||
         ['thefuck', 'ls'], 'ls'),
 | 
					 | 
				
			||||||
        (Mock(last_command='ls', last_fixed_command=''),
 | 
					 | 
				
			||||||
         ['thefuck', 'fuck'], 'ls')])
 | 
					 | 
				
			||||||
    def test_get_command_script(self, history, args, result):
 | 
					 | 
				
			||||||
        if result:
 | 
					        if result:
 | 
				
			||||||
            assert main.get_command(Mock(), history, args).script == result
 | 
					            assert main.get_command(Mock(env={}), args).script == result
 | 
				
			||||||
        else:
 | 
					        else:
 | 
				
			||||||
            assert main.get_command(Mock(), history, args) is None
 | 
					            assert main.get_command(Mock(env={}), args) is None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class TestGetMatchedRule(object):
 | 
					class TestGetMatchedRule(object):
 | 
				
			||||||
@@ -96,52 +110,41 @@ class TestGetMatchedRule(object):
 | 
				
			|||||||
    def test_when_rule_failed(self, capsys):
 | 
					    def test_when_rule_failed(self, capsys):
 | 
				
			||||||
        main.get_matched_rule(
 | 
					        main.get_matched_rule(
 | 
				
			||||||
            Command('ls'), [Rule('test', Mock(side_effect=OSError('Denied')))],
 | 
					            Command('ls'), [Rule('test', Mock(side_effect=OSError('Denied')))],
 | 
				
			||||||
            Mock(no_colors=True))
 | 
					            Mock(no_colors=True, debug=False))
 | 
				
			||||||
        assert capsys.readouterr()[1].split('\n')[0] == '[WARN] Rule test:'
 | 
					        assert capsys.readouterr()[1].split('\n')[0] == '[WARN] Rule test:'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class TestRunRule(object):
 | 
					class TestRunRule(object):
 | 
				
			||||||
    @pytest.fixture(autouse=True)
 | 
					    @pytest.fixture(autouse=True)
 | 
				
			||||||
    def confirm(self, monkeypatch):
 | 
					    def confirm(self, mocker):
 | 
				
			||||||
        mock = Mock(return_value=True)
 | 
					        return mocker.patch('thefuck.main.confirm', return_value=True)
 | 
				
			||||||
        monkeypatch.setattr('thefuck.main.confirm', mock)
 | 
					 | 
				
			||||||
        return mock
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_run_rule(self, capsys):
 | 
					    def test_run_rule(self, capsys):
 | 
				
			||||||
        main.run_rule(Rule(get_new_command=lambda *_: 'new-command'),
 | 
					        main.run_rule(Rule(get_new_command=lambda *_: 'new-command'),
 | 
				
			||||||
                      Command(), Mock(), None)
 | 
					                      Command(), None)
 | 
				
			||||||
        assert capsys.readouterr() == ('new-command\n', '')
 | 
					        assert capsys.readouterr() == ('new-command\n', '')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_run_rule_with_side_effect(self, capsys):
 | 
					    def test_run_rule_with_side_effect(self, capsys):
 | 
				
			||||||
        side_effect = Mock()
 | 
					        side_effect = Mock()
 | 
				
			||||||
        settings = Mock()
 | 
					        settings = Mock(debug=False)
 | 
				
			||||||
        command = Command()
 | 
					        command = Command()
 | 
				
			||||||
        main.run_rule(Rule(get_new_command=lambda *_: 'new-command',
 | 
					        main.run_rule(Rule(get_new_command=lambda *_: 'new-command',
 | 
				
			||||||
                           side_effect=side_effect),
 | 
					                           side_effect=side_effect),
 | 
				
			||||||
                      command, Mock(), settings)
 | 
					                      command, settings)
 | 
				
			||||||
        assert capsys.readouterr() == ('new-command\n', '')
 | 
					        assert capsys.readouterr() == ('new-command\n', '')
 | 
				
			||||||
        side_effect.assert_called_once_with(command, settings)
 | 
					        side_effect.assert_called_once_with(command, settings)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_hisotry_updated(self):
 | 
					 | 
				
			||||||
        history = Mock()
 | 
					 | 
				
			||||||
        main.run_rule(Rule(get_new_command=lambda *_: 'ls -lah'),
 | 
					 | 
				
			||||||
                      Command('ls'), history, None)
 | 
					 | 
				
			||||||
        history.update.assert_called_once_with(last_command='ls',
 | 
					 | 
				
			||||||
                                               last_fixed_command='ls -lah')
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def test_when_not_comfirmed(self, capsys, confirm):
 | 
					    def test_when_not_comfirmed(self, capsys, confirm):
 | 
				
			||||||
        confirm.return_value = False
 | 
					        confirm.return_value = False
 | 
				
			||||||
        main.run_rule(Rule(get_new_command=lambda *_: 'new-command'),
 | 
					        main.run_rule(Rule(get_new_command=lambda *_: 'new-command'),
 | 
				
			||||||
                      Command(), Mock(), None)
 | 
					                      Command(), None)
 | 
				
			||||||
        assert capsys.readouterr() == ('', '')
 | 
					        assert capsys.readouterr() == ('', '')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class TestConfirm(object):
 | 
					class TestConfirm(object):
 | 
				
			||||||
    @pytest.fixture
 | 
					    @pytest.fixture
 | 
				
			||||||
    def stdin(self, monkeypatch):
 | 
					    def stdin(self, mocker):
 | 
				
			||||||
        mock = Mock(return_value='\n')
 | 
					        return mocker.patch('sys.stdin.read', return_value='\n')
 | 
				
			||||||
        monkeypatch.setattr('sys.stdin.read', mock)
 | 
					 | 
				
			||||||
        return mock
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_when_not_required(self, capsys):
 | 
					    def test_when_not_required(self, capsys):
 | 
				
			||||||
        assert main.confirm('command', None, Mock(require_confirmation=False))
 | 
					        assert main.confirm('command', None, Mock(require_confirmation=False))
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,22 +1,71 @@
 | 
				
			|||||||
import pytest
 | 
					import pytest
 | 
				
			||||||
from mock import Mock
 | 
					 | 
				
			||||||
from thefuck import shells
 | 
					from thefuck import shells
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def builtins_open(mocker):
 | 
				
			||||||
 | 
					    return mocker.patch('six.moves.builtins.open')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					def isfile(mocker):
 | 
				
			||||||
 | 
					    return mocker.patch('os.path.isfile', return_value=True)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.fixture
 | 
				
			||||||
 | 
					@pytest.mark.usefixtures('isfile')
 | 
				
			||||||
 | 
					def history_lines(mocker):
 | 
				
			||||||
 | 
					    def aux(lines):
 | 
				
			||||||
 | 
					        mock = mocker.patch('io.open')
 | 
				
			||||||
 | 
					        mock.return_value.__enter__\
 | 
				
			||||||
 | 
					            .return_value.__iter__.return_value = lines
 | 
				
			||||||
 | 
					    return aux
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class TestGeneric(object):
 | 
					class TestGeneric(object):
 | 
				
			||||||
    def test_from_shell(self):
 | 
					    @pytest.fixture
 | 
				
			||||||
        assert shells.Generic().from_shell('pwd') == 'pwd'
 | 
					    def shell(self):
 | 
				
			||||||
 | 
					        return shells.Generic()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_to_shell(self):
 | 
					    def test_from_shell(self, shell):
 | 
				
			||||||
        assert shells.Bash().to_shell('pwd') == 'pwd'
 | 
					        assert shell.from_shell('pwd') == 'pwd'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_to_shell(self, shell):
 | 
				
			||||||
 | 
					        assert shell.to_shell('pwd') == 'pwd'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_put_to_history(self, builtins_open, shell):
 | 
				
			||||||
 | 
					        assert shell.put_to_history('ls') is None
 | 
				
			||||||
 | 
					        assert builtins_open.call_count == 0
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_and_(self, shell):
 | 
				
			||||||
 | 
					        assert shell.and_('ls', 'cd') == 'ls && cd'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_get_aliases(self, shell):
 | 
				
			||||||
 | 
					        assert shell.get_aliases() == {}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_app_alias(self, shell):
 | 
				
			||||||
 | 
					        assert 'alias fuck' in shell.app_alias()
 | 
				
			||||||
 | 
					        assert 'thefuck' in shell.app_alias()
 | 
				
			||||||
 | 
					        assert 'TF_ALIAS' in shell.app_alias()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_get_history(self, history_lines, shell):
 | 
				
			||||||
 | 
					        history_lines(['ls', 'rm'])
 | 
				
			||||||
 | 
					        # We don't know what to do in generic shell with history lines,
 | 
				
			||||||
 | 
					        # so just ignore them:
 | 
				
			||||||
 | 
					        assert list(shell.get_history()) == []
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.usefixtures('isfile')
 | 
				
			||||||
class TestBash(object):
 | 
					class TestBash(object):
 | 
				
			||||||
 | 
					    @pytest.fixture
 | 
				
			||||||
 | 
					    def shell(self):
 | 
				
			||||||
 | 
					        return shells.Bash()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    @pytest.fixture(autouse=True)
 | 
					    @pytest.fixture(autouse=True)
 | 
				
			||||||
    def Popen(self, monkeypatch):
 | 
					    def Popen(self, mocker):
 | 
				
			||||||
        mock = Mock()
 | 
					        mock = mocker.patch('thefuck.shells.Popen')
 | 
				
			||||||
        monkeypatch.setattr('thefuck.shells.Popen', mock)
 | 
					 | 
				
			||||||
        mock.return_value.stdout.read.return_value = (
 | 
					        mock.return_value.stdout.read.return_value = (
 | 
				
			||||||
 | 
					            b'alias fuck=\'eval $(thefuck $(fc -ln -1))\'\n'
 | 
				
			||||||
            b'alias l=\'ls -CF\'\n'
 | 
					            b'alias l=\'ls -CF\'\n'
 | 
				
			||||||
            b'alias la=\'ls -A\'\n'
 | 
					            b'alias la=\'ls -A\'\n'
 | 
				
			||||||
            b'alias ll=\'ls -alF\'')
 | 
					            b'alias ll=\'ls -alF\'')
 | 
				
			||||||
@@ -24,30 +73,159 @@ class TestBash(object):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    @pytest.mark.parametrize('before, after', [
 | 
					    @pytest.mark.parametrize('before, after', [
 | 
				
			||||||
        ('pwd', 'pwd'),
 | 
					        ('pwd', 'pwd'),
 | 
				
			||||||
 | 
					        ('fuck', 'eval $(thefuck $(fc -ln -1))'),
 | 
				
			||||||
 | 
					        ('awk', 'awk'),
 | 
				
			||||||
        ('ll', 'ls -alF')])
 | 
					        ('ll', 'ls -alF')])
 | 
				
			||||||
    def test_from_shell(self, before, after):
 | 
					    def test_from_shell(self, before, after, shell):
 | 
				
			||||||
        assert shells.Bash().from_shell(before) == after
 | 
					        assert shell.from_shell(before) == after
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_to_shell(self):
 | 
					    def test_to_shell(self, shell):
 | 
				
			||||||
        assert shells.Bash().to_shell('pwd') == 'pwd'
 | 
					        assert shell.to_shell('pwd') == 'pwd'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_put_to_history(self, builtins_open, shell):
 | 
				
			||||||
 | 
					        shell.put_to_history('ls')
 | 
				
			||||||
 | 
					        builtins_open.return_value.__enter__.return_value. \
 | 
				
			||||||
 | 
					            write.assert_called_once_with('ls\n')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_and_(self, shell):
 | 
				
			||||||
 | 
					        assert shell.and_('ls', 'cd') == 'ls && cd'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_get_aliases(self, shell):
 | 
				
			||||||
 | 
					        assert shell.get_aliases() == {'fuck': 'eval $(thefuck $(fc -ln -1))',
 | 
				
			||||||
 | 
					                                       'l': 'ls -CF',
 | 
				
			||||||
 | 
					                                       'la': 'ls -A',
 | 
				
			||||||
 | 
					                                       'll': 'ls -alF'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_app_alias(self, shell):
 | 
				
			||||||
 | 
					        assert 'alias fuck' in shell.app_alias()
 | 
				
			||||||
 | 
					        assert 'thefuck' in shell.app_alias()
 | 
				
			||||||
 | 
					        assert 'TF_ALIAS' in shell.app_alias()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_get_history(self, history_lines, shell):
 | 
				
			||||||
 | 
					        history_lines(['ls', 'rm'])
 | 
				
			||||||
 | 
					        assert list(shell.get_history()) == ['ls', 'rm']
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class TestZsh(object):
 | 
					@pytest.mark.usefixtures('isfile')
 | 
				
			||||||
 | 
					class TestFish(object):
 | 
				
			||||||
 | 
					    @pytest.fixture
 | 
				
			||||||
 | 
					    def shell(self):
 | 
				
			||||||
 | 
					        return shells.Fish()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    @pytest.fixture(autouse=True)
 | 
					    @pytest.fixture(autouse=True)
 | 
				
			||||||
    def Popen(self, monkeypatch):
 | 
					    def Popen(self, mocker):
 | 
				
			||||||
        mock = Mock()
 | 
					        mock = mocker.patch('thefuck.shells.Popen')
 | 
				
			||||||
        monkeypatch.setattr('thefuck.shells.Popen', mock)
 | 
					 | 
				
			||||||
        mock.return_value.stdout.read.return_value = (
 | 
					        mock.return_value.stdout.read.return_value = (
 | 
				
			||||||
 | 
					            b'cd\nfish_config\nfuck\nfunced\nfuncsave\ngrep\nhistory\nll\nls\n'
 | 
				
			||||||
 | 
					            b'man\nmath\npopd\npushd\nruby')
 | 
				
			||||||
 | 
					        return mock
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    @pytest.fixture
 | 
				
			||||||
 | 
					    def environ(self, monkeypatch):
 | 
				
			||||||
 | 
					        data = {'TF_OVERRIDDEN_ALIASES': 'cd, ls, man, open'}
 | 
				
			||||||
 | 
					        monkeypatch.setattr('thefuck.shells.os.environ', data)
 | 
				
			||||||
 | 
					        return data
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    @pytest.mark.usefixture('environ')
 | 
				
			||||||
 | 
					    def test_get_overridden_aliases(self, shell, environ):
 | 
				
			||||||
 | 
					        assert shell._get_overridden_aliases() == ['cd', 'ls', 'man', 'open']
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    @pytest.mark.parametrize('before, after', [
 | 
				
			||||||
 | 
					        ('cd', 'cd'),
 | 
				
			||||||
 | 
					        ('pwd', 'pwd'),
 | 
				
			||||||
 | 
					        ('fuck', 'fish -ic "fuck"'),
 | 
				
			||||||
 | 
					        ('find', 'find'),
 | 
				
			||||||
 | 
					        ('funced', 'fish -ic "funced"'),
 | 
				
			||||||
 | 
					        ('grep', 'grep'),
 | 
				
			||||||
 | 
					        ('awk', 'awk'),
 | 
				
			||||||
 | 
					        ('math "2 + 2"', r'fish -ic "math \"2 + 2\""'),
 | 
				
			||||||
 | 
					        ('man', 'man'),
 | 
				
			||||||
 | 
					        ('open', 'open'),
 | 
				
			||||||
 | 
					        ('vim', 'vim'),
 | 
				
			||||||
 | 
					        ('ll', 'fish -ic "ll"'),
 | 
				
			||||||
 | 
					        ('ls', 'ls')])  # Fish has no aliases but functions
 | 
				
			||||||
 | 
					    def test_from_shell(self, before, after, shell):
 | 
				
			||||||
 | 
					        assert shell.from_shell(before) == after
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_to_shell(self, shell):
 | 
				
			||||||
 | 
					        assert shell.to_shell('pwd') == 'pwd'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_put_to_history(self, builtins_open, mocker, shell):
 | 
				
			||||||
 | 
					        mocker.patch('thefuck.shells.time',
 | 
				
			||||||
 | 
					                     return_value=1430707243.3517463)
 | 
				
			||||||
 | 
					        shell.put_to_history('ls')
 | 
				
			||||||
 | 
					        builtins_open.return_value.__enter__.return_value. \
 | 
				
			||||||
 | 
					            write.assert_called_once_with('- cmd: ls\n   when: 1430707243\n')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_and_(self, shell):
 | 
				
			||||||
 | 
					        assert shell.and_('foo', 'bar') == 'foo; and bar'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_get_aliases(self, shell):
 | 
				
			||||||
 | 
					        assert shell.get_aliases() == {'fish_config': 'fish_config',
 | 
				
			||||||
 | 
					                                       'fuck': 'fuck',
 | 
				
			||||||
 | 
					                                       'funced': 'funced',
 | 
				
			||||||
 | 
					                                       'funcsave': 'funcsave',
 | 
				
			||||||
 | 
					                                       'history': 'history',
 | 
				
			||||||
 | 
					                                       'll': 'll',
 | 
				
			||||||
 | 
					                                       'math': 'math',
 | 
				
			||||||
 | 
					                                       'popd': 'popd',
 | 
				
			||||||
 | 
					                                       'pushd': 'pushd',
 | 
				
			||||||
 | 
					                                       'ruby': 'ruby'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_app_alias(self, shell):
 | 
				
			||||||
 | 
					        assert 'function fuck' in shell.app_alias()
 | 
				
			||||||
 | 
					        assert 'thefuck' in shell.app_alias()
 | 
				
			||||||
 | 
					        assert 'TF_ALIAS' in shell.app_alias()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.usefixtures('isfile')
 | 
				
			||||||
 | 
					class TestZsh(object):
 | 
				
			||||||
 | 
					    @pytest.fixture
 | 
				
			||||||
 | 
					    def shell(self):
 | 
				
			||||||
 | 
					        return shells.Zsh()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    @pytest.fixture(autouse=True)
 | 
				
			||||||
 | 
					    def Popen(self, mocker):
 | 
				
			||||||
 | 
					        mock = mocker.patch('thefuck.shells.Popen')
 | 
				
			||||||
 | 
					        mock.return_value.stdout.read.return_value = (
 | 
				
			||||||
 | 
					            b'fuck=\'eval $(thefuck $(fc -ln -1 | tail -n 1))\'\n'
 | 
				
			||||||
            b'l=\'ls -CF\'\n'
 | 
					            b'l=\'ls -CF\'\n'
 | 
				
			||||||
            b'la=\'ls -A\'\n'
 | 
					            b'la=\'ls -A\'\n'
 | 
				
			||||||
            b'll=\'ls -alF\'')
 | 
					            b'll=\'ls -alF\'')
 | 
				
			||||||
        return mock
 | 
					        return mock
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    @pytest.mark.parametrize('before, after', [
 | 
					    @pytest.mark.parametrize('before, after', [
 | 
				
			||||||
 | 
					        ('fuck', 'eval $(thefuck $(fc -ln -1 | tail -n 1))'),
 | 
				
			||||||
        ('pwd', 'pwd'),
 | 
					        ('pwd', 'pwd'),
 | 
				
			||||||
        ('ll', 'ls -alF')])
 | 
					        ('ll', 'ls -alF')])
 | 
				
			||||||
    def test_from_shell(self, before, after):
 | 
					    def test_from_shell(self, before, after, shell):
 | 
				
			||||||
        assert shells.Zsh().from_shell(before) == after
 | 
					        assert shell.from_shell(before) == after
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_to_shell(self):
 | 
					    def test_to_shell(self, shell):
 | 
				
			||||||
        assert shells.Zsh().to_shell('pwd') == 'pwd'
 | 
					        assert shell.to_shell('pwd') == 'pwd'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_put_to_history(self, builtins_open, mocker, shell):
 | 
				
			||||||
 | 
					        mocker.patch('thefuck.shells.time',
 | 
				
			||||||
 | 
					                     return_value=1430707243.3517463)
 | 
				
			||||||
 | 
					        shell.put_to_history('ls')
 | 
				
			||||||
 | 
					        builtins_open.return_value.__enter__.return_value. \
 | 
				
			||||||
 | 
					            write.assert_called_once_with(': 1430707243:0;ls\n')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_and_(self, shell):
 | 
				
			||||||
 | 
					        assert shell.and_('ls', 'cd') == 'ls && cd'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_get_aliases(self, shell):
 | 
				
			||||||
 | 
					        assert shell.get_aliases() == {
 | 
				
			||||||
 | 
					            'fuck': 'eval $(thefuck $(fc -ln -1 | tail -n 1))',
 | 
				
			||||||
 | 
					            'l': 'ls -CF',
 | 
				
			||||||
 | 
					            'la': 'ls -A',
 | 
				
			||||||
 | 
					            'll': 'ls -alF'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_app_alias(self, shell):
 | 
				
			||||||
 | 
					        assert 'alias fuck' in shell.app_alias()
 | 
				
			||||||
 | 
					        assert 'thefuck' in shell.app_alias()
 | 
				
			||||||
 | 
					        assert 'TF_ALIAS' in shell.app_alias()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_get_history(self, history_lines, shell):
 | 
				
			||||||
 | 
					        history_lines([': 1432613911:0;ls', ': 1432613916:0;rm'])
 | 
				
			||||||
 | 
					        assert list(shell.get_history()) == ['ls', 'rm']
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,6 +1,6 @@
 | 
				
			|||||||
import pytest
 | 
					import pytest
 | 
				
			||||||
from mock import Mock
 | 
					from mock import Mock
 | 
				
			||||||
from thefuck.utils import sudo_support, wrap_settings
 | 
					from thefuck.utils import git_support, sudo_support, wrap_settings, memoize, get_closest
 | 
				
			||||||
from thefuck.types import Settings
 | 
					from thefuck.types import Settings
 | 
				
			||||||
from tests.utils import Command
 | 
					from tests.utils import Command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -24,3 +24,38 @@ def test_sudo_support(return_value, command, called, result):
 | 
				
			|||||||
    fn = Mock(return_value=return_value, __name__='')
 | 
					    fn = Mock(return_value=return_value, __name__='')
 | 
				
			||||||
    assert sudo_support(fn)(Command(command), None) == result
 | 
					    assert sudo_support(fn)(Command(command), None) == result
 | 
				
			||||||
    fn.assert_called_once_with(Command(called), None)
 | 
					    fn.assert_called_once_with(Command(called), None)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.parametrize('called, command, stderr', [
 | 
				
			||||||
 | 
					    ('git co', 'git checkout', "19:22:36.299340 git.c:282   trace: alias expansion: co => 'checkout'"),
 | 
				
			||||||
 | 
					    ('git com file', 'git commit --verbose file', "19:23:25.470911 git.c:282   trace: alias expansion: com => 'commit' '--verbose'")])
 | 
				
			||||||
 | 
					def test_git_support(called, command, stderr):
 | 
				
			||||||
 | 
					    @git_support
 | 
				
			||||||
 | 
					    def fn(command, settings): return command.script
 | 
				
			||||||
 | 
					    assert fn(Command(script=called, stderr=stderr), None) == command
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def test_memoize():
 | 
				
			||||||
 | 
					    fn = Mock(__name__='fn')
 | 
				
			||||||
 | 
					    memoized = memoize(fn)
 | 
				
			||||||
 | 
					    memoized()
 | 
				
			||||||
 | 
					    memoized()
 | 
				
			||||||
 | 
					    fn.assert_called_once_with()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@pytest.mark.usefixtures('no_memoize')
 | 
				
			||||||
 | 
					def test_no_memoize():
 | 
				
			||||||
 | 
					    fn = Mock(__name__='fn')
 | 
				
			||||||
 | 
					    memoized = memoize(fn)
 | 
				
			||||||
 | 
					    memoized()
 | 
				
			||||||
 | 
					    memoized()
 | 
				
			||||||
 | 
					    assert fn.call_count == 2
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					class TestGetClosest(object):
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_when_can_match(self):
 | 
				
			||||||
 | 
					        assert 'branch' == get_closest('brnch', ['branch', 'status'])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_when_cant_match(self):
 | 
				
			||||||
 | 
					        assert 'status' == get_closest('st', ['status', 'reset'])
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,4 +1,5 @@
 | 
				
			|||||||
from thefuck import types
 | 
					from thefuck import types
 | 
				
			||||||
 | 
					from thefuck.conf import DEFAULT_PRIORITY
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def Command(script='', stdout='', stderr=''):
 | 
					def Command(script='', stdout='', stderr=''):
 | 
				
			||||||
@@ -8,6 +9,8 @@ def Command(script='', stdout='', stderr=''):
 | 
				
			|||||||
def Rule(name='', match=lambda *_: True,
 | 
					def Rule(name='', match=lambda *_: True,
 | 
				
			||||||
         get_new_command=lambda *_: '',
 | 
					         get_new_command=lambda *_: '',
 | 
				
			||||||
         enabled_by_default=True,
 | 
					         enabled_by_default=True,
 | 
				
			||||||
         side_effect=None):
 | 
					         side_effect=None,
 | 
				
			||||||
 | 
					         priority=DEFAULT_PRIORITY):
 | 
				
			||||||
    return types.Rule(name, match, get_new_command,
 | 
					    return types.Rule(name, match, get_new_command,
 | 
				
			||||||
                      enabled_by_default, side_effect)
 | 
					                      enabled_by_default, side_effect,
 | 
				
			||||||
 | 
					                      priority)
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -22,17 +22,23 @@ class _DefaultRulesNames(types.RulesNamesList):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
DEFAULT_RULES = _DefaultRulesNames([])
 | 
					DEFAULT_RULES = _DefaultRulesNames([])
 | 
				
			||||||
 | 
					DEFAULT_PRIORITY = 1000
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
DEFAULT_SETTINGS = {'rules': DEFAULT_RULES,
 | 
					DEFAULT_SETTINGS = {'rules': DEFAULT_RULES,
 | 
				
			||||||
                    'wait_command': 3,
 | 
					                    'wait_command': 3,
 | 
				
			||||||
                    'require_confirmation': False,
 | 
					                    'require_confirmation': True,
 | 
				
			||||||
                    'no_colors': False}
 | 
					                    'no_colors': False,
 | 
				
			||||||
 | 
					                    'debug': False,
 | 
				
			||||||
 | 
					                    'priority': {},
 | 
				
			||||||
 | 
					                    'env': {'LANG': 'C', 'GIT_TRACE': '1'}}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
ENV_TO_ATTR = {'THEFUCK_RULES': 'rules',
 | 
					ENV_TO_ATTR = {'THEFUCK_RULES': 'rules',
 | 
				
			||||||
               'THEFUCK_WAIT_COMMAND': 'wait_command',
 | 
					               'THEFUCK_WAIT_COMMAND': 'wait_command',
 | 
				
			||||||
               'THEFUCK_REQUIRE_CONFIRMATION': 'require_confirmation',
 | 
					               'THEFUCK_REQUIRE_CONFIRMATION': 'require_confirmation',
 | 
				
			||||||
               'THEFUCK_NO_COLORS': 'no_colors'}
 | 
					               'THEFUCK_NO_COLORS': 'no_colors',
 | 
				
			||||||
 | 
					               'THEFUCK_PRIORITY': 'priority',
 | 
				
			||||||
 | 
					               'THEFUCK_DEBUG': 'debug'}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
SETTINGS_HEADER = u"""# ~/.thefuck/settings.py: The Fuck settings file
 | 
					SETTINGS_HEADER = u"""# ~/.thefuck/settings.py: The Fuck settings file
 | 
				
			||||||
@@ -65,16 +71,29 @@ def _rules_from_env(val):
 | 
				
			|||||||
    return val
 | 
					    return val
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def _priority_from_env(val):
 | 
				
			||||||
 | 
					    """Gets priority pairs from env."""
 | 
				
			||||||
 | 
					    for part in val.split(':'):
 | 
				
			||||||
 | 
					        try:
 | 
				
			||||||
 | 
					            rule, priority = part.split('=')
 | 
				
			||||||
 | 
					            yield rule, int(priority)
 | 
				
			||||||
 | 
					        except ValueError:
 | 
				
			||||||
 | 
					            continue
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def _val_from_env(env, attr):
 | 
					def _val_from_env(env, attr):
 | 
				
			||||||
    """Transforms env-strings to python."""
 | 
					    """Transforms env-strings to python."""
 | 
				
			||||||
    val = os.environ[env]
 | 
					    val = os.environ[env]
 | 
				
			||||||
    if attr == 'rules':
 | 
					    if attr == 'rules':
 | 
				
			||||||
        val = _rules_from_env(val)
 | 
					        return _rules_from_env(val)
 | 
				
			||||||
 | 
					    elif attr == 'priority':
 | 
				
			||||||
 | 
					        return dict(_priority_from_env(val))
 | 
				
			||||||
    elif attr == 'wait_command':
 | 
					    elif attr == 'wait_command':
 | 
				
			||||||
        val = int(val)
 | 
					        return int(val)
 | 
				
			||||||
    elif attr in ('require_confirmation', 'no_colors'):
 | 
					    elif attr in ('require_confirmation', 'no_colors', 'debug'):
 | 
				
			||||||
        val = val.lower() == 'true'
 | 
					        return val.lower() == 'true'
 | 
				
			||||||
    return val
 | 
					    else:
 | 
				
			||||||
 | 
					        return val
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def _settings_from_env():
 | 
					def _settings_from_env():
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,27 +0,0 @@
 | 
				
			|||||||
import os
 | 
					 | 
				
			||||||
import shelve
 | 
					 | 
				
			||||||
from tempfile import gettempdir
 | 
					 | 
				
			||||||
from psutil import Process
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
class History(object):
 | 
					 | 
				
			||||||
    """Temporary history of commands/fixed-commands dependent on
 | 
					 | 
				
			||||||
    current shell instance.
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def __init__(self):
 | 
					 | 
				
			||||||
        self._path = os.path.join(gettempdir(), '.thefuck_history')
 | 
					 | 
				
			||||||
        self._pid = Process(os.getpid()).parent().pid
 | 
					 | 
				
			||||||
        self._db = shelve.open(self._path)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def _prepare_key(self, key):
 | 
					 | 
				
			||||||
        return '{}-{}'.format(self._pid, key)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def update(self, **kwargs):
 | 
					 | 
				
			||||||
        self._db.update({self._prepare_key(k): v for k,v in kwargs.items()})
 | 
					 | 
				
			||||||
        self._db.sync()
 | 
					 | 
				
			||||||
        return self
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def __getattr__(self, item):
 | 
					 | 
				
			||||||
        return self._db.get(self._prepare_key(item))
 | 
					 | 
				
			||||||
@@ -1,3 +1,4 @@
 | 
				
			|||||||
 | 
					from pprint import pformat
 | 
				
			||||||
import sys
 | 
					import sys
 | 
				
			||||||
from traceback import format_exception
 | 
					from traceback import format_exception
 | 
				
			||||||
import colorama
 | 
					import colorama
 | 
				
			||||||
@@ -52,3 +53,12 @@ def failed(msg, settings):
 | 
				
			|||||||
        msg=msg,
 | 
					        msg=msg,
 | 
				
			||||||
        red=color(colorama.Fore.RED, settings),
 | 
					        red=color(colorama.Fore.RED, settings),
 | 
				
			||||||
        reset=color(colorama.Style.RESET_ALL, settings)))
 | 
					        reset=color(colorama.Style.RESET_ALL, settings)))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def debug(msg, settings):
 | 
				
			||||||
 | 
					    if settings.debug:
 | 
				
			||||||
 | 
					        sys.stderr.write(u'{blue}{bold}DEBUG:{reset} {msg}\n'.format(
 | 
				
			||||||
 | 
					            msg=msg,
 | 
				
			||||||
 | 
					            reset=color(colorama.Style.RESET_ALL, settings),
 | 
				
			||||||
 | 
					            blue=color(colorama.Fore.BLUE, settings),
 | 
				
			||||||
 | 
					            bold=color(colorama.Style.BRIGHT, settings)))
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,12 +1,13 @@
 | 
				
			|||||||
from imp import load_source
 | 
					from imp import load_source
 | 
				
			||||||
from pathlib import Path
 | 
					from pathlib import Path
 | 
				
			||||||
from os.path import expanduser
 | 
					from os.path import expanduser
 | 
				
			||||||
 | 
					from pprint import pformat
 | 
				
			||||||
from subprocess import Popen, PIPE
 | 
					from subprocess import Popen, PIPE
 | 
				
			||||||
import os
 | 
					import os
 | 
				
			||||||
import sys
 | 
					import sys
 | 
				
			||||||
from psutil import Process, TimeoutExpired
 | 
					from psutil import Process, TimeoutExpired
 | 
				
			||||||
import colorama
 | 
					import colorama
 | 
				
			||||||
from .history import History
 | 
					import six
 | 
				
			||||||
from . import logs, conf, types, shells
 | 
					from . import logs, conf, types, shells
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -26,7 +27,17 @@ def load_rule(rule):
 | 
				
			|||||||
    return types.Rule(rule.name[:-3], rule_module.match,
 | 
					    return types.Rule(rule.name[:-3], rule_module.match,
 | 
				
			||||||
                      rule_module.get_new_command,
 | 
					                      rule_module.get_new_command,
 | 
				
			||||||
                      getattr(rule_module, 'enabled_by_default', True),
 | 
					                      getattr(rule_module, 'enabled_by_default', True),
 | 
				
			||||||
                      getattr(rule_module, 'side_effect', None))
 | 
					                      getattr(rule_module, 'side_effect', None),
 | 
				
			||||||
 | 
					                      getattr(rule_module, 'priority', conf.DEFAULT_PRIORITY))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def _get_loaded_rules(rules, settings):
 | 
				
			||||||
 | 
					    """Yields all available rules."""
 | 
				
			||||||
 | 
					    for rule in rules:
 | 
				
			||||||
 | 
					        if rule.name != '__init__.py':
 | 
				
			||||||
 | 
					            loaded_rule = load_rule(rule)
 | 
				
			||||||
 | 
					            if loaded_rule in settings.rules:
 | 
				
			||||||
 | 
					                yield loaded_rule
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def get_rules(user_dir, settings):
 | 
					def get_rules(user_dir, settings):
 | 
				
			||||||
@@ -35,11 +46,9 @@ def get_rules(user_dir, settings):
 | 
				
			|||||||
        .joinpath('rules') \
 | 
					        .joinpath('rules') \
 | 
				
			||||||
        .glob('*.py')
 | 
					        .glob('*.py')
 | 
				
			||||||
    user = user_dir.joinpath('rules').glob('*.py')
 | 
					    user = user_dir.joinpath('rules').glob('*.py')
 | 
				
			||||||
    for rule in sorted(list(bundled)) + list(user):
 | 
					    rules = _get_loaded_rules(sorted(bundled) + sorted(user), settings)
 | 
				
			||||||
        if rule.name != '__init__.py':
 | 
					    return sorted(rules, key=lambda rule: settings.priority.get(
 | 
				
			||||||
            loaded_rule = load_rule(rule)
 | 
					        rule.name, rule.priority))
 | 
				
			||||||
            if loaded_rule in settings.rules:
 | 
					 | 
				
			||||||
                yield loaded_rule
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def wait_output(settings, popen):
 | 
					def wait_output(settings, popen):
 | 
				
			||||||
@@ -54,30 +63,30 @@ def wait_output(settings, popen):
 | 
				
			|||||||
        proc.wait(settings.wait_command)
 | 
					        proc.wait(settings.wait_command)
 | 
				
			||||||
        return True
 | 
					        return True
 | 
				
			||||||
    except TimeoutExpired:
 | 
					    except TimeoutExpired:
 | 
				
			||||||
        for child in proc.get_children(recursive=True):
 | 
					        for child in proc.children(recursive=True):
 | 
				
			||||||
            child.kill()
 | 
					            child.kill()
 | 
				
			||||||
        proc.kill()
 | 
					        proc.kill()
 | 
				
			||||||
        return False
 | 
					        return False
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def get_command(settings, history, args):
 | 
					def get_command(settings, args):
 | 
				
			||||||
    """Creates command from `args` and executes it."""
 | 
					    """Creates command from `args` and executes it."""
 | 
				
			||||||
    if sys.version_info[0] < 3:
 | 
					    if six.PY2:
 | 
				
			||||||
        script = ' '.join(arg.decode('utf-8') for arg in args[1:])
 | 
					        script = ' '.join(arg.decode('utf-8') for arg in args[1:])
 | 
				
			||||||
    else:
 | 
					    else:
 | 
				
			||||||
        script = ' '.join(args[1:])
 | 
					        script = ' '.join(args[1:])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    if script == 'fuck' or script == history.last_command:
 | 
					 | 
				
			||||||
        script = history.last_fixed_command or history.last_command
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if not script:
 | 
					    if not script:
 | 
				
			||||||
        return
 | 
					        return
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    script = shells.from_shell(script)
 | 
					    script = shells.from_shell(script)
 | 
				
			||||||
    history.update(last_command=script,
 | 
					    logs.debug('Call: {}'.format(script), settings)
 | 
				
			||||||
                   last_fixed_command=None)
 | 
					
 | 
				
			||||||
    result = Popen(script, shell=True, stdout=PIPE, stderr=PIPE,
 | 
					    env = dict(os.environ)
 | 
				
			||||||
                   env=dict(os.environ, LANG='C'))
 | 
					    env.update(settings.env)
 | 
				
			||||||
 | 
					    logs.debug('Executing with env: {}'.format(env), settings)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    result = Popen(script, shell=True, stdout=PIPE, stderr=PIPE, env=env)
 | 
				
			||||||
    if wait_output(settings, result):
 | 
					    if wait_output(settings, result):
 | 
				
			||||||
        return types.Command(script, result.stdout.read().decode('utf-8'),
 | 
					        return types.Command(script, result.stdout.read().decode('utf-8'),
 | 
				
			||||||
                             result.stderr.read().decode('utf-8'))
 | 
					                             result.stderr.read().decode('utf-8'))
 | 
				
			||||||
@@ -87,6 +96,7 @@ def get_matched_rule(command, rules, settings):
 | 
				
			|||||||
    """Returns first matched rule for command."""
 | 
					    """Returns first matched rule for command."""
 | 
				
			||||||
    for rule in rules:
 | 
					    for rule in rules:
 | 
				
			||||||
        try:
 | 
					        try:
 | 
				
			||||||
 | 
					            logs.debug(u'Trying rule: {}'.format(rule.name), settings)
 | 
				
			||||||
            if rule.match(command, settings):
 | 
					            if rule.match(command, settings):
 | 
				
			||||||
                return rule
 | 
					                return rule
 | 
				
			||||||
        except Exception:
 | 
					        except Exception:
 | 
				
			||||||
@@ -108,33 +118,36 @@ def confirm(new_command, side_effect, settings):
 | 
				
			|||||||
        return False
 | 
					        return False
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def run_rule(rule, command, history, settings):
 | 
					def run_rule(rule, command, settings):
 | 
				
			||||||
    """Runs command from rule for passed command."""
 | 
					    """Runs command from rule for passed command."""
 | 
				
			||||||
    new_command = shells.to_shell(rule.get_new_command(command, settings))
 | 
					    new_command = shells.to_shell(rule.get_new_command(command, settings))
 | 
				
			||||||
    if confirm(new_command, rule.side_effect, settings):
 | 
					    if confirm(new_command, rule.side_effect, settings):
 | 
				
			||||||
        if rule.side_effect:
 | 
					        if rule.side_effect:
 | 
				
			||||||
            rule.side_effect(command, settings)
 | 
					            rule.side_effect(command, settings)
 | 
				
			||||||
        history.update(last_command=command.script,
 | 
					        shells.put_to_history(new_command)
 | 
				
			||||||
                       last_fixed_command=new_command)
 | 
					 | 
				
			||||||
        print(new_command)
 | 
					        print(new_command)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def alias():
 | 
					 | 
				
			||||||
    print(shells.app_alias())
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def main():
 | 
					def main():
 | 
				
			||||||
    colorama.init()
 | 
					    colorama.init()
 | 
				
			||||||
    user_dir = setup_user_dir()
 | 
					    user_dir = setup_user_dir()
 | 
				
			||||||
    settings = conf.get_settings(user_dir)
 | 
					    settings = conf.get_settings(user_dir)
 | 
				
			||||||
    history = History()
 | 
					    logs.debug('Run with settings: {}'.format(pformat(settings)), settings)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    command = get_command(settings, history, sys.argv)
 | 
					    command = get_command(settings, sys.argv)
 | 
				
			||||||
    if command:
 | 
					    if command:
 | 
				
			||||||
 | 
					        logs.debug('Received stdout: {}'.format(command.stdout), settings)
 | 
				
			||||||
 | 
					        logs.debug('Received stderr: {}'.format(command.stderr), settings)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        rules = get_rules(user_dir, settings)
 | 
					        rules = get_rules(user_dir, settings)
 | 
				
			||||||
 | 
					        logs.debug(
 | 
				
			||||||
 | 
					            'Loaded rules: {}'.format(', '.join(rule.name for rule in rules)),
 | 
				
			||||||
 | 
					            settings)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        matched_rule = get_matched_rule(command, rules, settings)
 | 
					        matched_rule = get_matched_rule(command, rules, settings)
 | 
				
			||||||
        if matched_rule:
 | 
					        if matched_rule:
 | 
				
			||||||
            run_rule(matched_rule, command, history, settings)
 | 
					            logs.debug('Matched rule: {}'.format(matched_rule.name), settings)
 | 
				
			||||||
 | 
					            run_rule(matched_rule, command, settings)
 | 
				
			||||||
            return
 | 
					            return
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    logs.failed('No fuck given', settings)
 | 
					    logs.failed('No fuck given', settings)
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,9 +1,12 @@
 | 
				
			|||||||
 | 
					from thefuck import shells
 | 
				
			||||||
 | 
					from thefuck.utils import sudo_support
 | 
				
			||||||
 | 
					
 | 
				
			||||||
try:
 | 
					try:
 | 
				
			||||||
    import CommandNotFound
 | 
					    import CommandNotFound
 | 
				
			||||||
except ImportError:
 | 
					except ImportError:
 | 
				
			||||||
    enabled_by_default = False
 | 
					    enabled_by_default = False
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@sudo_support
 | 
				
			||||||
def match(command, settings):
 | 
					def match(command, settings):
 | 
				
			||||||
    if 'not found' in command.stderr:
 | 
					    if 'not found' in command.stderr:
 | 
				
			||||||
        try:
 | 
					        try:
 | 
				
			||||||
@@ -15,9 +18,10 @@ def match(command, settings):
 | 
				
			|||||||
            # IndexError is thrown when no matching package is found
 | 
					            # IndexError is thrown when no matching package is found
 | 
				
			||||||
            return False
 | 
					            return False
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@sudo_support
 | 
				
			||||||
def get_new_command(command, settings):
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
    c = CommandNotFound.CommandNotFound()
 | 
					    c = CommandNotFound.CommandNotFound()
 | 
				
			||||||
    pkgs = c.getPackages(command.script.split(" ")[0])
 | 
					    pkgs = c.getPackages(command.script.split(" ")[0])
 | 
				
			||||||
    name, _ = pkgs[0]
 | 
					    name, _ = pkgs[0]
 | 
				
			||||||
    return "sudo apt-get install {} && {}".format(name, command.script)
 | 
					    formatme = shells.and_('sudo apt-get install {}', '{}')
 | 
				
			||||||
 | 
					    return formatme.format(name, command.script)
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,14 +1,14 @@
 | 
				
			|||||||
import difflib
 | 
					 | 
				
			||||||
import os
 | 
					import os
 | 
				
			||||||
import re
 | 
					import re
 | 
				
			||||||
from subprocess import check_output
 | 
					from subprocess import check_output
 | 
				
			||||||
 | 
					from thefuck.utils import get_closest
 | 
				
			||||||
import thefuck.logs
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
# Formulars are base on each local system's status
 | 
					# Formulars are base on each local system's status
 | 
				
			||||||
 | 
					
 | 
				
			||||||
brew_formulas = []
 | 
					brew_formulas = []
 | 
				
			||||||
try:
 | 
					try:
 | 
				
			||||||
    brew_path_prefix = check_output(['brew', '--prefix']).strip()
 | 
					    brew_path_prefix = check_output(['brew', '--prefix'],
 | 
				
			||||||
 | 
					                                    universal_newlines=True).strip()
 | 
				
			||||||
    brew_formula_path = brew_path_prefix + '/Library/Formula'
 | 
					    brew_formula_path = brew_path_prefix + '/Library/Formula'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    for file_name in os.listdir(brew_formula_path):
 | 
					    for file_name in os.listdir(brew_formula_path):
 | 
				
			||||||
@@ -18,8 +18,8 @@ except:
 | 
				
			|||||||
    pass
 | 
					    pass
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def _get_similar_formulars(formula_name):
 | 
					def _get_similar_formula(formula_name):
 | 
				
			||||||
    return difflib.get_close_matches(formula_name, brew_formulas, 1, 0.85)
 | 
					    return get_closest(formula_name, brew_formulas, 1, 0.85)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def match(command, settings):
 | 
					def match(command, settings):
 | 
				
			||||||
@@ -30,7 +30,7 @@ def match(command, settings):
 | 
				
			|||||||
    if is_proper_command:
 | 
					    if is_proper_command:
 | 
				
			||||||
        formula = re.findall(r'Error: No available formula for ([a-z]+)',
 | 
					        formula = re.findall(r'Error: No available formula for ([a-z]+)',
 | 
				
			||||||
                             command.stderr)[0]
 | 
					                             command.stderr)[0]
 | 
				
			||||||
        has_possible_formulas = len(_get_similar_formulars(formula)) > 0
 | 
					        has_possible_formulas = bool(_get_similar_formula(formula))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    return has_possible_formulas
 | 
					    return has_possible_formulas
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -38,6 +38,6 @@ def match(command, settings):
 | 
				
			|||||||
def get_new_command(command, settings):
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
    not_exist_formula = re.findall(r'Error: No available formula for ([a-z]+)',
 | 
					    not_exist_formula = re.findall(r'Error: No available formula for ([a-z]+)',
 | 
				
			||||||
                                   command.stderr)[0]
 | 
					                                   command.stderr)[0]
 | 
				
			||||||
    exist_formula = _get_similar_formulars(not_exist_formula)[0]
 | 
					    exist_formula = _get_similar_formula(not_exist_formula)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    return command.script.replace(not_exist_formula, exist_formula, 1)
 | 
					    return command.script.replace(not_exist_formula, exist_formula, 1)
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,8 +1,7 @@
 | 
				
			|||||||
import difflib
 | 
					 | 
				
			||||||
import os
 | 
					import os
 | 
				
			||||||
import re
 | 
					import re
 | 
				
			||||||
import subprocess
 | 
					import subprocess
 | 
				
			||||||
 | 
					from thefuck.utils import get_closest
 | 
				
			||||||
 | 
					
 | 
				
			||||||
BREW_CMD_PATH = '/Library/Homebrew/cmd'
 | 
					BREW_CMD_PATH = '/Library/Homebrew/cmd'
 | 
				
			||||||
TAP_PATH = '/Library/Taps'
 | 
					TAP_PATH = '/Library/Taps'
 | 
				
			||||||
@@ -12,7 +11,8 @@ TAP_CMD_PATH = '/%s/%s/cmd'
 | 
				
			|||||||
def _get_brew_path_prefix():
 | 
					def _get_brew_path_prefix():
 | 
				
			||||||
    """To get brew path"""
 | 
					    """To get brew path"""
 | 
				
			||||||
    try:
 | 
					    try:
 | 
				
			||||||
        return subprocess.check_output(['brew', '--prefix']).strip()
 | 
					        return subprocess.check_output(['brew', '--prefix'],
 | 
				
			||||||
 | 
					                                       universal_newlines=True).strip()
 | 
				
			||||||
    except:
 | 
					    except:
 | 
				
			||||||
        return None
 | 
					        return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -77,8 +77,8 @@ if brew_path_prefix:
 | 
				
			|||||||
        pass
 | 
					        pass
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def _get_similar_commands(command):
 | 
					def _get_similar_command(command):
 | 
				
			||||||
    return difflib.get_close_matches(command, brew_commands)
 | 
					    return get_closest(command, brew_commands)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def match(command, settings):
 | 
					def match(command, settings):
 | 
				
			||||||
@@ -89,7 +89,7 @@ def match(command, settings):
 | 
				
			|||||||
    if is_proper_command:
 | 
					    if is_proper_command:
 | 
				
			||||||
        broken_cmd = re.findall(r'Error: Unknown command: ([a-z]+)',
 | 
					        broken_cmd = re.findall(r'Error: Unknown command: ([a-z]+)',
 | 
				
			||||||
                                command.stderr)[0]
 | 
					                                command.stderr)[0]
 | 
				
			||||||
        has_possible_commands = len(_get_similar_commands(broken_cmd)) > 0
 | 
					        has_possible_commands = bool(_get_similar_command(broken_cmd))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    return has_possible_commands
 | 
					    return has_possible_commands
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -97,6 +97,6 @@ def match(command, settings):
 | 
				
			|||||||
def get_new_command(command, settings):
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
    broken_cmd = re.findall(r'Error: Unknown command: ([a-z]+)',
 | 
					    broken_cmd = re.findall(r'Error: Unknown command: ([a-z]+)',
 | 
				
			||||||
                            command.stderr)[0]
 | 
					                            command.stderr)[0]
 | 
				
			||||||
    new_cmd = _get_similar_commands(broken_cmd)[0]
 | 
					    new_cmd = _get_similar_command(broken_cmd)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    return command.script.replace(broken_cmd, new_cmd, 1)
 | 
					    return command.script.replace(broken_cmd, new_cmd, 1)
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										14
									
								
								thefuck/rules/brew_upgrade.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								thefuck/rules/brew_upgrade.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
				
			|||||||
 | 
					# Appends --all to the brew upgrade command
 | 
				
			||||||
 | 
					# 
 | 
				
			||||||
 | 
					# Example:
 | 
				
			||||||
 | 
					# > brew upgrade
 | 
				
			||||||
 | 
					# Warning: brew upgrade with no arguments will change behaviour soon!
 | 
				
			||||||
 | 
					# It currently upgrades all formula but this will soon change to require '--all'.
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# 
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (command.script == 'brew upgrade')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return command.script + ' --all'
 | 
				
			||||||
							
								
								
									
										6
									
								
								thefuck/rules/cargo.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								thefuck/rules/cargo.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,6 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return command.script == 'cargo'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return 'cargo build'
 | 
				
			||||||
							
								
								
									
										14
									
								
								thefuck/rules/cargo_no_command.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								thefuck/rules/cargo_no_command.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
				
			|||||||
 | 
					import re
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return ('cargo' in command.script
 | 
				
			||||||
 | 
					            and 'No such subcommand' in command.stderr
 | 
				
			||||||
 | 
					            and 'Did you mean' in command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    broken = command.script.split()[1]
 | 
				
			||||||
 | 
					    fix = re.findall(r'Did you mean `([^`]*)`', command.stderr)[0]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    return command.script.replace(broken, fix, 1)
 | 
				
			||||||
							
								
								
									
										53
									
								
								thefuck/rules/cd_correction.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								thefuck/rules/cd_correction.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,53 @@
 | 
				
			|||||||
 | 
					#!/usr/bin/env python
 | 
				
			||||||
 | 
					__author__ = "mmussomele"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					"""Attempts to spellcheck and correct failed cd commands"""
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import os
 | 
				
			||||||
 | 
					from difflib import get_close_matches
 | 
				
			||||||
 | 
					from thefuck.utils import sudo_support
 | 
				
			||||||
 | 
					from thefuck.rules import cd_mkdir
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					MAX_ALLOWED_DIFF = 0.6
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def _get_sub_dirs(parent):
 | 
				
			||||||
 | 
					    """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))]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@sudo_support
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    """Match function copied from cd_mkdir.py"""
 | 
				
			||||||
 | 
					    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):
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					    Attempt to rebuild the path string by spellchecking the directories.
 | 
				
			||||||
 | 
					    If it fails (i.e. no directories are a close enough match), then it 
 | 
				
			||||||
 | 
					    defaults to the rules of cd_mkdir. 
 | 
				
			||||||
 | 
					    Change sensitivity by changing MAX_ALLOWED_DIFF. Default value is 0.6
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					    dest = command.script.split()[1].split(os.sep)
 | 
				
			||||||
 | 
					    if dest[-1] == '':
 | 
				
			||||||
 | 
					        dest = dest[:-1]
 | 
				
			||||||
 | 
					    cwd = os.getcwd()
 | 
				
			||||||
 | 
					    for directory in dest:
 | 
				
			||||||
 | 
					        if directory == ".":
 | 
				
			||||||
 | 
					            continue
 | 
				
			||||||
 | 
					        elif directory == "..":
 | 
				
			||||||
 | 
					            cwd = os.path.split(cwd)[0]
 | 
				
			||||||
 | 
					            continue
 | 
				
			||||||
 | 
					        best_matches = get_close_matches(directory, _get_sub_dirs(cwd), cutoff=MAX_ALLOWED_DIFF)
 | 
				
			||||||
 | 
					        if best_matches:
 | 
				
			||||||
 | 
					            cwd = os.path.join(cwd, best_matches[0])
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            return cd_mkdir.get_new_command(command, settings)
 | 
				
			||||||
 | 
					    return 'cd "{0}"'.format(cwd)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					enabled_by_default = True
 | 
				
			||||||
@@ -1,4 +1,5 @@
 | 
				
			|||||||
import re
 | 
					import re
 | 
				
			||||||
 | 
					from thefuck import shells
 | 
				
			||||||
from thefuck.utils import sudo_support
 | 
					from thefuck.utils import sudo_support
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -11,4 +12,5 @@ def match(command, settings):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
@sudo_support
 | 
					@sudo_support
 | 
				
			||||||
def get_new_command(command, settings):
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
    return re.sub(r'^cd (.*)', 'mkdir -p \\1 && cd \\1', command.script)
 | 
					    repl = shells.and_('mkdir -p \\1', 'cd \\1')
 | 
				
			||||||
 | 
					    return re.sub(r'^cd (.*)', repl, command.script)
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -4,8 +4,9 @@ from thefuck.utils import sudo_support
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
@sudo_support
 | 
					@sudo_support
 | 
				
			||||||
def match(command, settings):
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    stderr = command.stderr.lower()
 | 
				
			||||||
    return command.script.startswith('cp ') \
 | 
					    return command.script.startswith('cp ') \
 | 
				
			||||||
        and 'cp: omitting directory' in command.stderr.lower()
 | 
					        and ('omitting directory' in stderr or 'is a directory' in stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@sudo_support
 | 
					@sudo_support
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										9
									
								
								thefuck/rules/cpp11.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								thefuck/rules/cpp11.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,9 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (('g++' in command.script or 'clang++' in command.script) and
 | 
				
			||||||
 | 
					            ('This file requires compiler and library support for the '
 | 
				
			||||||
 | 
					             'ISO C++ 2011 standard.' in command.stderr or
 | 
				
			||||||
 | 
					             '-Wc++11-extensions' in command.stderr))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return command.script + ' -std=c++11'
 | 
				
			||||||
							
								
								
									
										8
									
								
								thefuck/rules/django_south_ghost.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										8
									
								
								thefuck/rules/django_south_ghost.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,8 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return 'manage.py' in command.script and \
 | 
				
			||||||
 | 
					           'migrate' in command.script \
 | 
				
			||||||
 | 
					           and 'or pass --delete-ghost-migrations' in command.stderr
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return u'{} --delete-ghost-migrations'.format(command.script)
 | 
				
			||||||
							
								
								
									
										8
									
								
								thefuck/rules/django_south_merge.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										8
									
								
								thefuck/rules/django_south_merge.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,8 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return 'manage.py' in command.script and \
 | 
				
			||||||
 | 
					           'migrate' in command.script \
 | 
				
			||||||
 | 
					           and '--merge: will just attempt the migration' in command.stderr
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return u'{} --merge'.format(command.script)
 | 
				
			||||||
							
								
								
									
										12
									
								
								thefuck/rules/dry.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								thefuck/rules/dry.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    split_command = command.script.split()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    return len(split_command) >= 2 and split_command[0] == split_command[1]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return command.script[command.script.find(' ')+1:]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# it should be rare enough to actually have to type twice the same word, so
 | 
				
			||||||
 | 
					# this rule can have a higher priority to come before things like "cd cd foo"
 | 
				
			||||||
 | 
					priority = 900
 | 
				
			||||||
							
								
								
									
										19
									
								
								thefuck/rules/git_add.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								thefuck/rules/git_add.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,19 @@
 | 
				
			|||||||
 | 
					import re
 | 
				
			||||||
 | 
					from thefuck import utils, shells
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return ('git' in command.script
 | 
				
			||||||
 | 
					            and 'did not match any file(s) known to git.' in command.stderr
 | 
				
			||||||
 | 
					            and "Did you forget to 'git add'?" in command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    missing_file = re.findall(
 | 
				
			||||||
 | 
					            r"error: pathspec '([^']*)' "
 | 
				
			||||||
 | 
					            "did not match any file\(s\) known to git.", command.stderr)[0]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    formatme = shells.and_('git add -- {}', '{}')
 | 
				
			||||||
 | 
					    return formatme.format(missing_file, command.script)
 | 
				
			||||||
							
								
								
									
										7
									
								
								thefuck/rules/git_branch_delete.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								thefuck/rules/git_branch_delete.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,7 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return ('git branch -d' in command.script
 | 
				
			||||||
 | 
					            and 'If you are sure you want to delete it' in command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return command.script.replace('-d', '-D')
 | 
				
			||||||
							
								
								
									
										12
									
								
								thefuck/rules/git_branch_list.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								thefuck/rules/git_branch_list.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
				
			|||||||
 | 
					from thefuck import utils, shells
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    # catches "git branch list" in place of "git branch"
 | 
				
			||||||
 | 
					    return command.script.split() == 'git branch list'.split()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return shells.and_('git branch --delete list', 'git branch')
 | 
				
			||||||
							
								
								
									
										19
									
								
								thefuck/rules/git_checkout.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								thefuck/rules/git_checkout.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,19 @@
 | 
				
			|||||||
 | 
					import re
 | 
				
			||||||
 | 
					from thefuck import shells, utils
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return ('git' in command.script
 | 
				
			||||||
 | 
					            and 'did not match any file(s) known to git.' in command.stderr
 | 
				
			||||||
 | 
					            and "Did you forget to 'git add'?" not in command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    missing_file = re.findall(
 | 
				
			||||||
 | 
					            r"error: pathspec '([^']*)' "
 | 
				
			||||||
 | 
					            "did not match any file\(s\) known to git.", command.stderr)[0]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    formatme = shells.and_('git branch {}', '{}')
 | 
				
			||||||
 | 
					    return formatme.format(missing_file, command.script)
 | 
				
			||||||
							
								
								
									
										13
									
								
								thefuck/rules/git_diff_staged.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								thefuck/rules/git_diff_staged.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,13 @@
 | 
				
			|||||||
 | 
					from thefuck import utils
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return ('git' in command.script and
 | 
				
			||||||
 | 
					            'diff' in command.script and
 | 
				
			||||||
 | 
					            '--staged' not in command.script)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return '{} --staged'.format(command.script)
 | 
				
			||||||
@@ -1,16 +1,28 @@
 | 
				
			|||||||
import re
 | 
					import re
 | 
				
			||||||
 | 
					from thefuck.utils import get_closest, git_support
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@git_support
 | 
				
			||||||
def match(command, settings):
 | 
					def match(command, settings):
 | 
				
			||||||
    return ('git' in command.script
 | 
					    return ('git' in command.script
 | 
				
			||||||
            and " is not a git command. See 'git --help'." in command.stderr
 | 
					            and " is not a git command. See 'git --help'." in command.stderr
 | 
				
			||||||
            and 'Did you mean' in command.stderr)
 | 
					            and 'Did you mean' in command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def _get_all_git_matched_commands(stderr):
 | 
				
			||||||
 | 
					    should_yield = False
 | 
				
			||||||
 | 
					    for line in stderr.split('\n'):
 | 
				
			||||||
 | 
					        if 'Did you mean' in line:
 | 
				
			||||||
 | 
					            should_yield = True
 | 
				
			||||||
 | 
					        elif should_yield and line:
 | 
				
			||||||
 | 
					            yield line.strip()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@git_support
 | 
				
			||||||
def get_new_command(command, settings):
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
    broken_cmd = re.findall(r"git: '([^']*)' is not a git command",
 | 
					    broken_cmd = re.findall(r"git: '([^']*)' is not a git command",
 | 
				
			||||||
                            command.stderr)[0]
 | 
					                            command.stderr)[0]
 | 
				
			||||||
    new_cmd = re.findall(r'Did you mean[^\n]*\n\s*([^\n]*)',
 | 
					    new_cmd = get_closest(broken_cmd,
 | 
				
			||||||
                         command.stderr)[0]
 | 
					                          _get_all_git_matched_commands(command.stderr))
 | 
				
			||||||
    return command.script.replace(broken_cmd, new_cmd, 1)
 | 
					    return command.script.replace(broken_cmd, new_cmd, 1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										17
									
								
								thefuck/rules/git_pull.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								thefuck/rules/git_pull.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
				
			|||||||
 | 
					from thefuck import shells, utils
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return ('git' in command.script
 | 
				
			||||||
 | 
					            and 'pull' in command.script
 | 
				
			||||||
 | 
					            and 'set-upstream' in command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    line = command.stderr.split('\n')[-3].strip()
 | 
				
			||||||
 | 
					    branch = line.split(' ')[-1]
 | 
				
			||||||
 | 
					    set_upstream = line.replace('<remote>', 'origin')\
 | 
				
			||||||
 | 
					                       .replace('<branch>', branch)
 | 
				
			||||||
 | 
					    return shells.and_(set_upstream, command.script)
 | 
				
			||||||
@@ -1,8 +1,13 @@
 | 
				
			|||||||
 | 
					from thefuck import utils
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
def match(command, settings):
 | 
					def match(command, settings):
 | 
				
			||||||
    return ('git' in command.script
 | 
					    return ('git' in command.script
 | 
				
			||||||
            and 'push' in command.script
 | 
					            and 'push' in command.script
 | 
				
			||||||
            and 'set-upstream' in command.stderr)
 | 
					            and 'set-upstream' in command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
def get_new_command(command, settings):
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
    return command.stderr.split('\n')[-3].strip()
 | 
					    return command.stderr.split('\n')[-3].strip()
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										14
									
								
								thefuck/rules/git_stash.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								thefuck/rules/git_stash.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
				
			|||||||
 | 
					from thefuck import shells, utils
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    # catches "Please commit or stash them" and "Please, commit your changes or
 | 
				
			||||||
 | 
					    # stash them before you can switch branches."
 | 
				
			||||||
 | 
					    return 'git' in command.script and 'or stash them' in command.stderr
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@utils.git_support
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    formatme = shells.and_('git stash', '{}')
 | 
				
			||||||
 | 
					    return formatme.format(command.script)
 | 
				
			||||||
							
								
								
									
										14
									
								
								thefuck/rules/go_run.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								thefuck/rules/go_run.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
				
			|||||||
 | 
					# Appends .go when compiling go files
 | 
				
			||||||
 | 
					# 
 | 
				
			||||||
 | 
					# Example:
 | 
				
			||||||
 | 
					# > go run foo
 | 
				
			||||||
 | 
					# error: go run: no go files listed
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# 
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (command.script.startswith ('go run ')
 | 
				
			||||||
 | 
					            and not command.script.endswith('.go'))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return command.script + '.go'
 | 
				
			||||||
							
								
								
									
										7
									
								
								thefuck/rules/grep_recursive.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								thefuck/rules/grep_recursive.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,7 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (command.script.startswith('grep')
 | 
				
			||||||
 | 
					            and 'is a directory' in command.stderr.lower())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return 'grep -r {}'.format(command.script[5:])
 | 
				
			||||||
							
								
								
									
										24
									
								
								thefuck/rules/history.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										24
									
								
								thefuck/rules/history.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,24 @@
 | 
				
			|||||||
 | 
					from difflib import get_close_matches
 | 
				
			||||||
 | 
					from thefuck.shells import get_history
 | 
				
			||||||
 | 
					from thefuck.utils import get_closest, memoize
 | 
				
			||||||
 | 
					from thefuck.rules.no_command import get_all_callables
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@memoize
 | 
				
			||||||
 | 
					def _history_of_exists_without_current(command):
 | 
				
			||||||
 | 
					    callables = get_all_callables()
 | 
				
			||||||
 | 
					    return [line for line in get_history()
 | 
				
			||||||
 | 
					            if line != command.script
 | 
				
			||||||
 | 
					            and line.split(' ')[0] in callables]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return len(get_close_matches(command.script,
 | 
				
			||||||
 | 
					                                 _history_of_exists_without_current(command)))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return get_closest(command.script,
 | 
				
			||||||
 | 
					                       _history_of_exists_without_current(command))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					priority = 9999
 | 
				
			||||||
							
								
								
									
										14
									
								
								thefuck/rules/java.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								thefuck/rules/java.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
				
			|||||||
 | 
					# Fixes common java command mistake
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# Example:
 | 
				
			||||||
 | 
					# > java foo.java
 | 
				
			||||||
 | 
					# Error: Could not find or load main class foo.java
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (command.script.startswith('java ')
 | 
				
			||||||
 | 
					            and command.script.endswith('.java'))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return command.script[:-5]
 | 
				
			||||||
							
								
								
									
										15
									
								
								thefuck/rules/javac.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								thefuck/rules/javac.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,15 @@
 | 
				
			|||||||
 | 
					# Appends .java when compiling java files
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# Example:
 | 
				
			||||||
 | 
					# > javac foo
 | 
				
			||||||
 | 
					# error: Class names, 'foo', are only accepted if annotation
 | 
				
			||||||
 | 
					# processing is explicitly requested
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (command.script.startswith('javac ')
 | 
				
			||||||
 | 
					            and not command.script.endswith('.java'))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return command.script + '.java'
 | 
				
			||||||
@@ -1,8 +1,7 @@
 | 
				
			|||||||
enabled_by_default = False
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def match(command, settings):
 | 
					def match(command, settings):
 | 
				
			||||||
    return 'ls' in command.script and not ('ls -' in command.script)
 | 
					    return (command.script == 'ls'
 | 
				
			||||||
 | 
					            or command.script.startswith('ls ')
 | 
				
			||||||
 | 
					            and not ('ls -' in command.script))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def get_new_command(command, settings):
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										13
									
								
								thefuck/rules/man.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								thefuck/rules/man.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,13 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return command.script.strip().startswith('man ')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    if '3' in command.script:
 | 
				
			||||||
 | 
					        return command.script.replace("3", "2")
 | 
				
			||||||
 | 
					    if '2' in command.script:
 | 
				
			||||||
 | 
					        return command.script.replace("2", "3")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    split_cmd = command.script.split()
 | 
				
			||||||
 | 
					    split_cmd.insert(1, ' 3 ')
 | 
				
			||||||
 | 
					    return "".join(split_cmd)
 | 
				
			||||||
							
								
								
									
										9
									
								
								thefuck/rules/man_no_space.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								thefuck/rules/man_no_space.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,9 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (command.script.startswith(u'man')
 | 
				
			||||||
 | 
					            and u'command not found' in command.stderr.lower())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return u'man {}'.format(command.script[3:])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					priority = 2000
 | 
				
			||||||
							
								
								
									
										29
									
								
								thefuck/rules/mercurial.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								thefuck/rules/mercurial.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,29 @@
 | 
				
			|||||||
 | 
					import re
 | 
				
			||||||
 | 
					from thefuck.utils import get_closest
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def extract_possisiblities(command):
 | 
				
			||||||
 | 
					    possib = re.findall(r'\n\(did you mean one of ([^\?]+)\?\)', command.stderr)
 | 
				
			||||||
 | 
					    if possib:
 | 
				
			||||||
 | 
					        return possib[0].split(', ')
 | 
				
			||||||
 | 
					    possib = re.findall(r'\n    ([^$]+)$', command.stderr)
 | 
				
			||||||
 | 
					    if possib:
 | 
				
			||||||
 | 
					        return possib[0].split(' ')
 | 
				
			||||||
 | 
					    return possib
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (command.script.startswith('hg ')
 | 
				
			||||||
 | 
					            and ('hg: unknown command' in command.stderr
 | 
				
			||||||
 | 
					                 and '(did you mean one of ' in command.stderr
 | 
				
			||||||
 | 
					                 or "hg: command '" in command.stderr
 | 
				
			||||||
 | 
					                 and "' is ambiguous:" in command.stderr
 | 
				
			||||||
 | 
					                 )
 | 
				
			||||||
 | 
					            )
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    script = command.script.split(' ')
 | 
				
			||||||
 | 
					    possisiblities = extract_possisiblities(command)
 | 
				
			||||||
 | 
					    script[1] = get_closest(script[1], possisiblities)
 | 
				
			||||||
 | 
					    return ' '.join(script)
 | 
				
			||||||
@@ -1,7 +1,8 @@
 | 
				
			|||||||
from difflib import get_close_matches
 | 
					from difflib import get_close_matches
 | 
				
			||||||
import os
 | 
					import os
 | 
				
			||||||
from pathlib import Path
 | 
					from pathlib import Path
 | 
				
			||||||
from thefuck.utils import sudo_support
 | 
					from thefuck.utils import sudo_support, memoize
 | 
				
			||||||
 | 
					from thefuck.shells import thefuck_alias, get_aliases
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def _safe(fn, fallback):
 | 
					def _safe(fn, fallback):
 | 
				
			||||||
@@ -11,23 +12,29 @@ def _safe(fn, fallback):
 | 
				
			|||||||
        return fallback
 | 
					        return fallback
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def _get_all_bins():
 | 
					@memoize
 | 
				
			||||||
 | 
					def get_all_callables():
 | 
				
			||||||
 | 
					    tf_alias = thefuck_alias()
 | 
				
			||||||
    return [exe.name
 | 
					    return [exe.name
 | 
				
			||||||
            for path in os.environ.get('PATH', '').split(':')
 | 
					            for path in os.environ.get('PATH', '').split(':')
 | 
				
			||||||
            for exe in _safe(lambda: list(Path(path).iterdir()), [])
 | 
					            for exe in _safe(lambda: list(Path(path).iterdir()), [])
 | 
				
			||||||
            if not _safe(exe.is_dir, True)]
 | 
					            if not _safe(exe.is_dir, True)] + [
 | 
				
			||||||
 | 
					                alias for alias in get_aliases() if alias != tf_alias]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@sudo_support
 | 
					@sudo_support
 | 
				
			||||||
def match(command, settings):
 | 
					def match(command, settings):
 | 
				
			||||||
    return 'not found' in command.stderr and \
 | 
					    return 'not found' in command.stderr and \
 | 
				
			||||||
           bool(get_close_matches(command.script.split(' ')[0],
 | 
					           bool(get_close_matches(command.script.split(' ')[0],
 | 
				
			||||||
                                  _get_all_bins()))
 | 
					                                  get_all_callables()))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@sudo_support
 | 
					@sudo_support
 | 
				
			||||||
def get_new_command(command, settings):
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
    old_command = command.script.split(' ')[0]
 | 
					    old_command = command.script.split(' ')[0]
 | 
				
			||||||
    new_command = get_close_matches(old_command,
 | 
					    new_command = get_close_matches(old_command,
 | 
				
			||||||
                                    _get_all_bins())[0]
 | 
					                                    get_all_callables())[0]
 | 
				
			||||||
    return ' '.join([new_command] + command.script.split(' ')[1:])
 | 
					    return ' '.join([new_command] + command.script.split(' ')[1:])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					priority = 3000
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										30
									
								
								thefuck/rules/no_such_file.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										30
									
								
								thefuck/rules/no_such_file.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,30 @@
 | 
				
			|||||||
 | 
					import re
 | 
				
			||||||
 | 
					from thefuck import shells
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					patterns = (
 | 
				
			||||||
 | 
					    r"mv: cannot move '[^']*' to '([^']*)': No such file or directory",
 | 
				
			||||||
 | 
					    r"mv: cannot move '[^']*' to '([^']*)': Not a directory",
 | 
				
			||||||
 | 
					    r"cp: cannot create regular file '([^']*)': No such file or directory",
 | 
				
			||||||
 | 
					    r"cp: cannot create regular file '([^']*)': Not a directory",
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    for pattern in patterns:
 | 
				
			||||||
 | 
					        if re.search(pattern, command.stderr):
 | 
				
			||||||
 | 
					            return True
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    return False
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    for pattern in patterns:
 | 
				
			||||||
 | 
					        file = re.findall(pattern, command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if file:
 | 
				
			||||||
 | 
					            file = file[0]
 | 
				
			||||||
 | 
					            dir = file[0:file.rfind('/')]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            formatme = shells.and_('mkdir -p {}', '{}')
 | 
				
			||||||
 | 
					            return formatme.format(dir, command.script)
 | 
				
			||||||
							
								
								
									
										26
									
								
								thefuck/rules/open.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										26
									
								
								thefuck/rules/open.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,26 @@
 | 
				
			|||||||
 | 
					# Opens URL's in the default web browser
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# Example:
 | 
				
			||||||
 | 
					# > open github.com
 | 
				
			||||||
 | 
					# The file ~/github.com does not exist.
 | 
				
			||||||
 | 
					# Perhaps you meant 'http://github.com'?
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (command.script.startswith(('open', 'xdg-open', 'gnome-open', 'kde-open'))
 | 
				
			||||||
 | 
					            and (
 | 
				
			||||||
 | 
					                '.com' in command.script
 | 
				
			||||||
 | 
					                or '.net' in command.script
 | 
				
			||||||
 | 
					                or '.org' in command.script
 | 
				
			||||||
 | 
					                or '.ly' in command.script
 | 
				
			||||||
 | 
					                or '.io' in command.script
 | 
				
			||||||
 | 
					                or '.se' in command.script
 | 
				
			||||||
 | 
					                or '.edu' in command.script
 | 
				
			||||||
 | 
					                or '.info' in command.script
 | 
				
			||||||
 | 
					                or '.me' in command.script
 | 
				
			||||||
 | 
					                or 'www.' in command.script))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return 'open http://' + command.script[5:]
 | 
				
			||||||
							
								
								
									
										43
									
								
								thefuck/rules/pacman.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										43
									
								
								thefuck/rules/pacman.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,43 @@
 | 
				
			|||||||
 | 
					import subprocess
 | 
				
			||||||
 | 
					from thefuck.utils import DEVNULL, which
 | 
				
			||||||
 | 
					from thefuck import shells
 | 
				
			||||||
 | 
					from thefuck.utils import memoize
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@memoize
 | 
				
			||||||
 | 
					def __get_pkgfile(command):
 | 
				
			||||||
 | 
					    try:
 | 
				
			||||||
 | 
					        command = command.script
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if command.startswith('sudo'):
 | 
				
			||||||
 | 
					            command = command[5:]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        command = command.split(" ")[0]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        return subprocess.check_output(
 | 
				
			||||||
 | 
					            ['pkgfile', '-b', '-v', command],
 | 
				
			||||||
 | 
					            universal_newlines=True, stderr=DEVNULL
 | 
				
			||||||
 | 
					        ).split()
 | 
				
			||||||
 | 
					    except subprocess.CalledProcessError:
 | 
				
			||||||
 | 
					        return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return 'not found' in command.stderr and __get_pkgfile(command)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    package = __get_pkgfile(command)[0]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    formatme = shells.and_('{} -S {}', '{}')
 | 
				
			||||||
 | 
					    return formatme.format(pacman, package, command.script)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					if not which('pkgfile'):
 | 
				
			||||||
 | 
					    enabled_by_default = False
 | 
				
			||||||
 | 
					elif which('yaourt'):
 | 
				
			||||||
 | 
					    pacman = 'yaourt'
 | 
				
			||||||
 | 
					elif which('pacman'):
 | 
				
			||||||
 | 
					    pacman = 'sudo pacman'
 | 
				
			||||||
 | 
					else:
 | 
				
			||||||
 | 
					    enabled_by_default = False
 | 
				
			||||||
							
								
								
									
										14
									
								
								thefuck/rules/python_execute.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								thefuck/rules/python_execute.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
				
			|||||||
 | 
					# Appends .py when executing python files
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# Example:
 | 
				
			||||||
 | 
					# > python foo
 | 
				
			||||||
 | 
					# error: python: can't open file 'foo': [Errno 2] No such file or directory
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return (command.script.startswith('python ')
 | 
				
			||||||
 | 
					            and not command.script.endswith('.py'))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return command.script + '.py'
 | 
				
			||||||
							
								
								
									
										14
									
								
								thefuck/rules/quotation_marks.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								thefuck/rules/quotation_marks.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
				
			|||||||
 | 
					# Fixes careless " and ' usage
 | 
				
			||||||
 | 
					# 
 | 
				
			||||||
 | 
					# Example:
 | 
				
			||||||
 | 
					# > git commit -m 'My Message"
 | 
				
			||||||
 | 
					# 
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# 
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return ('\'' in command.script
 | 
				
			||||||
 | 
					            and '\"' in command.script)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return command.script.replace ('\'', '\"')
 | 
				
			||||||
							
								
								
									
										17
									
								
								thefuck/rules/sed_unterminated_s.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								thefuck/rules/sed_unterminated_s.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
				
			|||||||
 | 
					import shlex
 | 
				
			||||||
 | 
					from thefuck.utils import quote
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return ('sed' in command.script
 | 
				
			||||||
 | 
					            and "unterminated `s' command" in command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    script = shlex.split(command.script)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    for (i, e) in enumerate(script):
 | 
				
			||||||
 | 
					        if e.startswith(('s/', '-es/')) and e[-1] != '/':
 | 
				
			||||||
 | 
					            script[i] += '/'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    return ' '.join(map(quote, script))
 | 
				
			||||||
@@ -7,11 +7,14 @@ patterns = ['permission denied',
 | 
				
			|||||||
            'root privilege',
 | 
					            'root privilege',
 | 
				
			||||||
            'This command has to be run under the root user.',
 | 
					            'This command has to be run under the root user.',
 | 
				
			||||||
            'This operation requires root.',
 | 
					            'This operation requires root.',
 | 
				
			||||||
            'You need to be root to perform this command.',
 | 
					 | 
				
			||||||
            'requested operation requires superuser privilege',
 | 
					            'requested operation requires superuser privilege',
 | 
				
			||||||
            'must be run as root',
 | 
					            'must be run as root',
 | 
				
			||||||
            'must be superuser',
 | 
					            'must be superuser',
 | 
				
			||||||
            'Need to be root']
 | 
					            'must be root',
 | 
				
			||||||
 | 
					            'need to be root',
 | 
				
			||||||
 | 
					            'need root',
 | 
				
			||||||
 | 
					            'only root can do that',
 | 
				
			||||||
 | 
					            'authentication is required']
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def match(command, settings):
 | 
					def match(command, settings):
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										21
									
								
								thefuck/rules/systemctl.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								thefuck/rules/systemctl.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
				
			|||||||
 | 
					"""
 | 
				
			||||||
 | 
					The confusion in systemctl's param order is massive.
 | 
				
			||||||
 | 
					"""
 | 
				
			||||||
 | 
					from thefuck.utils import sudo_support
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@sudo_support
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    # Catches 'Unknown operation 'service'.' when executing systemctl with
 | 
				
			||||||
 | 
					    # misordered arguments
 | 
				
			||||||
 | 
					    cmd = command.script.split()
 | 
				
			||||||
 | 
					    return ('systemctl' in command.script and
 | 
				
			||||||
 | 
					            'Unknown operation \'' in command.stderr and
 | 
				
			||||||
 | 
					            len(cmd) - cmd.index('systemctl') == 3)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@sudo_support
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    cmd = command.script.split()
 | 
				
			||||||
 | 
					    cmd[-1], cmd[-2] = cmd[-2], cmd[-1]
 | 
				
			||||||
 | 
					    return ' '.join(cmd)
 | 
				
			||||||
							
								
								
									
										10
									
								
								thefuck/rules/test.py.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								thefuck/rules/test.py.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,10 @@
 | 
				
			|||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return command.script == 'test.py' and 'not found' in command.stderr
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    return 'py.test'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# make it come before the python_command rule
 | 
				
			||||||
 | 
					priority = 900
 | 
				
			||||||
							
								
								
									
										20
									
								
								thefuck/rules/tmux.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										20
									
								
								thefuck/rules/tmux.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,20 @@
 | 
				
			|||||||
 | 
					from thefuck.utils import get_closest
 | 
				
			||||||
 | 
					import re
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    return ('tmux' in command.script
 | 
				
			||||||
 | 
					            and 'ambiguous command:' in command.stderr
 | 
				
			||||||
 | 
					            and 'could be:' in command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    cmd = re.match(r"ambiguous command: (.*), could be: (.*)",
 | 
				
			||||||
 | 
					                   command.stderr)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    old_cmd = cmd.group(1)
 | 
				
			||||||
 | 
					    suggestions = [cmd.strip() for cmd in cmd.group(2).split(',')]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    new_cmd = get_closest(old_cmd, suggestions)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    return command.script.replace(old_cmd, new_cmd)
 | 
				
			||||||
							
								
								
									
										31
									
								
								thefuck/rules/whois.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								thefuck/rules/whois.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,31 @@
 | 
				
			|||||||
 | 
					# -*- encoding: utf-8 -*-
 | 
				
			||||||
 | 
					from six.moves.urllib.parse import urlparse
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def match(command, settings):
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					    What the `whois` command returns depends on the 'Whois server' it contacted
 | 
				
			||||||
 | 
					    and is not consistent through different servers. But there can be only two
 | 
				
			||||||
 | 
					    types of errors I can think of with `whois`:
 | 
				
			||||||
 | 
					        - `whois https://en.wikipedia.org/` → `whois en.wikipedia.org`;
 | 
				
			||||||
 | 
					        - `whois en.wikipedia.org` → `whois wikipedia.org`.
 | 
				
			||||||
 | 
					    So we match any `whois` command and then:
 | 
				
			||||||
 | 
					        - if there is a slash: keep only the FQDN;
 | 
				
			||||||
 | 
					        - if there is no slash but there is a point: removes the left-most
 | 
				
			||||||
 | 
					          subdomain.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    We cannot either remove all subdomains because we cannot know which part is
 | 
				
			||||||
 | 
					    the subdomains and which is the domain, consider:
 | 
				
			||||||
 | 
					        - www.google.fr → subdomain: www, domain: 'google.fr';
 | 
				
			||||||
 | 
					        - google.co.uk → subdomain: None, domain; 'google.co.uk'.
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					    return 'whois' in command.script and len(command.script.split()) > 1
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_new_command(command, settings):
 | 
				
			||||||
 | 
					    url = command.script.split()[1]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if '/' in command.script:
 | 
				
			||||||
 | 
					        return 'whois ' + urlparse(url).netloc
 | 
				
			||||||
 | 
					    elif '.' in command.script:
 | 
				
			||||||
 | 
					        return 'whois ' + '.'.join(urlparse(url).path.split('.')[1:])
 | 
				
			||||||
@@ -1,22 +1,24 @@
 | 
				
			|||||||
"""Module with shell specific actions, each shell class should
 | 
					"""Module with shell specific actions, each shell class should
 | 
				
			||||||
implement `from_shell` and `to_shell` methods.
 | 
					implement `from_shell`, `to_shell`, `app_alias`, `put_to_history` and `get_aliases`
 | 
				
			||||||
 | 
					methods.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
"""
 | 
					"""
 | 
				
			||||||
from collections import defaultdict
 | 
					from collections import defaultdict
 | 
				
			||||||
from subprocess import Popen, PIPE
 | 
					from subprocess import Popen, PIPE
 | 
				
			||||||
 | 
					from time import time
 | 
				
			||||||
import os
 | 
					import os
 | 
				
			||||||
 | 
					import io
 | 
				
			||||||
from psutil import Process
 | 
					from psutil import Process
 | 
				
			||||||
 | 
					from .utils import DEVNULL, memoize
 | 
				
			||||||
 | 
					 | 
				
			||||||
FNULL = open(os.devnull, 'w')
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class Generic(object):
 | 
					class Generic(object):
 | 
				
			||||||
    def _get_aliases(self):
 | 
					
 | 
				
			||||||
 | 
					    def get_aliases(self):
 | 
				
			||||||
        return {}
 | 
					        return {}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def _expand_aliases(self, command_script):
 | 
					    def _expand_aliases(self, command_script):
 | 
				
			||||||
        aliases = self._get_aliases()
 | 
					        aliases = self.get_aliases()
 | 
				
			||||||
        binary = command_script.split(' ')[0]
 | 
					        binary = command_script.split(' ')[0]
 | 
				
			||||||
        if binary in aliases:
 | 
					        if binary in aliases:
 | 
				
			||||||
            return command_script.replace(binary, aliases[binary], 1)
 | 
					            return command_script.replace(binary, aliases[binary], 1)
 | 
				
			||||||
@@ -32,46 +34,200 @@ class Generic(object):
 | 
				
			|||||||
        return command_script
 | 
					        return command_script
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def app_alias(self):
 | 
					    def app_alias(self):
 | 
				
			||||||
        return "\nalias fuck='eval $(thefuck $(fc -ln -1))'\n"
 | 
					        return "\nalias fuck='TF_ALIAS=fuck eval $(thefuck $(fc -ln -1))'\n"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_file_name(self):
 | 
				
			||||||
 | 
					        return ''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_line(self, command_script):
 | 
				
			||||||
 | 
					        return ''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def put_to_history(self, command_script):
 | 
				
			||||||
 | 
					        """Puts command script to shell history."""
 | 
				
			||||||
 | 
					        history_file_name = self._get_history_file_name()
 | 
				
			||||||
 | 
					        if os.path.isfile(history_file_name):
 | 
				
			||||||
 | 
					            with open(history_file_name, 'a') as history:
 | 
				
			||||||
 | 
					                history.write(self._get_history_line(command_script))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _script_from_history(self, line):
 | 
				
			||||||
 | 
					        """Returns prepared history line.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        Should return a blank line if history line is corrupted or empty.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        return ''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def get_history(self):
 | 
				
			||||||
 | 
					        """Returns list of history entries."""
 | 
				
			||||||
 | 
					        history_file_name = self._get_history_file_name()
 | 
				
			||||||
 | 
					        if os.path.isfile(history_file_name):
 | 
				
			||||||
 | 
					            with io.open(history_file_name, 'r',
 | 
				
			||||||
 | 
					                         encoding='utf-8', errors='ignore') as history:
 | 
				
			||||||
 | 
					                for line in history:
 | 
				
			||||||
 | 
					                    prepared = self._script_from_history(line)\
 | 
				
			||||||
 | 
					                                   .strip()
 | 
				
			||||||
 | 
					                    if prepared:
 | 
				
			||||||
 | 
					                        yield prepared
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def and_(self, *commands):
 | 
				
			||||||
 | 
					        return u' && '.join(commands)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class Bash(Generic):
 | 
					class Bash(Generic):
 | 
				
			||||||
 | 
					    def app_alias(self):
 | 
				
			||||||
 | 
					        return "\nTF_ALIAS=fuck alias fuck='eval $(thefuck $(fc -ln -1));" \
 | 
				
			||||||
 | 
					               " history -r'\n"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def _parse_alias(self, alias):
 | 
					    def _parse_alias(self, alias):
 | 
				
			||||||
        name, value = alias.replace('alias ', '', 1).split('=', 1)
 | 
					        name, value = alias.replace('alias ', '', 1).split('=', 1)
 | 
				
			||||||
        if value[0] == value[-1] == '"' or value[0] == value[-1] == "'":
 | 
					        if value[0] == value[-1] == '"' or value[0] == value[-1] == "'":
 | 
				
			||||||
            value = value[1:-1]
 | 
					            value = value[1:-1]
 | 
				
			||||||
        return name, value
 | 
					        return name, value
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def _get_aliases(self):
 | 
					    def get_aliases(self):
 | 
				
			||||||
        proc = Popen('bash -ic alias', stdout=PIPE, stderr=FNULL, shell=True)
 | 
					        proc = Popen('bash -ic alias', stdout=PIPE, stderr=DEVNULL,
 | 
				
			||||||
 | 
					                     shell=True)
 | 
				
			||||||
        return dict(
 | 
					        return dict(
 | 
				
			||||||
            self._parse_alias(alias)
 | 
					            self._parse_alias(alias)
 | 
				
			||||||
            for alias in proc.stdout.read().decode('utf-8').split('\n')
 | 
					            for alias in proc.stdout.read().decode('utf-8').split('\n')
 | 
				
			||||||
            if alias)
 | 
					            if alias and '=' in alias)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_file_name(self):
 | 
				
			||||||
 | 
					        return os.environ.get("HISTFILE",
 | 
				
			||||||
 | 
					                              os.path.expanduser('~/.bash_history'))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_line(self, command_script):
 | 
				
			||||||
 | 
					        return u'{}\n'.format(command_script)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _script_from_history(self, line):
 | 
				
			||||||
 | 
					        print(line)
 | 
				
			||||||
 | 
					        return line
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					class Fish(Generic):
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_overridden_aliases(self):
 | 
				
			||||||
 | 
					        overridden_aliases = os.environ.get('TF_OVERRIDDEN_ALIASES', '').strip()
 | 
				
			||||||
 | 
					        if overridden_aliases:
 | 
				
			||||||
 | 
					            return [alias.strip() for alias in overridden_aliases.split(',')]
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            return ['cd', 'grep', 'ls', 'man', 'open']
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def app_alias(self):
 | 
				
			||||||
 | 
					        return ("set TF_ALIAS fuck\n"
 | 
				
			||||||
 | 
					                "function fuck -d 'Correct your previous console command'\n"
 | 
				
			||||||
 | 
					                "    set -l exit_code $status\n"
 | 
				
			||||||
 | 
					                "    set -l eval_script"
 | 
				
			||||||
 | 
					                " (mktemp 2>/dev/null ; or mktemp -t 'thefuck')\n"
 | 
				
			||||||
 | 
					                "    set -l fucked_up_commandd $history[1]\n"
 | 
				
			||||||
 | 
					                "    thefuck $fucked_up_commandd > $eval_script\n"
 | 
				
			||||||
 | 
					                "    . $eval_script\n"
 | 
				
			||||||
 | 
					                "    rm $eval_script\n"
 | 
				
			||||||
 | 
					                "    if test $exit_code -ne 0\n"
 | 
				
			||||||
 | 
					                "        history --delete $fucked_up_commandd\n"
 | 
				
			||||||
 | 
					                "    end\n"
 | 
				
			||||||
 | 
					                "end")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def get_aliases(self):
 | 
				
			||||||
 | 
					        overridden = self._get_overridden_aliases()
 | 
				
			||||||
 | 
					        proc = Popen('fish -ic functions', stdout=PIPE, stderr=DEVNULL,
 | 
				
			||||||
 | 
					                     shell=True)
 | 
				
			||||||
 | 
					        functions = proc.stdout.read().decode('utf-8').strip().split('\n')
 | 
				
			||||||
 | 
					        return {func: func for func in functions if func not in overridden}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _expand_aliases(self, command_script):
 | 
				
			||||||
 | 
					        aliases = self.get_aliases()
 | 
				
			||||||
 | 
					        binary = command_script.split(' ')[0]
 | 
				
			||||||
 | 
					        if binary in aliases:
 | 
				
			||||||
 | 
					            return u'fish -ic "{}"'.format(command_script.replace('"', r'\"'))
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            return command_script
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def from_shell(self, command_script):
 | 
				
			||||||
 | 
					        """Prepares command before running in app."""
 | 
				
			||||||
 | 
					        return self._expand_aliases(command_script)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_file_name(self):
 | 
				
			||||||
 | 
					        return os.path.expanduser('~/.config/fish/fish_history')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_line(self, command_script):
 | 
				
			||||||
 | 
					        return u'- cmd: {}\n   when: {}\n'.format(command_script, int(time()))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def and_(self, *commands):
 | 
				
			||||||
 | 
					        return u'; and '.join(commands)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class Zsh(Generic):
 | 
					class Zsh(Generic):
 | 
				
			||||||
 | 
					    def app_alias(self):
 | 
				
			||||||
 | 
					        return "\nTF_ALIAS=fuck" \
 | 
				
			||||||
 | 
					               " alias fuck='eval $(thefuck $(fc -ln -1 | tail -n 1));" \
 | 
				
			||||||
 | 
					               " fc -R'\n"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def _parse_alias(self, alias):
 | 
					    def _parse_alias(self, alias):
 | 
				
			||||||
        name, value = alias.split('=', 1)
 | 
					        name, value = alias.split('=', 1)
 | 
				
			||||||
        if value[0] == value[-1] == '"' or value[0] == value[-1] == "'":
 | 
					        if value[0] == value[-1] == '"' or value[0] == value[-1] == "'":
 | 
				
			||||||
            value = value[1:-1]
 | 
					            value = value[1:-1]
 | 
				
			||||||
        return name, value
 | 
					        return name, value
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def _get_aliases(self):
 | 
					    def get_aliases(self):
 | 
				
			||||||
        proc = Popen('zsh -ic alias', stdout=PIPE, stderr=FNULL, shell=True)
 | 
					        proc = Popen('zsh -ic alias', stdout=PIPE, stderr=DEVNULL,
 | 
				
			||||||
 | 
					                     shell=True)
 | 
				
			||||||
        return dict(
 | 
					        return dict(
 | 
				
			||||||
            self._parse_alias(alias)
 | 
					            self._parse_alias(alias)
 | 
				
			||||||
            for alias in proc.stdout.read().decode('utf-8').split('\n')
 | 
					            for alias in proc.stdout.read().decode('utf-8').split('\n')
 | 
				
			||||||
            if alias)
 | 
					            if alias and '=' in alias)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_file_name(self):
 | 
				
			||||||
 | 
					        return os.environ.get("HISTFILE",
 | 
				
			||||||
 | 
					                              os.path.expanduser('~/.zsh_history'))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_line(self, command_script):
 | 
				
			||||||
 | 
					        return u': {}:0;{}\n'.format(int(time()), command_script)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _script_from_history(self, line):
 | 
				
			||||||
 | 
					        if ';' in line:
 | 
				
			||||||
 | 
					            return line.split(';', 1)[1]
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            return ''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					class Tcsh(Generic):
 | 
				
			||||||
 | 
					    def app_alias(self):
 | 
				
			||||||
 | 
					        return "\nalias fuck 'setenv TF_ALIAS fuck && set fucked_cmd=`history -h 2 | head -n 1` && eval `thefuck ${fucked_cmd}`'\n"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _parse_alias(self, alias):
 | 
				
			||||||
 | 
					        name, value = alias.split("\t", 1)
 | 
				
			||||||
 | 
					        return name, value
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def get_aliases(self):
 | 
				
			||||||
 | 
					        proc = Popen('tcsh -ic alias', stdout=PIPE, stderr=DEVNULL,
 | 
				
			||||||
 | 
					                     shell=True)
 | 
				
			||||||
 | 
					        return dict(
 | 
				
			||||||
 | 
					            self._parse_alias(alias)
 | 
				
			||||||
 | 
					            for alias in proc.stdout.read().decode('utf-8').split('\n')
 | 
				
			||||||
 | 
					            if alias and '\t' in alias)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_file_name(self):
 | 
				
			||||||
 | 
					        return os.environ.get("HISTFILE",
 | 
				
			||||||
 | 
					                              os.path.expanduser('~/.history'))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_history_line(self, command_script):
 | 
				
			||||||
 | 
					        return u'#+{}\n{}\n'.format(int(time()), command_script)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
shells = defaultdict(lambda: Generic(), {
 | 
					shells = defaultdict(lambda: Generic(), {
 | 
				
			||||||
    'bash': Bash(),
 | 
					    'bash': Bash(),
 | 
				
			||||||
    'zsh': Zsh()})
 | 
					    'fish': Fish(),
 | 
				
			||||||
 | 
					    'zsh': Zsh(),
 | 
				
			||||||
 | 
					    'csh': Tcsh(),
 | 
				
			||||||
 | 
					    'tcsh': Tcsh()})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def _get_shell():
 | 
					def _get_shell():
 | 
				
			||||||
    shell = Process(os.getpid()).parent().cmdline()[0]
 | 
					    try:
 | 
				
			||||||
 | 
					        shell = Process(os.getpid()).parent().name()
 | 
				
			||||||
 | 
					    except TypeError:
 | 
				
			||||||
 | 
					        shell = Process(os.getpid()).parent.name
 | 
				
			||||||
    return shells[shell]
 | 
					    return shells[shell]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -84,4 +240,26 @@ def to_shell(command):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def app_alias():
 | 
					def app_alias():
 | 
				
			||||||
    return _get_shell().app_alias()
 | 
					    print(_get_shell().app_alias())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def thefuck_alias():
 | 
				
			||||||
 | 
					    return os.environ.get('TF_ALIAS', 'fuck')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def put_to_history(command):
 | 
				
			||||||
 | 
					    return _get_shell().put_to_history(command)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def and_(*commands):
 | 
				
			||||||
 | 
					    return _get_shell().and_(*commands)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@memoize
 | 
				
			||||||
 | 
					def get_aliases():
 | 
				
			||||||
 | 
					    return list(_get_shell().get_aliases().keys())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@memoize
 | 
				
			||||||
 | 
					def get_history():
 | 
				
			||||||
 | 
					    return list(_get_shell().get_history())
 | 
				
			||||||
 
 | 
				
			|||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user