Improved `ipcounting`

Signed-off-by: Georg <georg@lysergic.dev>

1 files changed, 5 insertions, 3 deletions

diff --git a/plugin.py b/plugin.py
index 4d9a797..4b16961 100644
--- a/plugin.py
+++ b/plugin.py
@@ -141,7 +141,8 @@ class SnoParser(callbacks.Plugin):
     def whois_set_cache(self, key: str, value: str) -> bool:
         """Data to Redis."""
 
-        state = self.redis_client_whois.setex(key, timedelta(seconds=3600), value=value,)
+        duration = self.registryValue('whois.ttl')
+        state = self.redis_client_whois.setex(key, timedelta(seconds=duration), value=value,)
         return state
 
     def whois_run(self, sourceip: str) -> dict:
@@ -238,12 +239,13 @@ class SnoParser(callbacks.Plugin):
         """
 
         data = self.whois_get_cache(key=ipaddress)
-        decoded = data.decode('utf-8')
+        decoded_data = data.decode('utf-8')
         ttl = self.redis_client_whois.ttl(ipaddress)
         count = self.redis_client_ips.get(ipaddress)
+        decoded_count = count.decode('utf-8')
 
         print('SnoParser manual query: ', data, ' ', ttl)
-        irc.reply(f'{decoded} - Count: {count} - Remaining: {ttl}s')
+        irc.reply(f'{decoded_data} - Count: {decoded_count} - Remaining: {ttl}s')
         
     ipquery = wrap(ipquery, ['anything'])