translate.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import subprocess
import sys
import re
import argparse

parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('--keymap-source', dest='source', choices=["linux/input.h","xmodmap"],
                   default="xmodmap",help='specify which keymap to use')
parser.add_argument('--dumpfile', type=str, dest="dumpfile",
                   help='the path to the dumpfile, if not specified stdin will be used')
args = parser.parse_args()


if args.dumpfile:
    try:
        dump = open(args.dumpfile)
    except:
        print("%s is not readable or does not exist."%args.dumpfile)
        sys.exit(1)
else:
    dump = sys.stdin

positions_to_translate = [0,1,2]
offset = -8

def load_mapping_from_xmodmap():
    keymap = subprocess.check_output(["xmodmap","-pke"]).decode("utf-8")
    mapping = {}

    for line in keymap.split("\n"):
        parts = line.split()
        if len(parts) > 4:
            mapping[int(parts[1])+offset] = parts[3]
    return mapping
def load_mapping_from_linux_input_h():
    mapping = {}
    with open("/usr/include/linux/input.h") as headerfile:
        regex = "^#define KEY_([^\s]*)\s*(\d*)"
        finds = re.findall(regex, headerfile.read(), flags=re.MULTILINE)
        for keyname, scancode in finds:
            if keyname and scancode:
                mapping[int(scancode)] = keyname
    return mapping
def cmd_exists(cmd): # from http://stackoverflow.com/a/11069822/3890934
    return subprocess.call("type " + cmd, shell=True,
        stdout=subprocess.PIPE, stderr=subprocess.PIPE) == 0

if args.source == "linux/input.h":
    mapping = load_mapping_from_linux_input_h()
elif args.source == "xmodmap":
    if cmd_exists("xmodmap"):
        try:
            mapping = load_mapping_from_xmodmap()
        except:
            print("Could not load mapping from xmodmap.")
            sys.exit(1)

table = []

for line in dump:
    parts = line.split()
    if int(parts[3]) > 0:
        for position in positions_to_translate:
            parts[position] = mapping[int(parts[position])]
        table.append(parts)

table.sort(key=lambda x: int(x[3]), reverse=True)

for x in table:
    x[3] = x[3].lstrip("0");

for row in table:
    try:
        print(" ".join(row))
    except Exception:
        break