Futurize (stage 1) add #33

CHANGELOG.md

@@ -2,7 +2,11 @@
 
 ## master (unreleased)
 
-*
+* Bundle `refactoring` enhanced with `future`, `libfuturize`, `libpasteurize` modules
+* `script/refactoring/futurize.py` introduced (see [Python Future](https://github.com/PythonCharmers/python-future))
+    * Equivalent of `futurize -1 --all-imports -n -w --add-suffix 3 $editor.get_path()` (Stage 1 only)
+    * When futurizer ends, editor text is replaced with content of the `.py3` and `.py3` is trashed
+    * You can run futurizer script with `Cmd Option F`
 
 ## 1.4.0 (2017-11-01)
 

blackmamba/__init__.py

@@ -142,6 +142,8 @@ def _register_default_key_commands():
          'refactoring/organize_imports.py', 'Refactor - Organize imports'),
         ('E', UIKeyModifier.command | UIKeyModifier.alternate,
          'refactoring/expand_star_imports.py', 'Refactor - Expand star imports'),
+        ('F', UIKeyModifier.command | UIKeyModifier.alternate,
+         'refactoring/futurize.py', 'Refactor - Futurize'),
 
         # Still keyboard only
         ('0', UIKeyModifier.command,

blackmamba/bundle.py

@@ -8,7 +8,7 @@
 _BUNDLES = {
     'analyze': ['mccabe', 'flake8', 'pycodestyle', 'pyflakes', 'setuptools'],
     'docutils': ['docutils'],
-    'refactoring': ['rope']
+    'refactoring': ['rope', 'future', 'libfuturize', 'libpasteurize']
 }
 
 _BUNDLED_MODULES_PATH = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'lib'))

blackmamba/script/refactoring/futurize.py

@@ -0,0 +1,60 @@
+#!python3
+
+import console
+import os
+import unittest.mock as mock
+
+import blackmamba.ide.source as source
+import blackmamba.ide.tab as tab
+import editor
+
+
+_SUFFIX = '3'
+
+
+def _replace_file_content(path):
+    futurized_path = '{}{}'.format(path, _SUFFIX)
+
+    if not os.path.exists(futurized_path):
+        # This function is not called unless result is 0, thus if file doesn't
+        # exist, we can assume that there's no need to change anything -> success
+        return True
+
+    line_number = source.get_line_number()
+    content = editor.get_text()
+
+    if not line_number or content is None:
+        os.remove(futurized_path)
+        return False
+
+    new_content = open(futurized_path, 'r').read()
+    editor.replace_text(0, len(content) - 1, new_content)
+    source.scroll_to_line(line_number)
+    os.remove(futurized_path)
+    return True
+
+
+# I'm lazy one, cripples Pythonista logging
+@mock.patch('logging.basicConfig')
+def main(mock):
+    import libfuturize.main
+
+    path = editor.get_path()
+    if not path or not path.endswith('.py'):
+        console.hud_alert('Not a Python file', 'error')
+        return
+
+    tab.save()
+    args = ['-1', '-n', '-w', '--all-imports', '--add-suffix', _SUFFIX, path]
+
+    result = libfuturize.main.main(args)
+    if result == 0 and _replace_file_content(path):
+        console.hud_alert('Futurized')
+    else:
+        console.hud_alert('Failed to futurize', 'error')
+
+
+if __name__ == '__main__':
+    from blackmamba.bundle import bundle
+    with bundle('refactoring'):
+        main()