312 lines
10 KiB
Python
Executable File
312 lines
10 KiB
Python
Executable File
#!/usr/bin/python
|
|
# Terminator.util - misc utility functions
|
|
# Copyright (C) 2006-2010 cmsj@tenshu.net
|
|
#
|
|
# This program 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, version 2 only.
|
|
#
|
|
# This program 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 this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
"""Terminator.util - misc utility functions
|
|
|
|
>>> a = {'foo': 'bar', 'baz': 'bjonk'}
|
|
>>> b = {'foo': 'far', 'baz': 'bjonk'}
|
|
>>> dict_diff(a, b)
|
|
{'foo': 'far'}
|
|
|
|
"""
|
|
|
|
import sys
|
|
from gi.repository import Gtk
|
|
import os
|
|
import pwd
|
|
import inspect
|
|
import uuid
|
|
import subprocess
|
|
|
|
# set this to true to enable debugging output
|
|
DEBUG = False
|
|
# set this to true to additionally list filenames in debugging
|
|
DEBUGFILES = False
|
|
# list of classes to show debugging for. empty list means show all classes
|
|
DEBUGCLASSES = []
|
|
# list of methods to show debugging for. empty list means show all methods
|
|
DEBUGMETHODS = []
|
|
|
|
def dbg(log = ""):
|
|
"""Print a message if debugging is enabled"""
|
|
if DEBUG:
|
|
stackitem = inspect.stack()[1]
|
|
parent_frame = stackitem[0]
|
|
method = parent_frame.f_code.co_name
|
|
names, varargs, keywords, local_vars = inspect.getargvalues(parent_frame)
|
|
try:
|
|
self_name = names[0]
|
|
classname = local_vars[self_name].__class__.__name__
|
|
except IndexError:
|
|
classname = "noclass"
|
|
if DEBUGFILES:
|
|
line = stackitem[2]
|
|
filename = parent_frame.f_code.co_filename
|
|
extra = " (%s:%s)" % (filename, line)
|
|
else:
|
|
extra = ""
|
|
if DEBUGCLASSES != [] and classname not in DEBUGCLASSES:
|
|
return
|
|
if DEBUGMETHODS != [] and method not in DEBUGMETHODS:
|
|
return
|
|
try:
|
|
print >> sys.stderr, "%s::%s: %s%s" % (classname, method, log, extra)
|
|
except IOError:
|
|
pass
|
|
|
|
def err(log = ""):
|
|
"""Print an error message"""
|
|
try:
|
|
print >> sys.stderr, log
|
|
except IOError:
|
|
pass
|
|
|
|
def gerr(message = None):
|
|
"""Display a graphical error. This should only be used for serious
|
|
errors as it will halt execution"""
|
|
|
|
dialog = Gtk.MessageDialog(None, Gtk.DialogFlags.MODAL,
|
|
Gtk.MessageType.ERROR, Gtk.ButtonsType.OK, message)
|
|
dialog.run()
|
|
dialog.destroy()
|
|
|
|
def has_ancestor(widget, wtype):
|
|
"""Walk up the family tree of widget to see if any ancestors are of type"""
|
|
while widget:
|
|
widget = widget.get_parent()
|
|
if isinstance(widget, wtype):
|
|
return(True)
|
|
return(False)
|
|
|
|
def path_lookup(command):
|
|
'''Find a command in our path'''
|
|
if os.path.isabs(command):
|
|
if os.path.isfile(command):
|
|
return(command)
|
|
else:
|
|
return(None)
|
|
elif command[:2] == './' and os.path.isfile(command):
|
|
dbg('path_lookup: Relative filename %s found in cwd' % command)
|
|
return(command)
|
|
|
|
try:
|
|
paths = os.environ['PATH'].split(':')
|
|
if len(paths[0]) == 0:
|
|
raise(ValueError)
|
|
except (ValueError, NameError):
|
|
dbg('path_lookup: PATH not set in environment, using fallbacks')
|
|
paths = ['/usr/local/bin', '/usr/bin', '/bin']
|
|
|
|
dbg('path_lookup: Using %d paths: %s' % (len(paths), paths))
|
|
|
|
for path in paths:
|
|
target = os.path.join(path, command)
|
|
if os.path.isfile(target):
|
|
dbg('path_lookup: found %s' % target)
|
|
return(target)
|
|
|
|
dbg('path_lookup: Unable to locate %s' % command)
|
|
|
|
def shell_lookup():
|
|
"""Find an appropriate shell for the user"""
|
|
try:
|
|
usershell = pwd.getpwuid(os.getuid())[6]
|
|
except KeyError:
|
|
usershell = None
|
|
shells = [usershell, 'bash', 'zsh', 'tcsh', 'ksh', 'csh', 'sh']
|
|
|
|
for shell in shells:
|
|
if shell is None:
|
|
continue
|
|
elif os.path.isfile(shell):
|
|
return(shell)
|
|
else:
|
|
rshell = path_lookup(shell)
|
|
if rshell is not None:
|
|
dbg('shell_lookup: Found %s at %s' % (shell, rshell))
|
|
return(rshell)
|
|
dbg('shell_lookup: Unable to locate a shell')
|
|
|
|
def widget_pixbuf(widget, maxsize=None):
|
|
"""Generate a pixbuf of a widget"""
|
|
pixmap = widget.get_snapshot()
|
|
(width, height) = pixmap.get_size()
|
|
pixbuf = GdkPixbuf.Pixbuf(GdkPixbuf.Colorspace.RGB, False, 8, width, height)
|
|
pixbuf.get_from_drawable(pixmap, pixmap.get_colormap(), 0, 0, 0, 0, width,
|
|
height)
|
|
|
|
longest = max(width, height)
|
|
|
|
if maxsize is not None:
|
|
factor = float(maxsize) / float(longest)
|
|
|
|
if not maxsize or (width * factor) > width or (height * factor) > height:
|
|
factor = 1
|
|
|
|
scaledpixbuf = pixbuf.scale_simple(int(width * factor), int(height * factor), GdkPixbuf.InterpType.BILINEAR)
|
|
|
|
return(scaledpixbuf)
|
|
|
|
def get_config_dir():
|
|
"""Expand all the messy nonsense for finding where ~/.config/terminator
|
|
really is"""
|
|
try:
|
|
configdir = os.environ['XDG_CONFIG_HOME']
|
|
except KeyError:
|
|
configdir = os.path.join(os.path.expanduser('~'), '.config')
|
|
|
|
dbg('Found config dir: %s' % configdir)
|
|
return(os.path.join(configdir, 'terminator'))
|
|
|
|
def dict_diff(reference, working):
|
|
"""Examine the values in the supplied working set and return a new dict
|
|
that only contains those values which are different from those in the
|
|
reference dictionary"""
|
|
|
|
result = {}
|
|
|
|
for key in reference:
|
|
if reference[key] != working[key]:
|
|
result[key] = working[key]
|
|
|
|
return(result)
|
|
|
|
# Helper functions for directional navigation
|
|
def get_edge(allocation, direction):
|
|
"""Return the edge of the supplied allocation that we will care about for
|
|
directional navigation"""
|
|
if direction == 'left':
|
|
edge = allocation.x
|
|
p1, p2 = allocation.y, allocation.y + allocation.height
|
|
elif direction == 'up':
|
|
edge = allocation.y
|
|
p1, p2 = allocation.x, allocation.x + allocation.width
|
|
elif direction == 'right':
|
|
edge = allocation.x + allocation.width
|
|
p1, p2 = allocation.y, allocation.y + allocation.height
|
|
elif direction == 'down':
|
|
edge = allocation.y + allocation.height
|
|
p1, p2 = allocation.x, allocation.x + allocation.width
|
|
else:
|
|
raise ValueError('unknown direction %s' % direction)
|
|
|
|
return(edge, p1, p2)
|
|
|
|
def get_nav_possible(edge, allocation, direction, p1, p2):
|
|
"""Check if the supplied allocation is in the right direction of the
|
|
supplied edge"""
|
|
x1, x2 = allocation.x, allocation.x + allocation.width
|
|
y1, y2 = allocation.y, allocation.y + allocation.height
|
|
if direction == 'left':
|
|
return(x2 <= edge and y1 <= p2 and y2 >= p1)
|
|
elif direction == 'right':
|
|
return(x1 >= edge and y1 <= p2 and y2 >= p1)
|
|
elif direction == 'up':
|
|
return(y2 <= edge and x1 <= p2 and x2 >= p1)
|
|
elif direction == 'down':
|
|
return(y1 >= edge and x1 <= p2 and x2 >= p1)
|
|
else:
|
|
raise ValueError('Unknown direction: %s' % direction)
|
|
|
|
def get_nav_offset(edge, allocation, direction):
|
|
"""Work out how far edge is from a particular point on the allocation
|
|
rectangle, in the given direction"""
|
|
if direction == 'left':
|
|
return(edge - (allocation.x + allocation.width))
|
|
elif direction == 'right':
|
|
return(allocation.x - edge)
|
|
elif direction == 'up':
|
|
return(edge - (allocation.y + allocation.height))
|
|
elif direction == 'down':
|
|
return(allocation.y - edge)
|
|
else:
|
|
raise ValueError('Unknown direction: %s' % direction)
|
|
|
|
def get_nav_tiebreak(direction, cursor_x, cursor_y, rect):
|
|
"""We have multiple candidate terminals. Pick the closest by cursor
|
|
position"""
|
|
if direction in ['left', 'right']:
|
|
return(cursor_y >= rect.y and cursor_y <= (rect.y + rect.height))
|
|
elif direction in ['up', 'down']:
|
|
return(cursor_x >= rect.x and cursor_x <= (rect.x + rect.width))
|
|
else:
|
|
raise ValueError('Unknown direction: %s' % direction)
|
|
|
|
def enumerate_descendants(parent):
|
|
"""Walk all our children and build up a list of containers and
|
|
terminals"""
|
|
# FIXME: Does having to import this here mean we should move this function
|
|
# back to Container?
|
|
from factory import Factory
|
|
|
|
containerstmp = []
|
|
containers = []
|
|
terminals = []
|
|
maker = Factory()
|
|
|
|
if parent is None:
|
|
err('no parent widget specified')
|
|
return
|
|
|
|
for descendant in parent.get_children():
|
|
if maker.isinstance(descendant, 'Container'):
|
|
containerstmp.append(descendant)
|
|
elif maker.isinstance(descendant, 'Terminal'):
|
|
terminals.append(descendant)
|
|
|
|
while len(containerstmp) > 0:
|
|
child = containerstmp.pop(0)
|
|
for descendant in child.get_children():
|
|
if maker.isinstance(descendant, 'Container'):
|
|
containerstmp.append(descendant)
|
|
elif maker.isinstance(descendant, 'Terminal'):
|
|
terminals.append(descendant)
|
|
containers.append(child)
|
|
|
|
dbg('%d containers and %d terminals fall beneath %s' % (len(containers),
|
|
len(terminals), parent))
|
|
return(containers, terminals)
|
|
|
|
def make_uuid(str_uuid=None):
|
|
"""Generate a UUID for an object"""
|
|
if str_uuid:
|
|
return uuid.UUID(str_uuid)
|
|
return uuid.uuid4()
|
|
|
|
def inject_uuid(target):
|
|
"""Inject a UUID into an existing object"""
|
|
uuid = make_uuid()
|
|
if not hasattr(target, "uuid") or target.uuid == None:
|
|
dbg("Injecting UUID %s into: %s" % (uuid, target))
|
|
target.uuid = uuid
|
|
else:
|
|
dbg("Object already has a UUID: %s" % target)
|
|
|
|
def spawn_new_terminator(cwd, args):
|
|
"""Start a new terminator instance with the given arguments"""
|
|
cmd = sys.argv[0]
|
|
|
|
if not os.path.isabs(cmd):
|
|
# Command is not an absolute path. Figure out where we are
|
|
cmd = os.path.join (cwd, sys.argv[0])
|
|
if not os.path.isfile(cmd):
|
|
# we weren't started as ./terminator in a path. Give up
|
|
err('Unable to locate Terminator')
|
|
return False
|
|
|
|
dbg("Spawning: %s" % cmd)
|
|
subprocess.Popen([cmd]+args)
|