lang/org: major refactor of +ipython feature
This commit is contained in:
Henrik Lissner
parent
98b439e3bb
commit
ebfc5648ef
4 changed files with 112 additions and 158 deletions
|
|
@ -1,5 +1,6 @@
|
|||
;;; lang/org/autoload/ipython.el -*- lexical-binding: t; -*-
|
||||
;; * remote
|
||||
;;;###if (featurep! +ipython)
|
||||
|
||||
;;;###autoload
|
||||
(defun +org*org-babel-ipython-initiate-session (&optional session params)
|
||||
"Create a session named SESSION according to PARAMS."
|
||||
|
|
@ -7,7 +8,7 @@
|
|||
(error
|
||||
"ob-ipython currently only supports evaluation using a session.
|
||||
Make sure your src block has a :session param.")
|
||||
(when (not (s-ends-with-p ".json" session))
|
||||
(when (not (string-suffix-p ".json" session t))
|
||||
(ob-ipython--create-kernel
|
||||
(ob-ipython--normalize-session
|
||||
session)
|
||||
|
|
@ -22,50 +23,34 @@ Make sure your src block has a :session param.")
|
|||
"Create repl based on NAME and PARAMS.
|
||||
If PARAMS specifies remote kernel, copy the kernel config from remote server and
|
||||
create a repl connecting to remote session."
|
||||
(let ((cmd (s-join
|
||||
" "
|
||||
(ob-ipython--kernel-repl-cmd
|
||||
name))))
|
||||
(if (string= "default" name)
|
||||
(progn
|
||||
(run-python cmd nil nil)
|
||||
(format
|
||||
"*%s*"
|
||||
python-shell-buffer-name))
|
||||
(if (string-match
|
||||
"^remote-.*ssh.json"
|
||||
name)
|
||||
(when (not (ignore-errors
|
||||
(process-live-p
|
||||
(get-process
|
||||
(format
|
||||
"Python:ob-ipython-%s"
|
||||
name)))))
|
||||
(let* ((remote (s-split "-" name))
|
||||
(remote-host (nth 1 remote))
|
||||
(remote-session (nth 3 remote)))
|
||||
(+org/ob-ipython-generate-local-path-from-remote
|
||||
remote-session
|
||||
remote-host
|
||||
params)))
|
||||
(let* ((process-name (format
|
||||
"Python:ob-ipython-%s"
|
||||
name))
|
||||
(python-shell-prompt-detect-enabled nil)
|
||||
(python-shell-completion-native-enable nil)
|
||||
(buf (python-shell-make-comint
|
||||
cmd
|
||||
process-name
|
||||
t))
|
||||
(dir (cdr (assoc :pydir params))))
|
||||
(if dir
|
||||
(with-current-buffer
|
||||
buf
|
||||
(setq-local
|
||||
default-directory
|
||||
dir)))
|
||||
(sleep-for 1)
|
||||
(format "*%s*" process-name))))))
|
||||
(let ((cmd (string-join (ob-ipython--kernel-repl-cmd name) " ")))
|
||||
(cond ((string= "default" name)
|
||||
(run-python cmd nil nil)
|
||||
(format "*%s*" python-shell-buffer-name))
|
||||
((string-match "^remote-.*ssh.json" name)
|
||||
(when (not (ignore-errors
|
||||
(process-live-p
|
||||
(get-process
|
||||
(format
|
||||
"Python:ob-ipython-%s"
|
||||
name)))))
|
||||
(let* ((remote (s-split "-" name))
|
||||
(remote-host (nth 1 remote))
|
||||
(remote-session (nth 3 remote)))
|
||||
(+org/ob-ipython-generate-local-path-from-remote
|
||||
remote-session
|
||||
remote-host
|
||||
params))))
|
||||
((let* ((process-name (format "Python:ob-ipython-%s" name))
|
||||
(python-shell-prompt-detect-enabled nil)
|
||||
(python-shell-completion-native-enable nil)
|
||||
(buf (python-shell-make-comint cmd process-name t))
|
||||
(dir (cdr (assoc :pydir params))))
|
||||
(if dir
|
||||
(with-current-buffer buf
|
||||
(setq-local default-directory dir)))
|
||||
(sleep-for 1)
|
||||
(format "*%s*" process-name))))))
|
||||
|
||||
;;;###autoload
|
||||
(defun +org*org-babel-execute:ipython (body params)
|
||||
|
|
@ -73,17 +58,11 @@ create a repl connecting to remote session."
|
|||
This function is called by `org-babel-execute-src-block'."
|
||||
(message default-directory)
|
||||
(let ((session (cdr (assoc :session params))))
|
||||
(org-babel-ipython-initiate-session
|
||||
session
|
||||
params))
|
||||
(org-babel-ipython-initiate-session session params))
|
||||
(ob-ipython--clear-output-buffer)
|
||||
(if (cdr (assoc :async params))
|
||||
(ob-ipython--execute-async
|
||||
body
|
||||
params)
|
||||
(ob-ipython--execute-sync
|
||||
body
|
||||
params)))
|
||||
(ob-ipython--execute-async body params)
|
||||
(ob-ipython--execute-sync body params)))
|
||||
|
||||
;;;###autoload
|
||||
(defun +org/ob-ipython-generate-local-path-from-remote (session host params)
|
||||
|
|
@ -92,7 +71,10 @@ This function is called by `org-babel-execute-src-block'."
|
|||
(substring (shell-command-to-string (concat "ssh " host " jupyter --runtime-dir")) 0 -1))
|
||||
(runtime-file (concat runtime-dir "/" "kernel-" session ".json"))
|
||||
(tramp-path (concat "/ssh:" host ":" runtime-file))
|
||||
(tramp-copy (concat +ob-ipython-local-runtime-dir "/remote-" host "-kernel-" session ".json"))
|
||||
(tramp-copy (concat (or +ob-ipython-local-runtime-dir
|
||||
(substring (shell-command-to-string "jupyter --runtime-dir")
|
||||
0 -1))
|
||||
"/remote-" host "-kernel-" session ".json"))
|
||||
(local-path
|
||||
(concat
|
||||
"Python:ob-ipython-"
|
||||
|
|
@ -105,21 +87,17 @@ This function is called by `org-babel-execute-src-block'."
|
|||
(python-shell-prompt-detect-enabled nil)
|
||||
(python-shell-completion-native-enable nil)
|
||||
(buf (python-shell-make-comint
|
||||
(concat
|
||||
ob-ipython-command
|
||||
" console --simple-prompt --existing "
|
||||
tramp-copy
|
||||
" --ssh "
|
||||
host)
|
||||
(concat ob-ipython-command
|
||||
" console --simple-prompt --existing "
|
||||
tramp-copy " --ssh " host)
|
||||
(concat "" local-path)
|
||||
t))
|
||||
(proc (get-buffer-process buf))
|
||||
(dir (cdr (assoc :pydir params))))
|
||||
(sleep-for 3)
|
||||
(if dir
|
||||
(with-current-buffer
|
||||
buf
|
||||
(setq-local default-directory dir)))
|
||||
(when dir
|
||||
(with-current-buffer buf
|
||||
(setq-local default-directory dir)))
|
||||
(format "*%s*" proc))))
|
||||
|
||||
;; * org-src-edit
|
||||
|
|
@ -127,51 +105,31 @@ This function is called by `org-babel-execute-src-block'."
|
|||
(defun +org*org-babel-edit-prep:ipython (info)
|
||||
(let* ((params (nth 2 info))
|
||||
(session (cdr (assoc :session params))))
|
||||
(org-babel-ipython-initiate-session
|
||||
session
|
||||
params))
|
||||
(org-babel-ipython-initiate-session session params))
|
||||
;; Support for python.el's "send-code" commands within edit buffers.
|
||||
(setq-local
|
||||
python-shell-buffer-name
|
||||
(format
|
||||
"Python:ob-ipython-%s"
|
||||
(->>
|
||||
info
|
||||
(nth 2)
|
||||
(assoc :session)
|
||||
cdr ob-ipython--normalize-session)))
|
||||
(setq-local
|
||||
default-directory
|
||||
(format
|
||||
"%s"
|
||||
(->>
|
||||
info
|
||||
(nth 2)
|
||||
(assoc :pydir)
|
||||
cdr ob-ipython--normalize-session)))
|
||||
(setq-local python-shell-buffer-name
|
||||
(format "Python:ob-ipython-%s"
|
||||
(->> info (nth 2) (assoc :session)
|
||||
cdr ob-ipython--normalize-session)))
|
||||
(setq-local default-directory
|
||||
(format "%s"
|
||||
(->> info (nth 2) (assoc :pydir)
|
||||
cdr ob-ipython--normalize-session)))
|
||||
(ob-ipython-mode 1)
|
||||
;; hack on company mode to use company-capf rather than company-anaconda
|
||||
(when (featurep! :completion company)
|
||||
(setq-local
|
||||
company-backends
|
||||
'(company-capf
|
||||
company-dabbrev
|
||||
company-files
|
||||
company-yasnippet))
|
||||
(setq-local
|
||||
company-idle-delay
|
||||
nil))
|
||||
(setq-local company-backends
|
||||
'(company-capf
|
||||
company-dabbrev
|
||||
company-files
|
||||
company-yasnippet))
|
||||
(setq-local company-idle-delay nil))
|
||||
(when (featurep 'lpy)
|
||||
(setq lispy-python-proc
|
||||
(format
|
||||
"Python:ob-ipython-%s"
|
||||
(->>
|
||||
info
|
||||
(nth 2)
|
||||
(assoc :session)
|
||||
cdr ob-ipython--normalize-session)))
|
||||
(setq lispy--python-middleware-loaded-p
|
||||
nil)
|
||||
(format "Python:ob-ipython-%s"
|
||||
(->> info (nth 2) (assoc :session)
|
||||
cdr ob-ipython--normalize-session))
|
||||
lispy--python-middleware-loaded-p nil)
|
||||
(lispy--python-middleware-load)))
|
||||
|
||||
;; * retina
|
||||
|
|
@ -180,11 +138,11 @@ This function is called by `org-babel-execute-src-block'."
|
|||
"Return the name of high-resolution image file for FILENAME.
|
||||
The optional arg SCALE is scale factor, and defaults to 2."
|
||||
(let ((pos (or (string-match "\\.[^./]*\\'" filename) (length filename))))
|
||||
(format
|
||||
"%s@%dx%s"
|
||||
(substring filename 0 pos)
|
||||
(or scale 2)
|
||||
(substring filename pos))))
|
||||
(format "%s@%dx%s"
|
||||
(substring filename 0 pos)
|
||||
(or scale 2)
|
||||
(substring filename pos))))
|
||||
|
||||
;;;###autoload
|
||||
(defun +org*ob-ipython--write-base64-string (oldfunc &rest args)
|
||||
(let ((file (car args))
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue