240 lines
9.9 KiB
EmacsLisp
240 lines
9.9 KiB
EmacsLisp
;;; funcs.el --- Python Layer functions File for Spacemacs
|
||
;;
|
||
;; Copyright (c) 2012-2017 Sylvain Benner & Contributors
|
||
;;
|
||
;; Author: Sylvain Benner <sylvain.benner@gmail.com>
|
||
;; URL: https://github.com/syl20bnr/spacemacs
|
||
;;
|
||
;; This file is not part of GNU Emacs.
|
||
;;
|
||
;;; License: GPLv3
|
||
|
||
;; from http://pedrokroger.net/2010/07/configuring-emacs-as-a-python-ide-2/
|
||
(defun spacemacs/python-annotate-pdb ()
|
||
"Highlight break point lines."
|
||
(interactive)
|
||
(highlight-lines-matching-regexp "import \\(pdb\\|ipdb\\|pudb\\|wdb\\)")
|
||
(highlight-lines-matching-regexp "\\(pdb\\|ipdb\\|pudb\\|wdb\\).set_trace()"))
|
||
|
||
(defun spacemacs/pyenv-executable-find (command)
|
||
"Find executable taking pyenv shims into account."
|
||
(if (executable-find "pyenv")
|
||
(progn
|
||
(let ((pyenv-string (shell-command-to-string (concat "pyenv which " command))))
|
||
(unless (string-match "not found" pyenv-string)
|
||
(string-trim pyenv-string))))
|
||
(executable-find command)))
|
||
|
||
(defun spacemacs//python-setup-shell (&rest args)
|
||
(if (spacemacs/pyenv-executable-find "ipython")
|
||
(progn (setq python-shell-interpreter "ipython")
|
||
(if (version< (replace-regexp-in-string "\n$" "" (shell-command-to-string "ipython --version")) "5")
|
||
(setq python-shell-interpreter-args "-i")
|
||
(setq python-shell-interpreter-args "--simple-prompt -i")))
|
||
(progn
|
||
(setq python-shell-interpreter-args "-i")
|
||
(setq python-shell-interpreter "python"))))
|
||
|
||
(defun spacemacs//python-setup-hy (&rest args)
|
||
(setq hy-mode-inferior-lisp-command
|
||
(concat (or (spacemacs/pyenv-executable-find "hy") "hy")
|
||
" --spy")))
|
||
|
||
(defun spacemacs//python-setup-checkers (&rest args)
|
||
(when (fboundp 'flycheck-set-checker-executable)
|
||
(let ((pylint (spacemacs/pyenv-executable-find "pylint"))
|
||
(flake8 (spacemacs/pyenv-executable-find "flake8")))
|
||
(when pylint
|
||
(flycheck-set-checker-executable "python-pylint" pylint))
|
||
(when flake8
|
||
(flycheck-set-checker-executable "python-flake8" flake8)))))
|
||
|
||
(defun spacemacs/python-setup-everything (&rest args)
|
||
(apply 'spacemacs//python-setup-shell args)
|
||
(apply 'spacemacs//python-setup-hy args)
|
||
(apply 'spacemacs//python-setup-checkers args))
|
||
|
||
(defun spacemacs/python-toggle-breakpoint ()
|
||
"Add a break point, highlight it."
|
||
(interactive)
|
||
(let ((trace (cond ((spacemacs/pyenv-executable-find "wdb") "import wdb; wdb.set_trace()")
|
||
((spacemacs/pyenv-executable-find "ipdb") "import ipdb; ipdb.set_trace()")
|
||
((spacemacs/pyenv-executable-find "pudb") "import pudb; pudb.set_trace()")
|
||
((spacemacs/pyenv-executable-find "ipdb3") "import ipdb; ipdb.set_trace()")
|
||
((spacemacs/pyenv-executable-find "pudb3") "import pudb; pudb.set_trace()")
|
||
(t "import pdb; pdb.set_trace()")))
|
||
(line (thing-at-point 'line)))
|
||
(if (and line (string-match trace line))
|
||
(kill-whole-line)
|
||
(progn
|
||
(back-to-indentation)
|
||
(insert trace)
|
||
(insert "\n")
|
||
(python-indent-line)))))
|
||
|
||
;; from https://www.snip2code.com/Snippet/127022/Emacs-auto-remove-unused-import-statemen
|
||
(defun spacemacs/python-remove-unused-imports()
|
||
"Use Autoflake to remove unused function"
|
||
"autoflake --remove-all-unused-imports -i unused_imports.py"
|
||
(interactive)
|
||
(if (executable-find "autoflake")
|
||
(progn
|
||
(shell-command (format "autoflake --remove-all-unused-imports -i %s"
|
||
(shell-quote-argument (buffer-file-name))))
|
||
(revert-buffer t t t))
|
||
(message "Error: Cannot find autoflake executable.")))
|
||
|
||
(defun spacemacs//pyenv-mode-set-local-version ()
|
||
"Set pyenv version from \".python-version\" by looking in parent directories."
|
||
(interactive)
|
||
(let ((root-path (locate-dominating-file default-directory
|
||
".python-version")))
|
||
(when root-path
|
||
(let* ((file-path (expand-file-name ".python-version" root-path))
|
||
(version
|
||
(with-temp-buffer
|
||
(insert-file-contents-literally file-path)
|
||
(nth 0 (split-string (buffer-substring-no-properties
|
||
(line-beginning-position)
|
||
(line-end-position)))))))
|
||
(if (member version (pyenv-mode-versions))
|
||
(pyenv-mode-set version)
|
||
(message "pyenv: version `%s' is not installed (set by %s)"
|
||
version file-path))))))
|
||
|
||
(defun spacemacs//pyvenv-mode-set-local-virtualenv ()
|
||
"Set pyvenv virtualenv from \".venv\" by looking in parent directories."
|
||
(interactive)
|
||
(let ((root-path (locate-dominating-file default-directory
|
||
".venv")))
|
||
(when root-path
|
||
(let* ((file-path (expand-file-name ".venv" root-path))
|
||
(virtualenv
|
||
(with-temp-buffer
|
||
(insert-file-contents-literally file-path)
|
||
(buffer-substring-no-properties (line-beginning-position)
|
||
(line-end-position)))))
|
||
(pyvenv-workon virtualenv)))))
|
||
|
||
|
||
;; Tests
|
||
|
||
(defun spacemacs//disable-semantic-idle-summary-mode ()
|
||
"Disable semantic-idle-summary in Python mode.
|
||
Anaconda provides more useful information but can not do it properly
|
||
when this mode is enabled since the minibuffer is cleared all the time."
|
||
(semantic-idle-summary-mode 0))
|
||
|
||
(defun spacemacs//python-imenu-create-index-use-semantic-maybe ()
|
||
"Use semantic if the layer is enabled."
|
||
(setq imenu-create-index-function 'spacemacs/python-imenu-create-index))
|
||
|
||
;; fix for issue #2569 (https://github.com/syl20bnr/spacemacs/issues/2569) and
|
||
;; Emacs 24.5 and older. use `semantic-create-imenu-index' only when
|
||
;; `semantic-mode' is enabled, otherwise use `python-imenu-create-index'
|
||
(defun spacemacs/python-imenu-create-index ()
|
||
(if (bound-and-true-p semantic-mode)
|
||
(semantic-create-imenu-index)
|
||
(python-imenu-create-index)))
|
||
|
||
(defun spacemacs//python-get-main-testrunner ()
|
||
"Get the main test runner."
|
||
(if (listp python-test-runner) (car python-test-runner) python-test-runner))
|
||
|
||
(defun spacemacs//python-get-secondary-testrunner ()
|
||
"Get the secondary test runner"
|
||
(cdr (assoc (spacemacs//python-get-main-testrunner) '((pytest . nose)
|
||
(nose . pytest)))))
|
||
|
||
(defun spacemacs//python-call-correct-test-function (arg funcalist)
|
||
"Call a test function based on the chosen test framework.
|
||
ARG is the universal-argument which chooses between the main and
|
||
the secondary test runner. FUNCALIST is an alist of the function
|
||
to be called for each testrunner. "
|
||
(let* ((test-runner (if arg
|
||
(spacemacs//python-get-secondary-testrunner)
|
||
(spacemacs//python-get-main-testrunner)))
|
||
(test-function (assq test-runner funcalist)))
|
||
(if test-function
|
||
(funcall (cdr (assoc test-runner funcalist)))
|
||
(user-error "This test function is not available with the `%S' runner."
|
||
test-runner))))
|
||
|
||
(defun spacemacs/python-test-all (arg)
|
||
"Run all tests."
|
||
(interactive "P")
|
||
(spacemacs//python-call-correct-test-function arg '((pytest . pytest-all)
|
||
(nose . nosetests-all))))
|
||
|
||
(defun spacemacs/python-test-pdb-all (arg)
|
||
"Run all tests in debug mode."
|
||
(interactive "P")
|
||
(spacemacs//python-call-correct-test-function arg '((pytest . pytest-pdb-all)
|
||
(nose . nosetests-pdb-all))))
|
||
|
||
(defun spacemacs/python-test-module (arg)
|
||
"Run all tests in the current module."
|
||
(interactive "P")
|
||
(spacemacs//python-call-correct-test-function arg '((pytest . pytest-module)
|
||
(nose . nosetests-module))))
|
||
|
||
(defun spacemacs/python-test-pdb-module (arg)
|
||
"Run all tests in the current module in debug mode."
|
||
(interactive "P")
|
||
(spacemacs//python-call-correct-test-function
|
||
arg
|
||
'((pytest . pytest-pdb-module)
|
||
(nose . nosetests-pdb-module))))
|
||
|
||
(defun spacemacs/python-test-suite (arg)
|
||
"Run all tests in the current suite."
|
||
(interactive "P")
|
||
(spacemacs//python-call-correct-test-function arg '((nose . nosetests-suite))))
|
||
|
||
(defun spacemacs/python-test-pdb-suite (arg)
|
||
"Run all tests in the current suite in debug mode."
|
||
(interactive "P")
|
||
(spacemacs//python-call-correct-test-function arg '((nose . nosetests-pdb-suite))))
|
||
|
||
(defun spacemacs/python-test-one (arg)
|
||
"Run current test."
|
||
(interactive "P")
|
||
(spacemacs//python-call-correct-test-function arg '((pytest . pytest-one)
|
||
(nose . nosetests-one))))
|
||
|
||
(defun spacemacs/python-test-pdb-one (arg)
|
||
"Run current test in debug mode."
|
||
(interactive "P")
|
||
(spacemacs//python-call-correct-test-function arg '((pytest . pytest-pdb-one)
|
||
(nose . nosetests-pdb-one))))
|
||
|
||
(defun spacemacs//bind-python-testing-keys ()
|
||
"Bind the keys for testing in Python."
|
||
(spacemacs/declare-prefix-for-mode 'python-mode "mt" "test")
|
||
(spacemacs/set-leader-keys-for-major-mode 'python-mode
|
||
"tA" 'spacemacs/python-test-pdb-all
|
||
"ta" 'spacemacs/python-test-all
|
||
"tB" 'spacemacs/python-test-pdb-module
|
||
"tb" 'spacemacs/python-test-module
|
||
"tT" 'spacemacs/python-test-pdb-one
|
||
"tt" 'spacemacs/python-test-one
|
||
"tM" 'spacemacs/python-test-pdb-module
|
||
"tm" 'spacemacs/python-test-module
|
||
"tS" 'spacemacs/python-test-pdb-suite
|
||
"ts" 'spacemacs/python-test-suite))
|
||
|
||
(defun spacemacs//python-sort-imports ()
|
||
;; py-isort-before-save checks the major mode as well, however we can prevent
|
||
;; it from loading the package unnecessarily by doing our own check
|
||
(when (and python-sort-imports-on-save
|
||
(derived-mode-p 'python-mode))
|
||
(py-isort-before-save)))
|
||
|
||
|
||
;;* Anaconda
|
||
(defun spacemacs/anaconda-view-forward-and-push ()
|
||
"Find next button and hit RET"
|
||
(interactive)
|
||
(forward-button 1)
|
||
(call-interactively #'push-button))
|