[safety checks] test

app.py

@@ -36,41 +36,67 @@ intents.message_content = True
 bot = commands.Bot(command_prefix='!', intents=intents)
 
 
-#----------------------------------------------------------------------------------------------------------------------------------------------
-
+#---------------------------------------------------------------------------------------------------------------------------------------------
 @bot.event
 async def on_ready():
     print('Logged on as', bot.user)
     bot.log_channel = bot.get_channel(1100458786826747945) # 1100458786826747945 = bot-test, 1107006391547342910 = lunarbot server
-
+#---------------------------------------------------------------------------------------------------------------------------------------------
+async def safetychecks(ctx):
+    if ctx.author.bot:
+        return False
+
+    # check if the bot is offline
+    offline_bot_role_id = 1103676632667017266
+    if offline_bot_role_id in bot.roles:
+        return False
+
+    # check if the command is in the allowed channel(s)
+    channel = ctx.message.channel.id
+    if channel != 1100458786826747945:
+        return False
+        
+    # check if the user has the required role(s)   
+    guild = 879548962464493619
+    required_role_id = 900063512829755413  # @verified for now
+    required_role = guild.get_role(required_role_id)
+    if required_role not in ctx.author.roles:
+        return False
+    return True
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
 # jojo
 @bot.command()
 async def jojo(ctx):
-    # img + face ✅
+    # img + face    ✅
     # img + no face ✅
-    # no image ❌
+    # no image      ✅
     # no generation ✅
-    try:    
-        await ctx.message.add_reaction('🤖') 
-        thread = await ctx.message.create_thread(name=f'{ctx.author} Jojo Thread')         
-        if ctx.message.attachments:
-            await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...yare yare, daze ...')  
-            
-            attachment = ctx.message.attachments[0]
-            style = 'JoJo'
-            im = jojogan.predict(attachment.url, style)
-            #await ctx.message.reply(f'Here is the {style} version of it', file=discord.File(im))
+    # responsive?   ✅
+    # ratelimits?   ❌
+    # safety checks?❌
+    # bot no crash  ❌
     
-            await thread.send(f'{ctx.author.mention}Here is the {style} version of it', file=discord.File(im))
-            await ctx.message.add_reaction('✅') # img + face    
-        else: # no image
-            await thread.send(f"{ctx.author.mention}No attachments to be found...Can't animify dat! Try sending me an image 😉")
-            await ctx.message.add_reaction('❌')             
-    except Exception as e: # no generation / img + no face
-        print(f"Error: {e}")
-        await thread.send(f"{ctx.author.mention}Error: {e}")
-        await ctx.message.add_reaction('❌')            
+    if await safetychecks:
+        try:    
+            await ctx.message.add_reaction('🤖') 
+            thread = await ctx.message.create_thread(name=f'{ctx.author} Jojo Thread')         
+            if ctx.message.attachments:
+                await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...yare yare, daze ...')  
+                
+                attachment = ctx.message.attachments[0]
+                style = 'JoJo'
+                im = jojogan.predict(attachment.url, style)
+                #await ctx.message.reply(f'Here is the {style} version of it', file=discord.File(im))
+        
+                await thread.send(f'{ctx.author.mention}Here is the {style} version of it', file=discord.File(im))
+                await ctx.message.add_reaction('✅') # img + face    
+            else: # no image
+                await thread.send(f"{ctx.author.mention}No attachments to be found...Can't animify dat! Try sending me an image 😉")
+                await ctx.message.add_reaction('❌')             
+        except Exception as e: # no generation / img + no face
+            print(f"Error: {e}")
+            await thread.send(f"{ctx.author.mention}Error: {e}")
+            await ctx.message.add_reaction('❌')            
         
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
 # Disney