script for window group cycling on elementary OS

windows.py

from subprocess import Popen, PIPE, call
import logging
import logging.handlers
import sys
import os

# NOTE: this script assumes a debian system and requires the wmctrl and xdotool packages
#   sudo apt-get install wmctrl xdotool

# NOTE: To get [Alt + ` ]to register on Elementary OS requires removing the keybinding via dconf editor for switch-group/switch-group-backward
#   org.gnome.desktop.wm.keybindings.switch-group: []
#   org.gnome.desktop.wm.keybindings.switch-group-backward: []
# set custom hotkey for forward/backward window group scrolling (switch-group)
#   python /path/to/windows.py group -f
#   python /path/to/windows.py group -b
# set custom hotkey for forward/backward window scrolling (switch-windows)
#   python /path/to/windows.py window -f
#   python /path/to/windows.py window -b

# logging will be placed into
#   /path/to/windows.py.log

LOG_FORMAT = '%(asctime)s %(levelname)s (%(pathname)s:%(lineno)d): %(message)s'
fh = logging.handlers.RotatingFileHandler(
        os.path.join(os.path.dirname(__file__), 'windows.py.log'),
        maxBytes= 5 * 1024 * 1024, # 5 MB
        backupCount=1)
fh.setFormatter(logging.Formatter(LOG_FORMAT))

_logger = logging.getLogger('windows.py')
_logger.setLevel(logging.DEBUG)
_logger.addHandler(fh)

_logger.info('Args: {0}'.format(sys.argv))
_logger.info('looking for windows')

if 'group' in sys.argv:
    method = 'group'
else:
    method = 'window'
    
if '-b' in sys.argv:
    direction = '-b'
else:
    direction = '-f'

IGNORE_CLASS = [
    'plank.Plank',
    'wingpanel.Wingpanel'
]

try:
    windows = []
    for line in Popen("wmctrl -lx", shell=True, stdout=PIPE).stdout.read().split(b'\n'):
        win = line.split()
        if win and int(win[1]) > -1 and win[2] not in IGNORE_CLASS:
            # wine apps might have an unusual class like "notepad++.exe.notepad++.exe"
            # ultimately we would want this _class to resolve to "notepad++.exe"
            _class = str(win[2]).split('.')
            windows.append(dict(
                id=int(win[0], 16),
                desktop=int(win[1]),
                _class='.'.join(_class[int(len(_class) / 2):])
            ))
    _logger.info('windows found: {0}'.format(windows))

    current_window_id = int(Popen("xdotool getactivewindow", shell=True, stdout=PIPE).stdout.read().strip())
    _logger.info('current_window_id: {0}'.format(current_window_id))
    current_window_class = next((x['_class'] for x in windows if x['id'] == current_window_id), None)
    current_window_desktop = next((x['desktop'] for x in windows if x['id'] == current_window_id), None)

    if current_window_class and method == 'group':
        class_windows = [x for x in windows if x['_class'] == current_window_class and x['desktop'] == current_window_desktop]
        _logger.info('Class windows: {0}'.format(class_windows))
        if class_windows:
            if direction == '-f':
                index = 0
                for x in range(len(class_windows)):
                    if class_windows[x]['id'] == current_window_id:
                        index = x
                        _logger.info('Found window index: {0}'.format(index))
                        break
                if index + 1 >= len(class_windows):
                    index = 0
                else:
                    index  = index + 1
            elif direction == '-b':
                index = len(class_windows) - 1
                for x in range(len(class_windows) -1, -1, -1):
                    if class_windows[x]['id'] == current_window_id:
                        index = x
                        _logger.info('Found window index: {0}'.format(index))
                        break
                if index <= 0:
                    index = len(class_windows) - 1
                else:
                    index = index - 1
            _logger.info('Calling class window index {0}'.format(index))
            call(['wmctrl', '-i', '-a', str(class_windows[index]['id'])])
        else:
            _logger.info('class_windows not found')
    elif method == 'window':
        desktop_windows = [x for x in windows if x['desktop'] == current_window_desktop]
        if desktop_windows:            
            if direction == '-f':
                index = 0
                for x in range(len(desktop_windows)):
                    if desktop_windows[x]['id'] == current_window_id:
                        index = x
                        _logger.info('Found window index: {0}'.format(index))
                        break
                if index + 1 >= len(desktop_windows):
                    index = 0
                else:
                    index  = index + 1
            elif direction == '-b':
                index = len(desktop_windows) - 1
                for x in range(len(desktop_windows) -1, -1, -1):
                    if desktop_windows[x]['id'] == current_window_id:
                        index = x
                        _logger.info('Found window index: {0}'.format(index))
                        break
                if index <= 0:
                    index = len(desktop_windows) - 1
                else:
                    index = index - 1
            _logger.info('Calling window {0}'.format(desktop_windows[index]))
            call(['wmctrl', '-i', '-a', str(desktop_windows[index]['id'])])
        else:
            _logger.info('windows not found')
    else:
        _logger.info('current_window_class not found')
            
except Exception as e:
    _logger.exception("windows.py blew up")