Start youtube notifications

Author: gccody

lib/bot/__init__.py

@@ -60,7 +60,7 @@ async def on_disconnect(self):
         print(f"Bot Disconnected, Retrying to sign into user {self.user.display_name}#{self.user.discriminator}")
 
     async def on_guild_join(self, guild: Guild):
-        self.db.execute("""INSERT OR IGNORE INTO guilds VALUES (?,?)""", guild.id, None)
+        self.db.execute("""INSERT OR IGNORE INTO guilds VALUES (?,?,?)""", guild.id, None, None)
 
     async def on_guild_remove(self, guild: Guild):
         self.db.execute("""DELETE FROM guilds WHERE id=?""", guild.id)
@@ -90,7 +90,7 @@ async def on_interaction(self, interaction: discord.Interaction):
         if interaction.is_command():
             name_list, options = self.get_name(interaction.data, [])
             name = " ".join(name_list)
-            self.db.execute(f"""INSERT INTO commands VALUES (?,?,?,?,?)""", name, interaction.guild_id, interaction.user.id, json.dumps(options), datetime.now().timestamp())
+            # self.db.execute(f"""INSERT INTO commands VALUES (?,?,?,?,?)""", name, interaction.guild_id, interaction.user.id, json.dumps(options), datetime.now().timestamp())
         return await super().on_interaction(interaction=interaction)
 
     async def on_application_command_error(self, ctx: CustomContext, exc) -> None:

lib/cogs/Youtube.py

@@ -0,0 +1,76 @@
+import re
+
+from discord.ext.commands import Cog
+from discord.commands import slash_command, SlashCommandGroup, option
+from discord.embeds import Embed
+import aiohttp
+
+from bs4 import BeautifulSoup
+
+from lib.bot import Bot
+from lib.context import CustomContext
+
+
+class Youtube(Cog):
+    def __init__(self, bot):
+        self.bot: Bot = bot
+
+    youtube = SlashCommandGroup('youtube')
+
+    def check_channel(self, handle: str, message: str, guild_id: str):
+        async with aiohttp.ClientSession() as session:
+            res = await session.get(f"https://youtube.com/@{handle}/videos")
+            html = await res.text()
+            try:
+                latest_video_url = "https://www.youtube.com/watch?v=" + re.search('(?<="videoId":").*?(?=")',
+                                                                                  html).group()
+            except AttributeError:
+                latest_video_url = ""
+
+
+    @youtube.command(name='notify', description='Get a ping when this user uploads')
+    @option('handle',
+            str,
+            description='Youtube handle',
+            required=True)
+    @option('message',
+            str,
+            description='Message to send when video is uploaded',
+            required=False)
+    @option('channel',
+            str,
+            description='Channel to send announcement',
+            required=False)
+    async def youtube_notify(self, ctx: CustomContext, handle: str, message: str | None):
+        async with aiohttp.ClientSession() as session:
+            res = await session.get(f"https://youtube.com/@{handle}/videos")
+            if res.status == 404:
+                return await ctx.respond(embed=Embed(title=':x: | Channel not found', colour=0xff0000))
+            elif res.status not in range(200, 300):
+                return await ctx.respond(embed=Embed(title=':x: | Something went wrong', colour=0xff0000))
+
+            user = self.bot.db.record("""SELECT * FROM youtube WHERE handle = ? AND guild_id = ?""", handle, ctx.guild_id)
+            if user:
+                return await ctx.respond(embed=Embed(title=':x: | Already watching for new videos', colour=0xff0000))
+
+            res = await session.get(f"https://youtube.com/@{handle}/videos")
+            html = await res.text()
+
+        try:
+            latest_video_url = "https://www.youtube.com/watch?v=" + re.search('(?<="videoId":").*?(?=")', html).group()
+        except AttributeError:
+            latest_video_url = ""
+        self.bot.db.execute("""INSERT INTO youtube VALUES (?,?,?,?)""", handle, ctx.guild_id, message if message else f"@{handle} has uploaded a video", latest_video_url)
+        await ctx.respond(embed=Embed(title=f'✅ | Waiting for @{handle} to post more videos'))
+        self.bot.tasks.add_job(id=f"{handle}|{ctx.guild_id}", args=(handle, message, ctx.guild_id,), trigger='interval', minutes=1)
+
+
+
+
+
+
+
+
+
+def setup(bot):
+    bot.add_cog(Youtube(bot))

lib/db/__init__.py

@@ -17,8 +17,8 @@ class DB:
         },
         'guilds': {
             'id': 'TEXT PRIMARY KEY NOT NULL',
-            'l_id': 'TEXT',
-            'a_id': 'TEXT'
+            'level_id': 'TEXT',
+            'announce_id': 'TEXT'
         },
         'reports': {
             'guild_id': 'TEXT',
@@ -37,22 +37,24 @@ class DB:
             'guild_id': 'TEXT',
             'user_id': 'TEXT',
             'xp': 'INTEGER'
+        },
+        'youtube': {
+            'handle': 'TEXT',
+            'guild_id': 'TEXT',
+            'message': 'TEXT',
+            'latest_url': 'TEXT'
         }
     }
 
     def __init__(self):
         self.cxn = connect(self.DB_PATH, check_same_thread=False)
         self.cur = self.cxn.cursor()
 
-    @staticmethod
-    def diff(list1, list2):
-        return list(set(list1).symmetric_difference(set(list2)))
-
     def update_table(self, table: str, columns_str_list: list[str], columns_list: dict[str, str]):
         self.cur.execute(f"CREATE TABLE {table}_new ({' '.join(columns_str_list)});")
         self.cur.execute(f"INSERT INTO {table}_new SELECT * FROM {table};")
         self.cur.execute(f"DROP TABLE {table};")
-        self.cur.execute(f"ALTER TABLE {table}_new RENAME TO {table}")
+        self.cur.execute(f"ALTER TABLE {table}_new RENAME TO {table};")
 
     def build(self) -> None:
         for table, columns in self.tables.items():

template.txt

@@ -1,5 +1,6 @@
 from discord.ext.commands import Cog
-from discord.commands import slash_command
+from discord.commands import slash_command, SlashCommandGroup, option
+from discord.embeds import Embed
 
 from lib.bot import Bot
 from lib.context import CustomContext