key_logger.py

import time
import log
from collections import namedtuple
from entities import Transition


class KeyLogger:
    def __init__(self):
        self.first_key = True
        self.keys = list()

    def extract_to(self, key):
        return key.special if key.special != "" else key.char

    def time_ms(self):
        return time.time_ns() / 1000000

    def log_key(self, key, state):
        if self.first_key:
            self.keys.append(Transition("START", self.extract_to(key), state, 0))
            self.first_key = False
        else:
            self.keys.append(
                Transition(
                    self.keys[-1].end,
                    self.extract_to(key),
                    state,
                    self.time_ms() - self.last_time,
                )
            )
        self.last_time = self.time_ms()

    def transitions(self):
        return self.keys

    def summary(self):
        return "\n".join(["{}".format(k) for k in self.keys])