Merge pull request #17 from uzair-ashraf/YesVRC/ptb-canary

Support for PTB and Canary
shadorki · Jan 18, 2023 · 38e1305 · 38e1305
2 parents d047541 + 944ec88
commit 38e1305
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
 __pycache__
 build
-vrc-discord-osc.config.json
+vrc-discord-osc.config.json
+*.spec
diff --git a/discord_band.py b/discord_band.py
@@ -13,6 +13,7 @@ def __init__(self, port: int):
         self.disable_timer = 0
         self.thread = Thread(target=self.run)
         self.thread.start()
+        self.versions = ["Discord", "Discord Canary", "Discord PTB"]
 
     def is_call_notification(self, notification: UserNotification) -> bool:
         try:
@@ -36,7 +37,8 @@ def is_discord_notification(self, notification: UserNotification) -> bool:
         try:
             if hasattr(notification, "app_info"):
                 app_name = notification.app_info.display_info.display_name
-                if app_name == "Discord":
+                if app_name in self.versions:
+                    print(f'Notification from {app_name}')
                     return True
             return False
         except AttributeError: