mirror of
https://github.com/wting/autojump
synced 2024-10-27 20:34:07 +00:00
290 lines
9.1 KiB
Python
Executable File
290 lines
9.1 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
import subprocess
|
|
import cPickle
|
|
import os.path
|
|
import os
|
|
import sys
|
|
import pygtk
|
|
pygtk.require('2.0')
|
|
import gtk
|
|
from autojump import open_dic,get_dic_file
|
|
|
|
defaults={}
|
|
actions={}
|
|
|
|
#directory finding helpers, conforming to the XDG Base Directory Specification
|
|
def data_dir():
|
|
xdg_data_dir = os.environ.get('XDG_DATA_HOME') or os.path.join(os.environ['HOME'], '.local', 'share')
|
|
return os.path.join(xdg_data_dir, 'autojump')
|
|
|
|
def config_dir():
|
|
xdg_config_dir = os.environ.get('XDG_CONFIG_HOME') or os.path.join(os.environ['HOME'], '.config')
|
|
return os.path.join(xdg_config_dir, 'autojump')
|
|
|
|
#decorator truff
|
|
def action(validator,name=None):
|
|
if name is None:
|
|
def wrapper(action):
|
|
actions[action.__name__]=(action,validator)
|
|
else:
|
|
def wrapper(action):
|
|
actions[name]=(action,validator)
|
|
return wrapper
|
|
|
|
#validator helper
|
|
def has_child_dir(dirname,recursion=0):
|
|
def wrapper(path):
|
|
k=recursion
|
|
ii=""
|
|
while k>=0:
|
|
if os.path.isdir(os.path.join(path,ii,dirname)): return True
|
|
k-=1
|
|
ii=os.path.join("..",ii)
|
|
return False
|
|
return wrapper
|
|
|
|
|
|
def has_child_file(filename):
|
|
def wrapper(path):
|
|
return os.path.isfile(os.path.join(path,filename))
|
|
return wrapper
|
|
|
|
def load_paths(maxpath=10):
|
|
path_dict=open_dic(get_dic_file())
|
|
path_dict=path_dict.items()
|
|
path_dict.sort(key=lambda x: x[1],reverse=True)
|
|
return [path for path,score in path_dict[:maxpath]]
|
|
|
|
def load_settings_file():
|
|
filename = os.path.join(config_dir(), 'jumpapplet_py')
|
|
#migration from old location
|
|
old_filename = os.path.join(os.environ['HOME'], '.jumpapplet_py')
|
|
if not os.path.exists(filename) and os.path.exists(old_filename):
|
|
os.rename(old_filename, filename)
|
|
|
|
print "loading settings"
|
|
global defaults
|
|
dic_file = filename
|
|
|
|
try:
|
|
aj_file=open(dic_file,'r')
|
|
defaults=cPickle.load(aj_file)
|
|
aj_file.close()
|
|
except IOError:
|
|
print "no config file"
|
|
pass
|
|
|
|
if not "terminal" in defaults: defaults["terminal"]="gnome-terminal"
|
|
if not "navigator" in defaults: defaults["navigator"]="nautilus"
|
|
if not "maxpath" in defaults: defaults["maxpath"]=15
|
|
if not "invert" in defaults: defaults["invert"]=False
|
|
if not "collapse" in defaults: defaults["collapse"]=True
|
|
|
|
create_actions()
|
|
|
|
def save_settings_file(filename=None):
|
|
directory = config_dir()
|
|
if not filename:
|
|
filename = os.path.join(directory, 'jumpapplet_py')
|
|
|
|
if not os.path.exists(directory):
|
|
os.makedirs(directory)
|
|
|
|
print "saving settings"
|
|
dic_file= filename
|
|
|
|
aj_file=open(dic_file,'w')
|
|
cPickle.dump(defaults,aj_file)
|
|
aj_file.close()
|
|
|
|
|
|
def get_actions(path):
|
|
return [(name,action) for name,(action,validator) in actions.items() if validator(path)]
|
|
|
|
def popup(sender,button,activation):
|
|
paths=load_paths(maxpath=defaults["maxpath"])
|
|
|
|
if defaults["collapse"]:
|
|
def collapse_home(path):
|
|
return path.replace(os.path.expanduser('~'),"~")
|
|
else:
|
|
def collapse_home(path):
|
|
return path
|
|
|
|
menu=gtk.Menu()
|
|
if defaults["invert"]:
|
|
item=gtk.ImageMenuItem(stock_id=gtk.STOCK_QUIT)
|
|
item.connect("activate",quit)
|
|
menu.append(item)
|
|
item=gtk.ImageMenuItem(stock_id=gtk.STOCK_PREFERENCES)
|
|
item.connect("activate",settings)
|
|
menu.append(item)
|
|
|
|
menu.append(gtk.SeparatorMenuItem())
|
|
|
|
for path in reversed(paths):
|
|
actions=get_actions(path)
|
|
if not actions: continue
|
|
|
|
item=gtk.MenuItem(collapse_home(path),use_underline=False)
|
|
submenu=gtk.Menu()
|
|
item.set_submenu(submenu)
|
|
for name,action in actions:
|
|
subitem=gtk.MenuItem(name)
|
|
subitem.connect("activate",action,path)
|
|
submenu.append(subitem)
|
|
menu.append(item)
|
|
else:
|
|
for path in paths:
|
|
actions=get_actions(path)
|
|
if not actions: continue
|
|
|
|
item=gtk.MenuItem(collapse_home(path),use_underline=False)
|
|
submenu=gtk.Menu()
|
|
item.set_submenu(submenu)
|
|
for name,action in actions:
|
|
subitem=gtk.MenuItem(name)
|
|
subitem.connect("activate",action,path)
|
|
submenu.append(subitem)
|
|
menu.append(item)
|
|
|
|
menu.append(gtk.SeparatorMenuItem())
|
|
|
|
item=gtk.ImageMenuItem(stock_id=gtk.STOCK_PREFERENCES)
|
|
item.connect("activate",settings)
|
|
menu.append(item)
|
|
item=gtk.ImageMenuItem(stock_id=gtk.STOCK_QUIT)
|
|
item.connect("activate",quit)
|
|
menu.append(item)
|
|
|
|
menu.show_all()
|
|
menu.popup(None,None,gtk.status_icon_position_menu,button,activation,sender)
|
|
|
|
def settings(sender):
|
|
window=gtk.Dialog("jump applet preferences",None,gtk.DIALOG_MODAL|gtk.DIALOG_DESTROY_WITH_PARENT,(gtk.STOCK_SAVE,gtk.RESPONSE_OK,gtk.STOCK_CANCEL,gtk.RESPONSE_CANCEL))
|
|
window.set_border_width(3)
|
|
window.set_resizable(False)
|
|
if os.path.isfile("icon.png"): window.set_icon_from_file("icon.png")
|
|
elif os.path.isfile("/usr/share/autojump/icon.png"): window.set_icon_from_file("/usr/share/autojump/icon.png")
|
|
|
|
vbox=gtk.Table(5,2)
|
|
vbox.set_row_spacings(3)
|
|
window.get_child().add(vbox)
|
|
def add_string_setting(name,label,nsettings):
|
|
label=gtk.Label(label+' ')
|
|
label.set_alignment(1.,.5)
|
|
entry=gtk.Entry()
|
|
if name in defaults: entry.set_text(defaults[name])
|
|
|
|
vbox.attach(label,0,1,nsettings,nsettings+1)
|
|
vbox.attach(entry,1,2,nsettings,nsettings+1)
|
|
return (name,entry)
|
|
|
|
def add_integer_setting(name,label,nsettings):
|
|
label=gtk.Label(label+' ')
|
|
label.set_alignment(1.,.5)
|
|
entry=gtk.SpinButton()
|
|
entry.set_range(5,35)
|
|
entry.set_numeric(True)
|
|
entry.set_increments(1,5)
|
|
entry.set_snap_to_ticks(True)
|
|
if name in defaults: entry.set_value(defaults[name])
|
|
|
|
vbox.attach(label,0,1,nsettings,nsettings+1)
|
|
vbox.attach(entry,1,2,nsettings,nsettings+1)
|
|
return (name,entry)
|
|
|
|
def add_bool_setting(name,label,nsettings):
|
|
entry=gtk.CheckButton(label=label,use_underline=False)
|
|
if name in defaults: entry.set_active(defaults[name])
|
|
|
|
vbox.attach(entry,0,2,nsettings,nsettings+1)
|
|
return (name,entry)
|
|
|
|
entries=[]
|
|
entries.append(add_string_setting("terminal","Terminal program",0))
|
|
entries.append(add_string_setting("navigator","Navigator program",1))
|
|
entries.append(add_integer_setting("maxpath","Number of directories",2))
|
|
entries.append(add_bool_setting("invert","List directories in reverse order",3))
|
|
entries.append(add_bool_setting("collapse","Collapse home directory to ~",4))
|
|
|
|
window.connect("response",save_settings,entries,window)
|
|
window.show_all();
|
|
|
|
def save_settings(sender,response,entries,window):
|
|
window.hide_all()
|
|
if response!=gtk.RESPONSE_OK: return
|
|
|
|
global defaults
|
|
for name,entry in entries:
|
|
try:
|
|
defaults[name]=int(entry.get_text())
|
|
except (ValueError,AttributeError):
|
|
try:
|
|
defaults[name]=entry.get_active()
|
|
except AttributeError:
|
|
defaults[name]=entry.get_text()
|
|
|
|
save_settings_file()
|
|
create_actions()
|
|
|
|
def init():
|
|
load_settings_file()
|
|
if os.path.isfile("icon.png"): icon=gtk.status_icon_new_from_file("icon.png")
|
|
elif os.path.isfile("/usr/share/autojump/icon.png"): icon=gtk.status_icon_new_from_file("/usr/share/autojump/icon.png")
|
|
else: icon=gtk.status_icon_new_from_icon_name("help")
|
|
icon.set_visible(True)
|
|
icon.connect("popup-menu",popup)
|
|
|
|
def quit(sender):
|
|
gtk.main_quit()
|
|
|
|
######################################################
|
|
#insert other actions here using the action decorator#
|
|
######################################################
|
|
def create_actions():
|
|
global actions
|
|
actions={}
|
|
|
|
@action(has_child_dir(".git",recursion=3))
|
|
def gitk(sender,path):
|
|
if not os.fork():
|
|
os.chdir(path)
|
|
subprocess.Popen(['gitk']).wait()
|
|
sys.exit()
|
|
|
|
@action(has_child_file("CMakeCache.txt"),"configure")
|
|
def cmake(sender,path):
|
|
if not os.fork():
|
|
os.chdir(path)
|
|
subprocess.Popen(['cmake-gui','.']).wait()
|
|
sys.exit()
|
|
|
|
@action(os.path.isdir)
|
|
def terminal(sender,path):
|
|
print "launch terminal '%s'" % defaults["terminal"]
|
|
if not os.fork():
|
|
try:
|
|
if defaults["terminal"]=="konsole":
|
|
subprocess.Popen([defaults["terminal"],"--workdir=%s"%path]).wait()
|
|
else:
|
|
os.chdir(path)
|
|
subprocess.Popen([defaults["terminal"]]).wait()
|
|
except OSError:
|
|
pass
|
|
sys.exit()
|
|
|
|
@action(os.path.isdir)
|
|
def navigator(sender,path):
|
|
print "launch navigator '%s'" % defaults["navigator"]
|
|
if not os.fork():
|
|
try:
|
|
subprocess.Popen([defaults["navigator"],path]).wait()
|
|
except OSError:
|
|
pass
|
|
sys.exit()
|
|
|
|
if __name__=='__main__':
|
|
init()
|
|
gtk.main()
|