Buttons

cogs/error.py

@@ -1,25 +1,25 @@
-# import logging
+import logging
 
-from discord import Message
 from discord.ext import commands
 
 
+# for finding errors with the code.
+# TODO: SPRAV TO UŽ!!!!!
+
 class Error(commands.Cog):
     """Basic class for catching errors and sending a message"""
 
     def __init__(self, bot):
         self.bot = bot
 
-        # TODO: spravit logger
-
-        # self.logger = logging.getLogger('discord')
-        # self.logger.setLevel(logging.WARN)
-        # handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
-        # handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s'))
-        # self.logger.addHandler(handler)
+        self.logger = logging.getLogger('discord')
+        self.logger.setLevel(logging.WARN)
+        handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
+        handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s'))
+        self.logger.addHandler(handler)
 
     @commands.Cog.listener()
-    async def on_command_error(self, ctx: commands.Context, error: commands.CommandError) -> Message | None:
+    async def on_command_error(self, ctx: commands.Context, error: commands.CommandError):
         if isinstance(error, commands.ExpectedClosingQuoteError):
             return await ctx.send(f"Pozor! Chybí tady: {error.close_quote} uvozovka!")
 
@@ -35,15 +35,11 @@ async def on_command_error(self, ctx: commands.Context, error: commands.CommandE
         else:
             # self.logger.critical(f"{ctx.message.id}, {ctx.message.content} | {error}")
 
-            return await ctx.send(
-                f"O této chybě ještě nevím a nebyla zaznamenána. Napiš The Xero#1273 o této chybě.\n"
-                f"Text chyby: `{error}`\n"
-                f"Číslo chyby: `{ctx.message.id}`"
-            )
+            print(error)
 
-    # @commands.Cog.listener()
-    # async def on_command(self, ctx: commands.Context):
-    #     self.logger.info(f"{ctx.message.id} {ctx.message.content}")
+    @commands.Cog.listener()
+    async def on_command(self, ctx: commands.Context):
+        self.logger.info(f"{ctx.message.id} {ctx.message.content}")
 
 
 async def setup(bot):

cogs/newpollstyle.py

@@ -0,0 +1,43 @@
+from discord.ext import commands
+
+from db_folder.sqldatabase import PollDatabase, VoteButtonDatabase
+from poll_design.poll import Poll
+from poll_design.poll_view import PollView
+from ui.poll_embed import PollEmbed, PollEmbedBase
+
+
+def error_handling(answer: tuple[str]) -> str:
+    if len(answer) > Poll.MAX_OPTIONS:
+        return f"Zadal jsi příliš mnoho odpovědí, můžeš maximálně {Poll.MAX_OPTIONS}!"
+    elif len(answer) < Poll.MIN_OPTIONS:
+        return f"Zadal jsi příliš málo odpovědí, můžeš alespoň {Poll.MIN_OPTIONS}!"
+
+
+class PollCreate(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.command()
+    async def poll(self, ctx: commands.Context, question: str, *answer: str):
+        message = await ctx.send(embed=PollEmbedBase("Dělám na tom, vydrž!"))
+        if error_handling(answer):
+            return await message.edit(embed=PollEmbedBase(error_handling(answer)))
+
+        poll = Poll(
+            message_id=message.id,
+            channel_id=message.channel.id,
+            question=question,
+            options=answer,
+            user_id=ctx.message.author.id
+        )
+
+        embed = PollEmbed(poll)
+        view = PollView(poll, embed, db_poll=self.bot.pool)
+        await PollDatabase(self.bot.pool).add(poll)
+        await VoteButtonDatabase(self.bot.pool).add_options(poll)
+
+        await message.edit(embed=embed, view=view)
+
+
+async def setup(bot):
+    await bot.add_cog(PollCreate(bot))