Implement Members list command

This commit is contained in:
Jeremy Zhang 2018-08-11 06:08:31 +00:00
parent 6e886bcf1c
commit 5ae32aeb89
2 changed files with 51 additions and 1 deletions

View File

@ -1,4 +1,5 @@
import aiohttp
import discord
class Commands():
def __init__(self, client, config):
@ -73,4 +74,45 @@ class Commands():
await message.channel.send("Join the Titan Embeds Discord server! https://discord.gg/pFDDtcN")
async def shard(self, message):
await message.channel.send("This instance of Titan Embeds Discord Bot is running on shard **{}**. There are **{}** shards in total.".format(message.guild.shard_id, self.client.shard_count))
await message.channel.send("This instance of Titan Embeds Discord Bot is running on shard **{}**. There are **{}** shards in total.".format(message.guild.shard_id, self.client.shard_count))
async def members(self, message):
headers = {"Authorization": self.config["titan-web-app-secret"]}
payload = {
"guild_id": message.guild.id,
}
users = {"authenticated": [], "unauthenticated": []}
url = self.config["titan-web-url"] + "api/bot/members"
async with aiohttp.ClientSession() as aioclient:
async with aioclient.get(url, params=payload, headers=headers) as resp:
if resp.status >= 200 and resp.status < 300:
users = await resp.json()
embed_description = ""
if users["authenticated"]:
embed_description = embed_description + "__(Discord)__\n"
count = 1
for user in users["authenticated"]:
server_user = message.guild.get_member(int(user["id"]))
embed_description = embed_description + "**{}.** {}#{}".format(count, server_user.name, server_user.discriminator)
if server_user.nick:
embed_description = embed_description + " ({})".format(server_user.nick)
embed_description = embed_description + " {}\n".format(server_user.mention)
count = count + 1
if users["unauthenticated"]:
if users["authenticated"]:
embed_description = embed_description + "\n"
embed_description = embed_description + "__(Guest)__\n"
count = 1
for user in users["unauthenticated"]:
embed_description = embed_description + "**{}.** {}#{}\n".format(count, user["username"], user["discriminator"])
count = count + 1
if users["authenticated"] or users["unauthenticated"]:
embed_description = embed_description + "\n"
embed_description = embed_description + "**Total Members Online: __{}__**".format(len(users["authenticated"]) + len(users["unauthenticated"]))
embed = discord.Embed(
title = "Currently Online Embed Members",
url = "https://TitanEmbeds.com/",
color = 7964363,
description = embed_description
)
await message.channel.send(embed=embed)

View File

@ -617,6 +617,14 @@ def bot_revoke():
db.session.commit()
return jsonify(success="Successfully kicked **{}#{}**!".format(dbuser.username, dbuser.discriminator))
@api.route("/bot/members")
def bot_members():
if request.headers.get("Authorization", "") != config.get("app-secret", ""):
return jsonify(error="Authorization header does not match."), 403
guild_id = request.args.get("guild_id")
members = get_online_embed_users(guild_id)
return jsonify(members)
@api.route("/af/direct_message", methods=["POST"])
def af_direct_message_post():
cs = request.form.get('cs', None)