roxxers
/
roxbot
mirror of https://github.com/roxxers/roxbot
Watch 1
Star 0
Fork 0
Code Issues 0 Releases 38 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
975 Commits
8 Branches
Tree: 58276d2799
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '58276d2799'
${ noResults }
roxbot/roxbot/cogs/util.py

219 lines
9.5KB
Raw Blame History 

  1. # -*- coding: utf-8 -*-

  2. 

  3. # MIT License

  4. #

  5. # Copyright (c) 2017-2018 Roxanne Gibson

  6. #

  7. # Permission is hereby granted, free of charge, to any person obtaining a copy

  8. # of this software and associated documentation files (the "Software"), to deal

  9. # in the Software without restriction, including without limitation the rights

  10. # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

  11. # copies of the Software, and to permit persons to whom the Software is

  12. # furnished to do so, subject to the following conditions:

  13. #

  14. # The above copyright notice and this permission notice shall be included in all

  15. # copies or substantial portions of the Software.

  16. #

  17. # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

  18. # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

  19. # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

  20. # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

  21. # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

  22. # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

  23. # SOFTWARE.

  24. 

  25. 

  26. import os

  27. import re

  28. 

  29. import discord

  30. import emoji

  31. from discord.ext import commands

  32. 

  33. import roxbot

  34. 

  35. 

  36. class Util(commands.Cog):

  37.     """

  38.     The Util cog is a cog filled with a number of utility commands to help more advanced users of Discord.

  39.     """

  40.     def __init__(self, bot_client):

  41.         self.bot = bot_client

  42. 

  43.     @commands.command()

  44.     async def avatar(self, ctx, *, user: discord.User = None):

  45.         """

  46.         Uploads a downloadable picture of an avatar.

  47. 

  48.         Example:

  49.             # Get my avatar

  50.             ;avatar

  51.             # Get USER's avatar

  52.             ;avatar USER#0001

  53.         """

  54.         if not user:

  55.             user = ctx.author

  56. 

  57.         url = user.avatar_url_as(static_format="png")

  58.         filename = re.sub(' |\?|\.|!|/|\\|:|\"|\[|\]|;|=|\||\*|,', '', user.name+str(user.id))

  59. 

  60.         if ".gif" in url:

  61.             avaimg = '{0}.gif'.format(filename)

  62.         else:

  63.             avaimg = '{0}.png'.format(filename)

  64. 

  65.         await roxbot.http.download_file(url, avaimg)

  66.         await ctx.send(file=discord.File(avaimg))

  67.         os.remove(avaimg)

  68. 

  69.     @commands.command(aliases=["user"])

  70.     async def info(self, ctx, member: discord.Member = None):

  71.         """

  72.         Provides information (account creation date, ID, roles [if in a guild]) on your or another persons account.

  73. 

  74.         Example:

  75.             # Get account information for yourself

  76.             ;info

  77.             # Get account information for a user called USER

  78.             ;info @USER

  79.         """

  80.         if not member:

  81.             member = ctx.author

  82.         if member.activity:

  83.             if member.activity.type == discord.ActivityType.playing:

  84.                 activity = "Playing **{}**".format(member.activity.name)

  85.             elif member.activity.type == discord.ActivityType.streaming:

  86.                 activity = "Streaming **{}**".format(member.activity.name)

  87.             elif member.activity.type == discord.ActivityType.listening:

  88.                 activity = "Listening to **{} by {}**".format(member.activity.title, member.activity.artist)

  89.             else:

  90.                 activity = ""

  91.         else:

  92.             activity = ""

  93. 

  94.         colour = member.colour.value

  95.         avatar = member.avatar_url

  96. 

  97.         embed = discord.Embed(colour=colour, description=activity)

  98.         embed.set_thumbnail(url=avatar)

  99.         embed.set_author(name=str(member), icon_url=avatar)

  100. 

  101.         embed.add_field(name="ID", value=member.id)

  102.         embed.add_field(name="Status", value=member.status)

  103.         if member.nick:

  104.             embed.add_field(name="Nickname", value=member.nick)

  105.         embed.add_field(name="Account Created", value="{:%a %Y/%m/%d %H:%M:%S} UTC".format(member.created_at), inline=True)

  106.         embed.add_field(name="Joined Server", value="{:%a %Y/%m/%d %H:%M:%S} UTC".format(member.joined_at), inline=True)

  107. 

  108.         roles = ""

  109.         count = 0

  110. 

  111.         for role in member.roles:

  112.             if role == ctx.guild.default_role:

  113.                 pass

  114.             else:

  115.                 roles += role.name + ", "

  116.                 count += 1

  117.         if not roles:

  118.             roles = "None"

  119.             count = 0

  120.         embed.add_field(name="Roles [{}]".format(count), value=roles.strip(", "))

  121.         return await ctx.send(embed=embed)

  122. 

  123.     @commands.guild_only()

  124.     @commands.command(aliases=["server"])

  125.     async def guild(self, ctx):

  126.         """Gives information (creation date, owner, ID) on the guild this command is executed in."""

  127.         guild = ctx.guild

  128.         guild_icon_url = "https://cdn.discordapp.com/icons/{}/{}.png".format(guild.id, guild.icon)

  129.         guild_splash_url = "https:/cdn.discordapp.com/splashes/{}/{}.png".format(guild.id, guild.splash)

  130.         embed = discord.Embed(title=guild.name, colour=guild.me.colour.value)

  131.         embed.set_thumbnail(url=guild_icon_url)

  132.         embed.add_field(name="ID", value=guild.id, inline=False)

  133.         embed.add_field(name="Owner", value="{} ({})".format(self.bot.get_user(guild.owner_id), guild.owner_id), inline=False)

  134.         embed.add_field(name="Created at", value="{:%a %Y/%m/%d %H:%M:%S} UTC".format(guild.created_at, inline=False))

  135.         embed.add_field(name="Voice Region", value=guild.region, inline=False)

  136.         embed.add_field(name="AFK Timeout", value="{} Minutes".format(guild.afk_timeout/60), inline=False)

  137.         if guild.afk_channel:

  138.             embed.add_field(name="AFK Channel", value=guild.afk_channel, inline=False)

  139. 

  140.         embed.add_field(name="Verification Level", value=guild.verification_level, inline=False)

  141.         embed.add_field(name="Explicit Content Filtering", value=guild.explicit_content_filter, inline=False)

  142.         embed.add_field(name="Members", value=guild.member_count)

  143.         number_of_bots = 0

  144.         for member in guild.members:

  145.             if member.bot:

  146.                 number_of_bots += 1

  147.         human_members = guild.member_count - number_of_bots

  148.         embed.add_field(name="Human Members", value=human_members, inline=False)

  149.         embed.add_field(name="Roles".format(), value=str(len(guild.roles)), inline=False)

  150.         embed.add_field(name="Emotes".format(), value=str(len(guild.emojis)), inline=False)

  151.         embed.add_field(name="Channels [{}]".format(len(guild.channels)), value="{} Channel Categories\n{} Text Channels\n{} Voice Channels".format(len(guild.categories), len(guild.text_channels), len(guild.voice_channels)))

  152. 

  153.         if guild.features:

  154.             embed.add_field(name="Extra Features", value=guild.features, inline=False)

  155.         if guild.splash:

  156.             embed.set_image(url=guild_splash_url)

  157. 

  158.         return await ctx.send(embed=embed)

  159. 

  160.     @commands.guild_only()

  161.     @commands.command()

  162.     async def role(self, ctx, *, role: discord.Role):

  163.         """Gives information (creation date, colour, ID) on the role given. Can only work if the role is in the guild you execute this command in.

  164.         Examples:

  165.             # Get information on the role called Admin

  166.             ;role Admin

  167.         """

  168.         embed = discord.Embed(title="Role '{}'".format(role.name), colour=role.colour.value)

  169.         embed.add_field(name="ID", value=role.id, inline=False)

  170.         embed.add_field(name="Created at", value="{:%a %Y/%m/%d %H:%M:%S} UTC".format(discord.utils.snowflake_time(role.id)), inline=False)

  171.         embed.add_field(name="Colour", value="#{}".format(str(hex(role.colour.value)).strip("0x")), inline=False)

  172.         embed.add_field(name="Hoisted", value=str(role.hoist), inline=False)

  173.         embed.add_field(name="Managed", value=str(role.managed), inline=False)

  174.         return await ctx.send(embed=embed)

  175. 

  176.     @commands.command(aliases=["emoji"])

  177.     async def emote(self, ctx, emote):

  178.         """

  179.         Displays information (creation date, guild, ID) and an easily downloadable version of the given custom emote.

  180. 

  181.         Example:

  182.             # Get information of the emoji ":Kappa:"

  183.             ;emote :Kappa:

  184.         """

  185.         try:  # If emote given is custom emote

  186.             emote = await roxbot.converters.Emoji().convert(ctx, emote)

  187.             em = discord.Embed(title=emote.name, colour=roxbot.EmbedColours.blue)

  188.             em.add_field(name="ID", value=str(emote.id), inline=False)

  189.             if isinstance(emote, discord.Emoji):

  190.                 em.add_field(name="Guild", value=str(emote.guild), inline=False)

  191.                 em.add_field(name="Created At", value=roxbot.datetime.format(emote.created_at), inline=False)

  192.             em.set_image(url=emote.url)

  193.             return await ctx.send(embed=em)

  194.         except commands.errors.BadArgument:  # unicode emoji

  195.             title = emoji.demojize(emote)

  196.             if not emoji.EMOJI_UNICODE.get(title):

  197.                 raise commands.BadArgument("Could not convert input to either unicode emoji or Discord custom emote.")

  198. 

  199.             emojis = []

  200.             for char in emote:

  201.                 emojis.append(hex(ord(char))[2:])

  202. 

  203.             if len(emojis) > 1:

  204.                 svg_url = "https://twemoji.maxcdn.com/2/svg/{0}-{1}.svg".format(*emojis)

  205.                 png_url = "https://twemoji.maxcdn.com/2/72x72/{0}-{1}.png".format(*emojis)

  206.             else:

  207.                 svg_url = "https://twemoji.maxcdn.com/2/svg/{0}.svg".format(*emojis)

  208.                 png_url = "https://twemoji.maxcdn.com/2/72x72/{0}.png".format(*emojis)

  209. 

  210.             em = discord.Embed(title=title, colour=roxbot.EmbedColours.blue)

  211.             em.description = "[SVG Link]({0})\n[PNG Link]({1})".format(svg_url, png_url)

  212.             em.set_image(url=png_url)

  213. 

  214.             return await ctx.send(embed=em)

  215. 

  216. 

  217. def setup(bot_client):

  218.     bot_client.add_cog(Util(bot_client))