From a0949b1102018288dc0d801d120f64bcc16559a4 Mon Sep 17 00:00:00 2001
From: Cami Diez <diezcami@gmail.com>
Date: Sun, 21 Jun 2015 09:24:27 +0800
Subject: [PATCH] Added Python Compile Rule

---
 tests/rules/test_python_compile.py | 17 +++++++++++++++++
 thefuck/rules/python_compile.py    | 15 +++++++++++++++
 2 files changed, 32 insertions(+)
 create mode 100644 tests/rules/test_python_compile.py
 create mode 100644 thefuck/rules/python_compile.py

diff --git a/tests/rules/test_python_compile.py b/tests/rules/test_python_compile.py
new file mode 100644
index 00000000..d7bf9889
--- /dev/null
+++ b/tests/rules/test_python_compile.py
@@ -0,0 +1,17 @@
+import pytest
+from thefuck.rules.python_compile 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
diff --git a/thefuck/rules/python_compile.py b/thefuck/rules/python_compile.py
new file mode 100644
index 00000000..39adcafa
--- /dev/null
+++ b/thefuck/rules/python_compile.py
@@ -0,0 +1,15 @@
+# Appends .py when compiling 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'