Initial commit

thefuck/main.py

@@ -0,0 +1,85 @@
+from collections import namedtuple
+from imp import load_source
+from pathlib import Path
+from os.path import expanduser
+from subprocess import Popen, PIPE
+import sys
+
+
+Command = namedtuple('Command', ('script', 'stdout', 'stderr'))
+Settings = namedtuple('Settings', ('rules',))
+Rule = namedtuple('Rule', ('match', 'get_new_command'))
+
+
+def setup_user_dir() -> Path:
+    """Returns user config dir, create it when it doesn't exists."""
+    user_dir = Path(expanduser('~/.thefuck'))
+    if not user_dir.is_dir():
+        user_dir.mkdir()
+        user_dir.joinpath('rules').mkdir()
+        user_dir.joinpath('settings.py').touch()
+    return user_dir
+
+
+def get_settings(user_dir: Path) -> Settings:
+    """Returns prepared settings module."""
+    settings = load_source('settings',
+                           str(user_dir.joinpath('settings.py')))
+    return Settings(getattr(settings, 'rules', None))
+
+
+def is_rule_enabled(settings: Settings, rule: Path) -> bool:
+    """Returns `True` when rule mentioned in `rules` or `rules`
+    isn't defined.
+
+    """
+    return settings.rules is None or rule.name[:-3] in settings.rules
+
+
+def load_rule(rule: Path) -> Rule:
+    """Imports rule module and returns it."""
+    rule_module = load_source(rule.name[:-3], str(rule))
+    return Rule(rule_module.match, rule_module.get_new_command)
+
+
+def get_rules(user_dir: Path, settings: Settings) -> [Rule]:
+    """Returns all enabled rules."""
+    bundled = Path(__file__).parent\
+                            .joinpath('rules')\
+                            .glob('*.py')
+    user = user_dir.joinpath('rules').glob('*.py')
+    return [load_rule(rule) for rule in list(bundled) + list(user)
+            if is_rule_enabled(settings, rule)]
+
+
+def get_command(args: [str]) -> Command:
+    """Creates command from `args` and executes it."""
+    script = ' '.join(args[1:])
+    result = Popen(script, shell=True, stdout=PIPE, stderr=PIPE)
+    return Command(script, result.stdout.read().decode(),
+                   result.stderr.read().decode())
+
+
+def get_matched_rule(command: Command, rules: [Rule]) -> Rule:
+    """Returns first matched rule for command."""
+    for rule in rules:
+        if rule.match(command):
+            return rule
+
+
+def run_rule(rule: Rule, command: Command):
+    """Runs command from rule for passed command."""
+    new_command = rule.get_new_command(command)
+    print(new_command)
+
+
+def main():
+    command = get_command(sys.argv)
+    user_dir = setup_user_dir()
+    settings = get_settings(user_dir)
+    rules = get_rules(user_dir, settings)
+    matched_rule = get_matched_rule(command, rules)
+    if matched_rule:
+        run_rule(matched_rule, command)
+    else:
+        print('echo No fuck given')

thefuck/rules/git_push.py

@@ -0,0 +1,8 @@
+def match(command):
+    return ('git' in command.script
+            and 'push' in command.script
+            and 'set-upstream' in command.stderr)
+
+
+def get_new_command(command):
+    return command.stderr.split('\n')[-3].strip()

thefuck/rules/sudo.py

@@ -0,0 +1,6 @@
+def match(command):
+    return 'permission denied' in command.stderr.lower()
+
+
+def get_new_command(command):
+    return 'sudo {}'.format(command.script)