fix python keybindings, auto cleanup python imports on save

[emacs-init.git] / python / setup.el

diff --git a/python/setup.el b/python/setup.el
index f325608..ffb1b4d 100644 (file)
--- a/python/setup.el
+++ b/python/setup.el
@@ -4,4 +4,17 @@
   (add-to-list 'load-path self-dir))
 
 (load-library "auto-completion")
+
 (require 'init_python)
+
+(defun write-file-py-cleanup-imports ()
+  (save-excursion
+    (condition-case nil
+        (py-cleanup-imports)
+      (error . nil)))
+  nil)
+
+(defun python-init-auto-cleanup-imports-on-save ()
+  (add-hook 'write-file-functions 'write-file-py-cleanup-imports nil t))
+
+(add-hook 'python-mode-hook 'python-init-auto-cleanup-imports-on-save)