From 467a615e8bccb3124d7acd9d75b489a9f3cc6ee2 Mon Sep 17 00:00:00 2001
From: bobokun <jon.cy.lee98@gmail.com>
Date: Fri, 17 Dec 2021 20:21:27 -0500
Subject: [PATCH] Updated payload for notifiarr

---
 config/config.yml.sample | 3 +++
 modules/config.py        | 3 ++-
 modules/notifiarr.py     | 3 ++-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/config/config.yml.sample b/config/config.yml.sample
index 4054b92..09ad710 100644
--- a/config/config.yml.sample
+++ b/config/config.yml.sample
@@ -130,7 +130,10 @@ orphaned:
 
 #Notifiarr integration with webhooks
 notifiarr:
+  #Mandatory to fill out API Key
   apikey: ####################################
+  #Your qBittorrent instance, can be set to any unique value
+  instance:
   test: true
   develop: true
 
diff --git a/modules/config.py b/modules/config.py
index 46439bb..6a718f7 100644
--- a/modules/config.py
+++ b/modules/config.py
@@ -83,7 +83,8 @@ class Config:
                     self.NotifiarrFactory = Notifiarr(self, {
                         "apikey": self.util.check_for_attribute(self.data, "apikey", parent="notifiarr", throw=True),
                         "develop": self.util.check_for_attribute(self.data, "develop", parent="notifiarr", var_type="bool", default=False, do_print=False, save=False),
-                        "test": self.util.check_for_attribute(self.data, "test", parent="notifiarr", var_type="bool", default=False, do_print=False, save=False)
+                        "test": self.util.check_for_attribute(self.data, "test", parent="notifiarr", var_type="bool", default=False, do_print=False, save=False),
+                        "instance": self.util.check_for_attribute(self.data, "instance", parent="notifiarr", default=False, do_print=False, save=False)
                     })
                 except Failed as e:
                     logger.error(e)
diff --git a/modules/notifiarr.py b/modules/notifiarr.py
index d67a081..a2aeed9 100644
--- a/modules/notifiarr.py
+++ b/modules/notifiarr.py
@@ -14,6 +14,7 @@ class Notifiarr:
         self.apikey = params["apikey"]
         self.develop = params["develop"]
         self.test = params["test"]
+        self.instance = params["instance"]
         url, _ = self.get_url("user/validate/")
         response = self.config.get(url)
         response_json = response.json()
@@ -28,5 +29,5 @@ class Notifiarr:
         if self.config.trace_mode:
             logger.debug(url.replace(self.apikey, "APIKEY"))
         test_payload = (f"qbitManage-{self.apikey[:5]}")
-        params = {"event": test_payload, "qbit_client":self.config.data["qbt"]["host"] if self.test else "notify"}
+        params = {"event": test_payload, "qbit_client":self.config.data["qbt"]["host"], "instance":self.instance if self.test else "notify"}
         return url, params
\ No newline at end of file