emacs-ipython-notebook/ein-pytools.el
2012-06-07 00:38:25 +02:00

93 lines
3.1 KiB
EmacsLisp

;;; ein-pytools.el --- Python tools build on top of kernel
;; Copyright (C) 2012- Takafumi Arakaki
;; Author: Takafumi Arakaki
;; This file is NOT part of GNU Emacs.
;; ein-pytools.el is free software: you can redistribute it and/or modify
;; it under the terms of the GNU General Public License as published by
;; the Free Software Foundation, either version 3 of the License, or
;; (at your option) any later version.
;; ein-pytools.el is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
;; GNU General Public License for more details.
;; You should have received a copy of the GNU General Public License
;; along with ein-pytools.el. If not, see <http://www.gnu.org/licenses/>.
;;; Commentary:
;;
;;; Code:
(eval-when-compile (require 'cl))
(require 'ein-kernel)
(eval-when-compile (defvar ein:notebook)
(defvar ein:@connect))
(declare-function ein:$notebook-kernel "ein-notebook")
(declare-function ein:connect-get-kernel "ein-connect")
(defun ein:pytools-get-kernel ()
(cond
(ein:notebook (ein:$notebook-kernel ein:notebook))
(ein:@connect (ein:connect-get-kernel))))
(defun ein:pytools-setup-hooks (kernel)
(push (cons #'ein:pytools-add-sys-path kernel)
(ein:$kernel-after-start-hook kernel)))
(defun ein:pytools-add-sys-path (kernel)
(ein:kernel-execute
kernel
(format "__import__('sys').path.append('%s')" ein:source-dir)))
(defun ein:pytools-jump-to-source (kernel object)
(ein:log 'info "Jumping to the source of %s..." object)
(ein:kernel-execute
kernel
(format "__import__('ein').find_source('%s')" object)
(list
:output
(cons
(lambda (object msg-type content)
(ein:case-equal msg-type
(("stream")
(ein:aif (plist-get content :data)
(if (or (string-match "^WARNING: .*" it)
(string-match
"^Traceback (most recent call last):\n" it)
(string-match "^.*<ipython-input-[^>\n]+>\n" it))
(ein:log 'info
"Jumping to the source of %s...Not found" object)
(let* ((filename-lineno (split-string it "\n"))
(filename (car filename-lineno))
(lineno (string-to-number (cadr filename-lineno))))
(unless (equal filename "")
(find-file-other-window filename)
(goto-char (point-min))
(forward-line (1- lineno))
(ein:log 'info
"Jumping to the source of %s...Done" object))))))
(("pyerr")
(ein:log 'info
"Jumping to the source of %s...Not found" object))))
object))))
(defun ein:pytools-jump-to-source-command ()
(interactive)
(let ((kernel (ein:pytools-get-kernel))
(object (ein:object-at-point)))
(assert (ein:kernel-ready-p kernel) nil "Kernel is not ready.")
(assert object nil "Object at point not found.")
(ein:pytools-jump-to-source kernel object)))
(provide 'ein-pytools)
;;; ein-pytools.el ends here