add python-module-error rule

Author: dvjn

README.md

@@ -281,6 +281,7 @@ following rules are enabled by default:
 * `pyenv_no_such_command` – fixes wrong pyenv commands like `pyenv isntall` or `pyenv list`;
 * `python_command` – prepends `python` when you try to run non-executable/without `./` python script;
 * `python_execute` – appends missing `.py` when executing Python files;
+* `python_module_error` – fixes ModuleNotFoundError by trying to `pip install` that module;
 * `quotation_marks` – fixes uneven usage of `'` and `"` when containing args';
 * `path_from_history` – replaces not found path with similar absolute path from history;
 * `react_native_command_unrecognized` – fixes unrecognized `react-native` commands;

tests/rules/test_python_module_error.py

@@ -0,0 +1,60 @@
+import pytest
+from thefuck.rules.python_module_error import match, get_new_command
+from thefuck.types import Command
+from thefuck.shells import shell
+
+from collections import namedtuple
+
+ModuleNotFoundTest = namedtuple("ModuleNotFoundTest", ["script", "corrected"])
+
+positive_tests = [
+    ModuleNotFoundTest(
+        Command(
+            "python some_script.py",
+            """Traceback (most recent call last):
+  File "some_script.py", line 1, in <module>
+    import more_itertools
+ModuleNotFoundError: No module named 'more_itertools'""",
+        ),
+        shell.and_("pip install more_itertools", "python some_script.py"),
+    ),
+    ModuleNotFoundTest(
+        Command(
+            "./some_other_script.py",
+            """Traceback (most recent call last):
+  File "some_other_script.py", line 1, in <module>
+    from a_module import a_function
+ModuleNotFoundError: No module named 'a_module'""",
+        ),
+        shell.and_("pip install a_module", "./some_other_script.py"),
+    ),
+]
+
+negative_tests = [
+    ModuleNotFoundTest(Command("python hello_world.py", "Hello World"), None),
+    ModuleNotFoundTest(
+        Command(
+            "./hello_world.py",
+            """Traceback (most recent call last):
+  File "hello_world.py", line 1, in <module>
+    pritn("Hello World")
+NameError: name 'pritn' is not defined""",
+        ),
+        None,
+    ),
+]
+
+
+@pytest.mark.parametrize("test", negative_tests)
+def test_not_match(test):
+    assert not match(test.script)
+
+
+@pytest.mark.parametrize("test", positive_tests)
+def test_match(test):
+    assert match(test.script)
+
+
+@pytest.mark.parametrize("test", positive_tests)
+def test_get_new_command(test):
+    assert get_new_command(test.script) == test.corrected

thefuck/rules/python_module_error.py

@@ -0,0 +1,13 @@
+import re
+from thefuck.shells import shell
+
+MISSING_MODULE = r"ModuleNotFoundError: No module named '([^']+)'"
+
+
+def match(command):
+    return "ModuleNotFoundError: No module named '" in command.output
+
+
+def get_new_command(command):
+    missing_module = re.findall(MISSING_MODULE, command.output)[0]
+    return shell.and_("pip install {}".format(missing_module), command.script)