# Alright peeps! Some of you were curious as to how Mod-Bot worked in our Discord server. I went back in time (on GitHub) and pulled some older versions of a few of the commands (for security reasons, of course). These commands were 100% made by me from scratch. I am the sole author. You may use this code and its logic within your own bots and Discord servers ... BUT ... if you display the code itself, you MUST credit me. Even though this is an older version of the current logic being implemented, I still worked fairly hard on it .. ^-^" .. Have fun guys! Oh, and don't hesitate to experiment with it. Learning is what we're here for, after all. import os, re, discord, asyncio, psycopg2, imgkit, pydf, threading, io # BE SURE TO INSTALL IMGKIT. THAT MODULE MAKES IT ALL POSSIBLE. from discord.ext import commands # MODULE/TOOLS FOR INTERACTION WITH DISCORD. from PIL import Image # MODULE/TOOLS FOR OPERATING ON IMAGES, ETC. from io import BytesIO # PRIMARILY USED (HERE) FOR IMAGE GENERATION. from psycopg2 import * # MODULE/TOOLS FOR POSTGRESQL DBMS TOOLS owner_id = int(input('Enter your Discord ID: ')) bot = commands.Bot(command_prefix = 'bot.', owner_id = owner_id) bot.remove_command('help') # ALWAYS MAKE YOUR OWN HELP-COMMAND. database = psycopg2.connect(' '.join(['%s=%s' % (i, os.environ[i]) for i in ('host', 'port', 'dbname', 'user', 'password')])) # CONNECTING DATABASE VIA PRE-DEFINED ENVIRONMENT VARIABLES. #wkhtmltoimage_binaries = imgkit.config(wkhtmltoimage = '/app/bin/wkhtmltoimage') # ONLY UNCOMMENT THIS IF YOU'RE USING A MAC/LINUX PC. wkhtml_options = {'quiet' : ''} # OUTPUT-OPTIONS FOR IMGKIT MODULE/TOOL, USED FOR SCREENSHOTTING, ETC. database.autocommit = True # DATABASE MODIFICATIONS ARE SAVED IMMEDIATELY. CHANGE THIS TO FALSE FOR MORE SECURITY, BUT BE SURE TO ALSO PLACE `database.commit()` AFTER EVERY MODIFICATION (WITH INCLUDED SECURITY-CHECKS BEFORE THE COMMIT, ETC.). db = database.cursor() @bot.event async def on_ready(): print('The bot and database are connected and running!\n\nORIGINAL AUTHOR: CipherFox/Fox (A.B.Carter)') # THIS IS OUTPUTTED TO YOUR CONSOLE/TERMINAL/COMMAND-PROMPT WHENEVER THE BOT IS BROUGHT ONLINE. IF YOU DO NOT SEE THIS MESSAGE, THEN EITHER YOUR ENVIRONMENT VARIABLES ARE NOT SET CORRECTLY OR THERE IS AN ISSUE WITH COMMUNICATION BETWEEN THE BOT AND THE INTERNET. def build_html(query): # AN HTML STRING THAT BUILDS A VISUAL REPRESENTATION OF YOUR SQL QUERY'S OUTPUT. tmp = '' html_doc = ''' {}

''' db.execute(query) dd = db.fetchall() for i, x in enumerate(([i[0] for i in db.description],) + (*[list(i) for i in dd],)): tmp += '\n' + '\t' * 3 + '\n{}'.format(''.join(['\t' * 4 + '{1}\n'.format('h' if not i else 'd', c) for c in x])) + '\t' * 3 + '' return html_doc.format(tmp) @bot.command(pass_context = 1) # THERE ARE SEVERAL OUTPUT OPTIONS (html, text, txt, pdf, default (PNG)). THE OPTION IS TO BE LISTED AS AN ARGUMENT DIRECTLY AFTER `bot.sql`. EXAMPLE: `bot.sql html SELECT * FROM tablename;`. LEAVE THE OUTPUT-OPTION BLANK FOR AN IMAGE-OUTPUT. async def sql(ctx): query = ' '.join(ctx.message.content.split()[1:]) if 'html' == query.split()[0].lower(): try: html_doc = build_html(' '.join(query.split()[1:])) f = open('queried_data.html', 'w') f.write(html_doc) f.close() await ctx.send(file = discord.File('queried_data.html', filename = 'queried_data.html')) os.system('rm queried_data.html') del f except discord.errors.HTTPException: await ctx.send(':exclamation: **`The file\'s size is too large to send. Try modifying the layout of your query so that you may produce accurate results.`**') elif 'text' == query.split()[0].lower(): try: q = '' db.execute(' '.join(query.split()[1:])) for i in db.fetchall(): q += ''.join(str(i).strip('( ,)')) + '\n' emb = discord.Embed(title = msg, description = q, color = 0x3338e6) await ctx.send(embed = emb) del q except discord.errors.HTTPException: await ctx.send(':exclamation: **`The queried data cannot be displayed due to it exceeding Discord\'s char-limit (2000).`**') elif 'txt' == query.split()[0].lower(): try: q = '' db.execute(' '.join(query.split()[1:])) for i in db.fetchall(): q += ''.join(str(i).strip('( ,)')) + '\n' query_data = open('queried_data.txt', 'w') query_data.write(q) query_data.close() await ctx.send(file = discord.File('queried_data.txt', filename = 'queried_data.txt')) os.system('rm queried_data.txt') del query_data except discord.errors.HTTPException: await ctx.send(':exclamation: **`The file\'s size is too large to send. Try modifying the layout of your query so that you may produce accurate results.`**') elif 'pdf' == query.split()[0].lower(): try: html_doc = build_html(' '.join(query.split()[1:])) pdf = pydf.generate_pdf(html_doc) with open('queried_data.pdf', 'wb') as f: f.write(pdf) await ctx.send(file = discord.File('queried_data.pdf', filename = 'queried_data.pdf')) except discord.errors.HTTPException: await ctx.send(':exclamation: **`The file\'s size is too large to send. Try modifying the layout of your query so that you may produce accurate results.`**') else: try: html_doc = build_html(query) img = imgkit.from_string(html_doc, False, config = wkhtmltoimage_binaries, options = wkhtml_options) bytes_io_img = BytesIO(img) pImg = Image.open(bytes_io_img) imgfileObj = BytesIO() pImg.save(imgfileObj, format = 'png') imgfileObj.seek(0) await ctx.send(file = discord.File(imgfileObj.read(), filename = 'query.png')) del html_doc except discord.errors.HTTPException: await ctx.send(':exclamation: **`The image\'s size is too large to send. Try modifying the layout of your query so that you may produce accurate results.`**') @bot.command(pass_context = 1) # WEAR THE BOT AS A MASK. async def repeat(ctx, *, args : str = None): await ctx.message.delete() await ctx.send(args) @bot.command(pass_context = 1) # USED IF YOU WISH TO END THE CONNECTION BETWEEN THE BOT AND THE DATABASE. async def database_logout(ctx): await ctx.message.delete() db.close() database.close() embed = discord.Embed(title = 'DATABASE CONNECTION CLOSED!', description = 'The bot\'s database connection has been manually closed due to unexpected circumstances!\n\nPlease refer to the logs for more information!', color = 0xff0000) embed.set_footer(text = 'Database closed by ' + str(ctx.message.author)) await ctx.send(embed = embed) @bot.command(pass_context = 1) # USED IF YOU WISH TO OFFLINE THE BOT. async def shutdown(ctx): emb = discord.Embed(description = ':exclamation: The bot is shutting down ...', color = 0xffffff) emb.set_footer(text = 'Executed by %s' % ctx.message.author.name) await ctx.send(embed = emb) await bot.logout() @bot.command(pass_context = 1) # TEST THE RESPONSE-TIME OF THE BOT. async def ping(ctx): before = time.monotonic() message = await ctx.send("**Ping-time:**") ping = (time.monotonic() - before) * 1000 await message.edit(content = "**Ping-time:** `%.2f milliseconds`" % ping) @bot.command(pass_context = 1) # YOUR CUSTOM HELP-MENU. async def help(ctx, option = None): cmds = """ **__BOT COMMANDS__**\n `database_logout` `help [ COMMAND [ ARG ] ]` `ping` `repeat { *MESSAGE }` `shutdown` `sql [ html | pdf | text | txt ] { *QUERY }`\n""" options = { None : cmds, 'database_logout' : '`database_logout`', 'help' : '`help [ COMMAND [ ARG ] ]`', 'ping' : '`ping`', 'repeat' : '`repeat { *MESSAGE }`', 'shutdown' : '`shutdown`', 'sql' : '`sql [ html | pdf | text | txt ] { *QUERY }`', } await ctx.send(embed = discord.Embed(description = options[option], color = 0xfffff)) if __name__ == '__main__': # KEEPS THE BOT BREATHING. loop = asyncio.get_event_loop() loop.create_task(bot.start(os.environ['bot_token_env_var'])) loop.run_forever()

Fun Discord Bot Commands!