Set pythonpath process property for remote processes.

Author: proofit404

pythonic.el

@@ -90,6 +90,12 @@ It will use `python-shell-exec-path' for PATH variable,
     (pythonic-set-path-variable)
     (pythonic-set-extra-variables)))
 
+(defun pythonic-get-pythonpath ()
+  "Get appropriate PYTHONPATH variable for pythonic process."
+  (if (pythonic-remote-p)
+      (pythonic-get-pythonpath-variable-tramp)
+    (pythonic-get-pythonpath-variable)))
+
 (defun pythonic-get-pythonpath-variable ()
   "Get PYTHONPATH variable from `python-shell-extra-pythonpaths' variable."
   (s-join path-separator
@@ -99,26 +105,30 @@ It will use `python-shell-exec-path' for PATH variable,
                (-remove 's-blank?)
                (-distinct))))
 
+(defun pythonic-get-pythonpath-variable-tramp ()
+  "Get PYTHONPATH variable form `python-shell-extra-pythonpaths' on the remote host."
+  (let ((connection (tramp-dissect-file-name (pythonic-tramp-connection))))
+    (s-join path-separator
+            (--> (progn
+                   (tramp-send-command connection "echo $PYTHONPATH")
+                   (with-current-buffer (tramp-get-connection-buffer connection)
+                     (buffer-string)))
+                 (s-trim it)
+                 (s-split ":" it t)
+                 (--remove (member it python-shell-extra-pythonpaths) it)
+                 (append python-shell-extra-pythonpaths it)))))
+
 (defun pythonic-set-pythonpath-variable ()
   "Set PYTHONPATH variable from `python-shell-extra-pythonpaths' variable."
   (setenv "PYTHONPATH" (pythonic-get-pythonpath-variable)))
 
 (defun pythonic-set-pythonpath-variable-tramp ()
   "Set PYTHONPATH variable from `python-shell-extra-pythonpaths' variable on remote host."
-  (let ((connection (tramp-dissect-file-name (pythonic-tramp-connection))))
-    (tramp-send-command
-     connection
-     (format
-      "export PYTHONPATH=%s"
-      (s-join ":"
-              (--> (progn
-                     (tramp-send-command connection "echo $PYTHONPATH")
-                     (with-current-buffer (tramp-get-connection-buffer connection)
-                       (buffer-string)))
-                   (s-trim it)
-                   (s-split ":" it t)
-                   (--remove (member it python-shell-extra-pythonpaths) it)
-                   (append python-shell-extra-pythonpaths it)))))))
+  (tramp-send-command
+   (tramp-dissect-file-name (pythonic-tramp-connection))
+   (format
+    "export PYTHONPATH=%s"
+    (pythonic-get-pythonpath-variable-tramp))))
 
 (defun pythonic-get-path ()
   "Return appropriate PATH variable for pythonic process."
@@ -199,7 +209,7 @@ process flag."
       (process-put process 'default-directory default-directory)
       (process-put process 'environment python-shell-process-environment)
       (process-put process 'path (pythonic-get-path))
-      (process-put process 'pythonpath (pythonic-get-pythonpath-variable))
+      (process-put process 'pythonpath (pythonic-get-pythonpath))
       process)))
 
 ;;;###autoload

test/pythonic-test.el

@@ -320,7 +320,6 @@ remote host."
   "Set `python-shell-exec-path' as `path' process property on remote process."
   (unwind-protect
       (let* ((python-shell-interpreter "/ssh:test@localhost:/path/to/the/python")
-             (home (f-expand "~"))
              (python-shell-exec-path '("/home/test/bin")))
         (should (equal "/home/test/bin:/bin:/usr/bin:/sbin:/usr/sbin:/usr/local/bin:/usr/local/sbin"
                        (process-get
@@ -340,6 +339,22 @@ remote host."
                     (start-pythonic :process "out" :args '("-V"))
                     'pythonpath)))))
 
+(ert-deftest test-start-pythonic-pythonpath-property-tramp ()
+  "Set `python-shell-extra-pythonpaths' as `pythonpath' process property on remote host.
+Respect remote PYTHONPATH value."
+  (unwind-protect
+      (let* ((python-shell-interpreter "/ssh:test@localhost:/path/to/the/python")
+             (python-shell-extra-pythonpaths '("/home/test/modules")))
+        (tramp-send-command ["ssh" "test" "localhost" "" nil]
+                            "export PYTHONPATH=/home/me/modules")
+        (should (equal "/home/test/modules:/home/me/modules"
+                       (process-get
+                        (start-pythonic :process "out" :args '("-V"))
+                        'pythonpath))))
+    (kill-buffer "*tramp/ssh test@localhost*")
+    (setq tramp-current-connection)
+    (sleep-for 0.5)))
+
 (ert-deftest test-start-pythonic-default-directory-property ()
   "Set `pythonic-default-directory' result as `default-directory' process property."
   (let ((default-directorys "~"))