From 5ba4e27e3d9879de7e28eb15b999a339fee30349 Mon Sep 17 00:00:00 2001
From: adator85 <>
Date: Mon, 19 Aug 2024 00:27:45 +0200
Subject: [PATCH] update mod_votekick.py

---
 mods/mod_votekick.py | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/mods/mod_votekick.py b/mods/mod_votekick.py
index 54933aa..254140f 100644
--- a/mods/mod_votekick.py
+++ b/mods/mod_votekick.py
@@ -45,7 +45,7 @@ class Votekick():
         # Créer les nouvelles commandes du module
         self.commands_level = {
             0: ['vote_for', 'vote_against'],
-            1: ['activate', 'deactivate', 'submit', 'vote_stat', 'vote_verdict']
+            1: ['activate', 'deactivate', 'submit', 'vote_stat', 'vote_verdict', 'vote_cancel']
         }
 
         # Init the module
@@ -227,10 +227,44 @@ class Votekick():
 
         command = str(cmd[0]).lower()
         dnickname = self.Config.SERVICE_NICKNAME
+        if len(fullcmd) >= 3:
+            fromchannel = str(fullcmd[2]).lower() if self.Base.Is_Channel(str(fullcmd[2]).lower()) else None
+        else:
+            fromchannel = None
+        
+        if len(cmd) >= 2:
+            sentchannel = str(cmd[1]).lower() if self.Base.Is_Channel(str(cmd[1]).lower()) else None
+        else:
+            sentchannel = None
+        
+        if not fromchannel is None:
+            channel = fromchannel
+        elif not sentchannel is None:
+            channel = sentchannel
+        else:
+            channel = None
+
+
         fromuser = user
 
         match command:
 
+            case 'vote_cancel':
+                try:
+                    if fromchannel is None:
+                        channel = str(cmd[1]).lower()
+                    else:
+                        channel = fromchannel
+                    
+                    for vote in self.VOTE_CHANNEL_DB:
+                        if vote.channel_name == channel:
+                            self.init_vote_system(channel)
+                            self.Irc.send2socket(f':{dnickname} PRIVMSG {channel} :Vote system re-initiated')
+
+                except IndexError as ke:
+                    self.Irc.send2socket(f':{dnickname} NOTICE {fromuser} :/msg {dnickname} vote_cancel #channel')
+                    self.Logs.error(f'Index Error: {ke}')
+
             case 'vote_for':
                 try:
                     # vote_for