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,51 +51,69 @@ async def on_ready():

@bot.event
 async def on_message(message):
- # Binflop
- if len(message.attachments) > 0:
- if not message.attachments[0].url.endswith(
- ('.png', '.jpg', '.jpeg', '.mp4', '.mov', '.avi', '.gif', '.image')):
- download = message.attachments[0].url
- async with aiohttp.ClientSession() as session:
- async with session.get(download, allow_redirects=True) as r:
-
- # r = requests.get(download, allow_redirects=True)
- text = await r.text()
- text = "\n".join(text.splitlines())
- if '�' not in text: # If it's not an image/gif
- truncated = False
- if len(text) > 100000:
- text = text[:99999]
- truncated = True
- req = requests.post('https://bin.birdflop.com/documents', data=text)
- key = json.loads(req.content)['key']
- response = ""
- response = response + "https://bin.birdflop.com/" + key
- response = response + "\nRequested by " + message.author.mention
- if truncated:
- response = response + "\n(file was truncated because it was too long.)"
- embed_var = discord.Embed(title="Please use a paste service", color=0x1D83D4)
- embed_var.description = response
- await message.channel.send(embed=embed_var)
+ if not running_on_panel:
+ # Binflop
+ if len(message.attachments) > 0:
+ if not message.attachments[0].url.endswith(
+ ('.png', '.jpg', '.jpeg', '.mp4', '.mov', '.avi', '.gif', '.image')):
+ download = message.attachments[0].url
+ async with aiohttp.ClientSession() as session:
+ async with session.get(download, allow_redirects=True) as r:
+
+ # r = requests.get(download, allow_redirects=True)
+ text = await r.text()
+ text = "\n".join(text.splitlines())
+ if '�' not in text: # If it's not an image/gif
+ truncated = False
+ if len(text) > 100000:
+ text = text[:99999]
+ truncated = True
+ req = requests.post('https://bin.birdflop.com/documents', data=text)
+ key = json.loads(req.content)['key']
+ response = ""
+ response = response + "https://bin.birdflop.com/" + key
+ response = response + "\nRequested by " + message.author.mention
+ if truncated:
+ response = response + "\n(file was truncated because it was too long.)"
+ embed_var = discord.Embed(title="Please use a paste service", color=0x1D83D4)
+ embed_var.description = response
+ await message.channel.send(embed=embed_var)
 timings = bot.get_cog('Timings')
 await timings.analyze_timings(message)
 await bot.process_commands(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):
- 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'):
- bot.load_extension(f'cogs.{file_name[:-3]}')
+ 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)
+
+
+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)