main.py

from pynput import keyboard
from datetime import datetime
from time import time

from log_analyzer import analyze

LOG_FILE = "logs/keystrokes.log"


def start_session():
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")

    with open(LOG_FILE, "a", encoding="utf-8") as f:
        f.write(f"\n--- Session started {timestamp} ---\n")

def format_key(key):
    try:
        return key.char
    except AttributeError:
        return f"[ {key.name} ]"

def log(key, log_file):
    k_text = format_key(key)
    timestamp = time()
    log_file.write(f"{timestamp:.3f} {k_text}"+"\n")


def finish(key):
    if key==keyboard.Key.esc:
        print('\n')
        return False

def main():
    start_session()
    with open(LOG_FILE, "a", encoding="utf-8") as log_file:
        def keyPressed(key):
            log(key, log_file)

        with keyboard.Listener(on_press=keyPressed, on_release=finish) as listener:
            listener.join()

    analyze()

if __name__ == '__main__':
    main()