Added hotkeys, work setup

Author: mrdon

.gitignore

@@ -0,0 +1,2 @@
+.secrets
+.env

Makefile

@@ -1,5 +1,7 @@
 .PHONY: run tunnel client help venv db db-migrate db-makemigrations format
 
+SHELL := /bin/bash
+
 
 # Help system from https://marmelab.com/blog/2016/02/29/auto-documented-makefile.html
 .DEFAULT_GOAL := help
@@ -18,4 +20,5 @@ format: ## Format the code
 	venv/bin/reorder-python-imports --py38-plus `find src -name "*.py"` || venv/bin/black src --target-version py38
 
 run: ## Run the app
-	cd src && ../venv/bin/python -m main
+	source .secrets
+	cd src && ../venv/bin/python -m sleuthdeck.cli work.py

requirements.in

@@ -6,6 +6,7 @@ pydub
 selenium
 obs-websocket-py
 watchdog
+pyautogui
 
 pytest
 black

requirements.txt

@@ -37,6 +37,8 @@ idna==3.3
     #   urllib3
 iniconfig==1.1.1
     # via pytest
+mouseinfo==0.1.3
+    # via pyautogui
 mypy-extensions==0.4.3
     # via black
 numpy==1.22.1
@@ -61,16 +63,34 @@ psutil==5.9.0
     # via -r requirements.in
 py==1.11.0
     # via pytest
+pyautogui==0.9.53
+    # via -r requirements.in
 pycparser==2.21
     # via cffi
 pydub==0.25.1
     # via -r requirements.in
+pygetwindow==0.0.9
+    # via pyautogui
+pymsgbox==1.0.9
+    # via pyautogui
 pyopenssl==21.0.0
     # via urllib3
 pyparsing==3.0.6
     # via packaging
+pyperclip==1.8.2
+    # via mouseinfo
+pyrect==0.1.4
+    # via pygetwindow
+pyscreeze==0.1.28
+    # via pyautogui
 pytest==6.2.5
     # via -r requirements.in
+python3-xlib==0.15
+    # via
+    #   mouseinfo
+    #   pyautogui
+pytweening==1.0.4
+    # via pyautogui
 reorder-python-imports==2.6.0
     # via -r requirements.in
 selenium==4.1.0

src/main.py

@@ -1,12 +1,16 @@
 import signal
 import subprocess
+from time import sleep
+from typing import Optional, Union
+
+from pyautogui import hotkey
 
 from sleuthdeck.deck import Action
 from sleuthdeck.deck import ClickType
 from sleuthdeck.deck import Key
 from sleuthdeck.deck import KeyScene
 from sleuthdeck.deck import Scene
-from sleuthdeck.windows import get_window
+from sleuthdeck.windows import get_window, By
 
 
 class Command(Action):
@@ -39,7 +43,7 @@ def execute(self, scene: KeyScene, key: Key, click: ClickType):
 
 
 class MaximizeWindow(Action):
-    def __init__(self, title: str):
+    def __init__(self, title: Union[str, By]):
         self.title = title
 
     def execute(self, scene: KeyScene, key: Key, click: ClickType):
@@ -50,20 +54,28 @@ def execute(self, scene: KeyScene, key: Key, click: ClickType):
             print("No window found")
 
 
+class Pause(Action):
+    def __init__(self, seconds: Union[float, int]):
+        self.seconds = seconds
+
+    def execute(self, scene: KeyScene, key: Key, click: ClickType):
+        sleep(self.seconds)
+
+
 class CloseWindow(Action):
-    def __init__(self, title: str):
+    def __init__(self, title: Union[str, By]):
         self.title = title
 
     def execute(self, scene: KeyScene, key: Key, click: ClickType):
         window = get_window(self.title, attempts=5 * 10)
         if window:
             window.close()
         else:
-            print("No window found")
+            print(f"No window found with {self.title}")
 
 
 class MoveWindow(Action):
-    def __init__(self, title: str, x: int, y: int, width: int, height: int):
+    def __init__(self, title: Union[str, By], x: int, y: int, width: int, height: int):
         self.x = x
         self.y = y
         self.width = width
@@ -76,3 +88,17 @@ def execute(self, scene: KeyScene, key: Key, click: ClickType):
             window.move(self.x, self.y, self.width, self.height)
         else:
             print("No window found")
+
+
+class SendHotkey(Action):
+    def __init__(self, title: Union[str, By], *hotkey: str):
+        self.title = title
+        self.hotkey = hotkey
+
+    def execute(self, scene: KeyScene, key: Key, click: ClickType):
+        print("sending key")
+        window = get_window(self.title, attempts=5 * 10)
+        print("got window")
+        window.focus()
+        hotkey(*self.hotkey)
+        print("sent")

src/sleuthdeck/actions.py

@@ -4,6 +4,7 @@
 import sys
 import threading
 import time
+import traceback
 from datetime import datetime
 from datetime import timedelta
 from multiprocessing import Queue
@@ -39,7 +40,13 @@ def run(self):
 
         try:
             while self._reloading_queue.get():
-                mod = runpy.run_module(mod_name)
+                try:
+                    mod = runpy.run_module(mod_name)
+                except:
+                    print(f"Error loading script")
+                    traceback.print_exc()
+                    continue
+
                 if "run" not in mod:
                     raise ValueError(f"Script {self.script} missing 'run' function")
 
@@ -50,7 +57,12 @@ def run(self):
                 print(f"Loaded new deck from {self.script}")
                 self.deck = deck
                 with deck:
-                    mod["run"](deck)
+                    try:
+                        mod["run"](deck)
+                    except Exception as e:
+                        print(f"Error: {e}")
+                        traceback.print_exc()
+                        continue
         except EOFError:
             if self.deck and self.deck.stream_deck.is_open():
                 self.deck.close()

src/assets/Roboto-Regular.ttf src/sleuthdeck/assets/Roboto-Regular.ttf

@@ -3,6 +3,7 @@
 import asyncio
 import threading
 import time
+import traceback
 from asyncio import Future
 from asyncio import Task
 from dataclasses import dataclass
@@ -53,8 +54,11 @@ def connect(self, scene: KeyScene):
 
 class Deck:
     def __init__(self):
+        print("Scanning for stream decks")
         streamdecks = DeviceManager().enumerate()
         self.stream_deck: StreamDeck = streamdecks[0]
+        print(f"Found stream deck: {self.stream_deck.id()}")
+
         self._animation = Animations(self.stream_deck)
         self._scene = Scene()
         self._last_scene: Scene = self._scene
@@ -191,12 +195,14 @@ def _check_long_click(self):
 
     def _run_actions(self, click: ClickType, key: Key):
         key.clicked_on = None
-        actions = self._actions.get(key, key.actions)
+        actions = list(self._actions.get(key, key.actions))
         for action in actions:
             try:
+                print(f"Running {action.__class__.__name__}")
                 action.execute(self, key, click)
             except Exception as e:
                 print(f"Error running action {action}: {e}")
+                traceback.print_exc()
 
     def add(
         self,

src/sleuthdeck/cli.py

@@ -2,6 +2,7 @@
 
 import asyncio
 from os import path
+from os.path import dirname
 from typing import Callable
 from typing import List
 from typing import Optional
@@ -56,7 +57,7 @@ def load_image(
             # Load a custom TrueType font and use it to overlay the key index, draw key
             # label onto the image a few pixels from the bottom of the key.
             draw = ImageDraw.Draw(image)
-            font = ImageFont.truetype("Roboto-Regular.ttf", 14)
+            font = ImageFont.truetype(path.join(dirname(__file__), "assets", "Roboto-Regular.ttf"), 14)
             draw.text(
                 (image.width / 2, image.height - 5),
                 text=text,