From ecb5986858d4d832c81c9da78669c25267282589 Mon Sep 17 00:00:00 2001
From: Paillat <git@electronique.cc>
Date: Sat, 10 Dec 2022 00:28:42 +0100
Subject: [PATCH] Create premiumcode.py

---
 code/premiumcode.py | 59 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 code/premiumcode.py

diff --git a/code/premiumcode.py b/code/premiumcode.py
new file mode 100644
index 0000000..07fd2bd
--- /dev/null
+++ b/code/premiumcode.py
@@ -0,0 +1,59 @@
+import discord # pip install pycord
+import asyncio # pip install asyncio
+import sqlite3 # pip install sqlite3
+import logging # pip install logging
+import os # pip install os
+intents = discord.Intents.all()
+conn = sqlite3.connect('premium.db')
+c = conn.cursor()
+c.execute('''CREATE TABLE IF NOT EXISTS data (user_id text, guild_id text, premium boolean)''')
+bot = discord.Bot()
+logging.basicConfig(level=logging.INFO)
+@bot.command()
+@discord.commands.option(name="server id", description="The server id for which you want to activate premium features", required=True)
+async def activate_premium(ctx, server_id):
+    #first check if the user is already in the database, select guuild_id and premium from the data table where user_id is the author's id
+    logging.info("Activating premium for user " + str(ctx.author.id))
+    c.execute("SELECT guild_id, premium FROM data WHERE user_id = ?", (ctx.author.id,))
+    #if a guild_id is found, override the old settings with the new ones
+    if c.fetchone() is not None:
+        c.execute("UPDATE data SET guild_id = ?, premium = ? WHERE user_id = ?", (server_id, True, ctx.author.id))
+        conn.commit()
+        logging.info("Premium activated for server " + server_id + "by user " + str(ctx.author.id))
+        await ctx.respond("Premium activated for server " + server_id, ephemeral=True)
+    #if no guild_id is found, insert the new settings
+    else:
+        c.execute("INSERT INTO data VALUES (?, ?, ?)", (ctx.author.id, server_id, True))
+        conn.commit()
+        logging.info("Premium updated for server " + server_id + "by user " + str(ctx.author.id))
+        await ctx.respond("Premium activated for server " + server_id, ephemeral=True)
+
+#each 24 hours, check if each user if they have the premium role "1050823446445178900" in the server "1050769643180146749"
+async def check_premium():
+    while True:
+        #select user_id and guild_id from the data table
+        c.execute("SELECT user_id, guild_id FROM data")
+        for row in c.fetchall():
+            #get the guild and the user
+            guild = bot.get_guild(int(row[1]))
+            user = guild.get_member(int(row[0]))
+            #if the user has the premium role, set premium to true
+            logging.info("Checking premium for user " + str(row[0]))
+            if discord.utils.get(user.roles, id=1050823446445178900) is not None:
+                c.execute("UPDATE data SET premium = ? WHERE user_id = ?", (True, row[0]))
+                conn.commit()
+                logging.info("Premium activated for server " + str(row[1]) + "by user " + str(row[0]))
+            #if the user does not have the premium role, set premium to false
+            else:
+                c.execute("UPDATE data SET premium = ? WHERE user_id = ?", (False, row[0]))
+                conn.commit()
+                logging.info("Premium deactivated for server " + str(row[1]) + "by user " + str(row[0]))
+        await asyncio.sleep(86400)
+
+#add a task to the bot that runs check_premium every 24 hours
+bot.loop.create_task(check_premium())
+#run the bot
+# Replace the following with your bot's token
+with open("./premium-key.txt") as f:
+    key = f.read()
+bot.run(key)
\ No newline at end of file