code-style issues

1 files changed, 53 insertions, 33 deletions

diff --git a/cogs/vc_logs.py b/cogs/vclogs.py
index e4a8454..4bc7104 100644
--- a/cogs/vc_logs.py
+++ b/cogs/vclogs.py
@@ -1,27 +1,59 @@
+import datetime
+import json
+import os
 
 import discord
 from discord.ext import commands
-import datetime
 from gtts import gTTS
-import os
-import json
 
-class vc_logs(commands.Cog):
+
+class VcLogs(commands.Cog):
     def __init__(self, bot):
         self.bot = bot
-    
+
+    @staticmethod
+    async def handle_state_change(member, channel, state_type, is_enabled, current_time, log_channel):
+        """Helper method to handle voice state changes"""
+        state_messages = {
+            'deaf': {
+                True: (
+                    f"[{current_time}] {member.name} ogłuchnął się na {channel.name}",
+                    f"{member.display_name} jest bity przez ojca"
+                ),
+                False: (
+                    f"[{current_time}] {member.name} odgłuchał się na {channel.name}",
+                    f"{member.display_name} uciek od ojca"
+                )
+            },
+            'mute': {
+                True: (
+                    f"[{current_time}] {member.name} wyciszył się na {channel.name}",
+                    f"{member.display_name} idzie robić jajecznice"
+                ),
+                False: (
+                    f"[{current_time}] {member.name} odciszył się na {channel.name}",
+                    f"{member.display_name} zrobił jajecznice"
+                )
+            }
+        }
+
+        message, tts_text = state_messages[state_type][is_enabled]
+        await log_channel.send(message)
+        print(message)
+        return tts_text, channel
+
     async def play_tts(self, channel, text):
         # Generate TTS audio
         tts = gTTS(text=text, lang='pl')
         tts.save('temp_tts.mp3')
-        
-        # Connect to voice channel if not already connected
+
+        # Connect to the voice channel if not already connected
         voice_client = discord.utils.get(self.bot.voice_clients, guild=channel.guild)
         if not voice_client or not voice_client.is_connected():
             voice_client = await channel.connect()
         elif voice_client.channel != channel:
             await voice_client.move_to(channel)
-        
+
         # Play the TTS audio
         if not voice_client.is_playing():
             if os.path.exists('settings.json'):
@@ -31,7 +63,7 @@ class vc_logs(commands.Cog):
             else:
                 with open('settings.json', 'w') as f:
                     json.dump({'volume': '5'}, f, indent=4)
-                           
+
             audio_source = discord.FFmpegPCMAudio('temp_tts.mp3', options=f'-filter:a "volume={volume}"')
             voice_client.play(audio_source)
             while voice_client.is_playing():
@@ -52,8 +84,6 @@ class vc_logs(commands.Cog):
         except Exception as e:
             await ctx.respond(e, ephemeral=True)
             print(e)
-            
-            
 
     @commands.Cog.listener()
     async def on_voice_state_update(self, member, before, after):
@@ -61,9 +91,11 @@ class vc_logs(commands.Cog):
         current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         tts_text = None
         tts_channel = None
-        # Ignore if user is a bot
+
+        # Ignore if the user is a bot
         if member.bot:
             return
+
         if before.channel is None and after.channel is not None:
             message = f"[{current_time}] {member.name} dołączył na {after.channel.name}"
             tts_text = f"{member.display_name} sie zespawnił"
@@ -76,35 +108,23 @@ class vc_logs(commands.Cog):
             tts_channel = before.channel
             await log_channel.send(message)
             print(message)
-        
+
         if before.self_deaf != after.self_deaf:
-            if after.self_deaf:
-                message = f"[{current_time}] {member.name} ogłuchnął się na {after.channel.name}"
-                tts_text = f"{member.display_name} jest bity przez ojca"
-            else:
-                message = f"[{current_time}] {member.name} odgłuchął się na {after.channel.name}"
-                tts_text = f"{member.display_name} uciek od ojca"
-            tts_channel = after.channel
-            await log_channel.send(message)
-            print(message)
+            tts_text, tts_channel = await self.handle_state_change(
+                member, after.channel, 'deaf', after.self_deaf, current_time, log_channel
+            )
             if tts_text and tts_channel:
                 await self.play_tts(tts_channel, tts_text)
             return
 
         if before.self_mute != after.self_mute:
-            if after.self_mute:
-                message = f"[{current_time}] {member.name} wyciszył się na {after.channel.name}"
-                tts_text = f"{member.display_name} idzie robić jajecznice"
-            else:
-                message = f"[{current_time}] {member.name} odciszył się na {after.channel.name}"
-                tts_text = f"{member.display_name} zrobił jajecznice"
-            tts_channel = after.channel
-            await log_channel.send(message)
-            print(message)
-        
+            tts_text, tts_channel = await self.handle_state_change(
+                member, after.channel, 'mute', after.self_mute, current_time, log_channel
+            )
+
         if tts_text and tts_channel:
             await self.play_tts(tts_channel, tts_text)
 
 
 def setup(bot):
-    bot.add_cog(vc_logs(bot))
\ No newline at end of file
+    bot.add_cog(VcLogs(bot))
\ No newline at end of file