wting_autojump/tools/autojump_ipython.py

# This module was contributed by Mario Pastorelli <pastorelli.mario@gmail.com>
# It is released in the public domain

# This tool provides "j" for ipython
# To use it, copy it in your ~/.ipython directory
# and add the following line to ipy_user_conf.py:
# import autojump_ipython

import os
import subprocess as sub
from IPython.ipapi import get
from IPython.iplib import InteractiveShell

ip = get()


def magic_j(self, parameter_s=''):
    cmd = ['autojump'] + parameter_s.split()
    # print 'executing autojump with args %s' % str(cmd)
    newpath = sub.Popen(cmd, stdout=sub.PIPE, shell=False).communicate(
    )[0][:-1]  # delete last '\n'
    # print 'Autojump answer: \'%s\'' % newpath
    if newpath:
        ip.magic('cd \'%s\'' % newpath)


def cd_decorator(f):
    def autojump_cd_monitor(self, parameter_s=''):
        f(self, parameter_s)
        sub.call(['autojump', '-a', os.getcwd()])
    return autojump_cd_monitor

# Add the new magic function to the class dict and decorate magic_cd:
InteractiveShell.magic_j = magic_j
InteractiveShell.magic_cd = cd_decorator(InteractiveShell.magic_cd)

# And remove the global name to keep global namespace clean.
del magic_j
del cd_decorator