Update bot.py

3 years ago · 3bd26c0663
1 changed files with 72 additions and 35 deletions
--- a/bot.py
+++ b/bot.py
@ -15,7 +15,26 @@ bot = commands.Bot(command_prefix=".", intents=discord.Intents.default(),
 load_dotenv()

 token = os.getenv('token')
+crabwings_role_id = int(os.getenv('crabwings_role_id'))
+duckfeet_role_id = int(os.getenv('duckfeet_role_id'))
+elktail_role_id = int(os.getenv('elktail_role_id'))
+client_role_id = int(os.getenv('client_role_id'))
+subuser_role_id = int(os.getenv('subuser_role_id'))
+verified_role_id = int(os.getenv('verified_role_id'))
 guild_id = int(os.getenv('guild_id'))
+verification_channel = int(os.getenv('verification_channel'))
+verification_message = int(os.getenv('verification_message'))
+application_api_key = os.getenv('application_api_key')
+running_on_panel = str(os.getenv('running_on_panel'))
+crabwings_ip = os.getenv('crabwings_ip')
+crabwings_port = int(os.getenv('crabwings_port'))
+crabwings_username = os.getenv('crabwings_username')
+crabwings_password = os.getenv('crabwings_password')
+
+if running_on_panel == "False":
+ running_on_panel = False
+else:
+ running_on_panel = True

 logging.basicConfig(filename='console.log',
 level=logging.INFO,
@ -32,6 +51,7 @@ async def on_ready():

@bot.event
 async def on_message(message):
+ if not running_on_panel:
 # Binflop
 if len(message.attachments) > 0:
 if not message.attachments[0].url.endswith(
@ -64,19 +84,36 @@ async def on_message(message):

@bot.command()
 async def ping(ctx):
- await ctx.send(f'Botflop\'s ping is {round(bot.latency * 1000)}ms')
+ if running_on_panel:
+ if guild_id == ctx.guild.id:
+ await ctx.send(f'Private bot ping is {round(bot.latency * 1000)}ms')
+ if not running_on_panel:
+ await ctx.send(f'Public bot ping is {round(bot.latency * 1000)}ms')

@bot.command(name="react", pass_context=True)
@has_permissions(administrator=True)
 async def react(ctx, url, reaction):
+ if not running_on_panel:
 channel = await bot.fetch_channel(int(url.split("/")[5]))
 message = await channel.fetch_message(int(url.split("/")[6]))
 await message.add_reaction(reaction)
 logging.info('reacted to ' + url + ' with ' + reaction)

-for file_name in os.listdir('./cogs'):
- if file_name.endswith('.py'):
+
+if running_on_panel:
+ for file_name in os.listdir('./cogs'):
+ if file_name.endswith('_panel.py'):
 bot.load_extension(f'cogs.{file_name[:-3]}')
+else:
+ for file_name in os.listdir('./cogs'):
+ if file_name.endswith('_public.py'):
+ bot.load_extension(f'cogs.{file_name[:-3]}')
+
+if running_on_panel:
+ print("running on panel, starting loops")
+ updater.start()
+ linking_updater = bot.get_cog('Linking_updater')
+ linking_updater.linking_updater.start()

 bot.run(token)