Exemplo n.º 1
0
 async def apod(self, ctx: aoi.AoiContext, *, date: dtime() = None):
     if not date:
         date = datetime.datetime.now()
     dt = date.strftime('%Y-%m-%d')
     if dt not in self.apod_cache:
         async with ctx.typing():
             async with aiohttp.ClientSession() as sess:
                 async with sess.get(
                         f"https://api.nasa.gov/planetary/apod?api_key={self.bot.nasa}&"
                         f"date={dt}") as resp:
                     js = await resp.json()
         if js.get("code", None) in [404, 400, 403, 401]:
             self.apod_cache[dt] = (str(js["code"]), "404", "404",
                                    js["msg"])
             return await ctx.send_error(js["msg"])
         url = js["url"]
         hdurl = js.get("hdurl", url)
         expl = js["explanation"][:1900]
         title = js["title"] + " " + dt
         self.apod_cache[dt] = (title, hdurl, url, expl)
     else:
         title, hdurl, url, expl = self.apod_cache[dt]
         if title == "404":
             await ctx.send_error(expl)
     await ctx.embed(
         title=title,
         description=
         f"{expl}\n\n[Normal Resolution]({url})  [High Resolution]({hdurl})",
         image=url)
Exemplo n.º 2
0
 async def wxhourly(self, ctx: aoi.AoiContext, *,
                    location: gmaps.LocationCoordinates):
     async with ctx.typing():
         conditions = (await self.wx.lookup_hourly(location))
     await ctx.paginate(
         fmt=f"Resolved Address: {location.location or location}```%s```\n",
         lst=[cond.line() for cond in conditions],
         n=24,
         title="Weather lookup",
         thumbnails=[c.icon for c in conditions[3::24]])
Exemplo n.º 3
0
 async def deletequote(self, ctx: aoi.AoiContext, quote: int):
     qid, user, guild = (await (await self.bot.db.conn.execute(
         "select id, user, guild from quotes where id=?",
         (quote, ))).fetchone())
     if user != ctx.author.id and not ctx.author.guild_permissions.administrator:
         return ctx.send_error(
             "You must be administrator to delete quotes that aren't yours."
         )
     await self.bot.db.conn.execute("delete from quotes where id=?",
                                    (qid, ))
     await self.bot.db.conn.commit()
Exemplo n.º 4
0
 async def warnp(self,
                 ctx: aoi.AoiContext,
                 warns: int,
                 *,
                 action: str = None):
     if not action:
         await self.bot.db.del_warnp(ctx.guild.id, warns)
         return await ctx.send(
             f"No punishment will be applied at {warns} warns")
     res = await self._validate_warnp(action)
     if res:
         return ctx.send_error(res)
     await self.bot.db.set_warnp(ctx.guild.id, warns, action)
     return await ctx.send_ok(f"`{action}` will be applied at {warns} warns"
                              )
Exemplo n.º 5
0
 async def landsat(self,
                   ctx: aoi.AoiContext,
                   coords: gmaps.LocationCoordinates,
                   date: dtime() = None):
     lat = coords.lat
     long = coords.long
     dt = date or datetime.datetime.now()
     url = f"https://api.nasa.gov/planetary/earth/imagery?" \
           f"lon={long}&lat={lat}&dim=0.15&api_key={self.bot.nasa}" \
           f"&date={dt.strftime('%Y-%m-%d')}"
     buf = io.BytesIO()
     async with ctx.typing():
         async with aiohttp.ClientSession() as sess:
             async with sess.get(url) as resp:
                 buf.write(await resp.content.read())
     await ctx.embed(title=f"{lat} {long} {dt.strftime('%Y-%m-%d')}",
                     image=buf)
Exemplo n.º 6
0
 async def punishmentclear(self,
                           ctx: aoi.AoiContext,
                           member: discord.Member,
                           num: int = 1):
     punishments: List[PunishmentModel] = sorted(
         await self.bot.db.lookup_punishments(member.id),
         key=lambda punishment: punishment.time,
         reverse=True)
     if num < 1 or num > len(punishments):
         return ctx.send_error("Invalid warning number")
     p = punishments[num - 1]
     if "del" in ctx.flags:
         await self.bot.db.conn.execute(
             "delete from punishments where user=? and guild=? and timestamp=?",
             (p.user, p.guild, p.time.timestamp()))
     else:
         await self.bot.db.conn.execute(
             "update punishments set cleared=1,cleared_by=? where user=? and guild=? "
             "and timestamp=?",
             (ctx.author.id, p.user, p.guild, p.time.timestamp()))
     await self.bot.db.conn.commit()
     await ctx.send_ok(f"Cleared punishment #{num} for {member}")
Exemplo n.º 7
0
 async def addrero(self, ctx: aoi.AoiContext, message: discord.Message, *,
                   args: str):  # noqa c901
     split = ctx.group_list(args.split(), 2)
     role_converter = commands.RoleConverter()
     for i in split:
         print(i)
         try:
             emoji: discord.PartialEmoji = await partial_emoji_convert(
                 ctx, i[0])
             role: discord.Role = await role_converter.convert(ctx, i[1])
         except commands.PartialEmojiConversionFailure:
             return await ctx.send_error(f"Emoji {i[0]} invalid")
         except commands.RoleNotFound:
             return await ctx.send_error(f"Role {i[1]} invalid")
         if role >= ctx.author.top_role and ctx.guild.owner_id != ctx.author.id:
             raise aoi.RoleHierarchyError(
                 f"Role {role} must be lower than your highest")
         if role >= ctx.me.top_role:
             raise aoi.RoleHierarchyError(
                 f"Role {role} must be lower than mine")
         try:
             await message.add_reaction(emoji)
         except discord.Forbidden:
             return await ctx.send_error("I can't react to that message!")
         except discord.HTTPException:
             return await ctx.send_error(f"Emoji {i[0]} invalid")
         if message.id not in self._roles:
             self._roles[message.id] = {}
         if emoji.name in self._roles[message.id] or str(
                 emoji.id) in self._roles[message.id]:
             return await ctx.send_error("That emoji is already being used")
         self._roles[message.id][str(emoji.id) if emoji.id else emoji.
                                 name] = _ReactionRoleData(role)
         await self._db.conn.execute(
             "insert into rero values (?,?,?,?,?,0,0)",
             (ctx.guild.id, message.channel.id, message.id,
              str(emoji.id) if emoji.id else emoji.name, role.id))
         await self._db.conn.commit()
         await ctx.send_ok("Added!")
Exemplo n.º 8
0
 async def satellite(self, ctx: aoi.AoiContext,
                     location: gmaps.LocationCoordinates):
     res = await self.wx.lookup_grid(location.lat, location.long)
     radar = res.radar_station[-3:]
     if radar in self.sat_cache:
         diff = (datetime.now() - self.sat_cache[radar][0]).seconds
         if diff < 30 * 60:
             img = self.sat_cache[radar][1]
             buf = io.BytesIO()
             img.save(buf, format="png")
             return await ctx.embed(
                 image=buf,
                 footer=f"Cached from {diff // 60}m{diff % 60:2} ago")
         del self.sat_cache[radar]
     urls = [
         f"https://radar.weather.gov/ridge/Overlays/Topo/Short/{radar}_Topo_Short.jpg",
         f"https://radar.weather.gov/ridge/RadarImg/N0R/{radar}_N0R_0.gif",
         f"https://radar.weather.gov/ridge/Overlays/County/Short/{radar}_County_Short.gif",
         f"https://radar.weather.gov/ridge/Overlays/Rivers/Short/{radar}_Rivers_Short.gif",
         f"https://radar.weather.gov/ridge/Overlays/Highways/Short/{radar}_Highways_Short.gif",
         f"https://radar.weather.gov/ridge/Overlays/Cities/Short/{radar}_City_Short.gif",
         f"https://radar.weather.gov/ridge/Warnings/Short/{radar}_Warnings_0.gif",
         f"https://radar.weather.gov/ridge/Legend/N0R/{radar}_N0R_Legend_0.gif"
     ]
     imgs = []
     async with ctx.typing():
         async with aiohttp.ClientSession() as sess:
             for url in urls:
                 async with sess.get(url) as resp:
                     buf = io.BytesIO()
                     buf.write(await resp.content.read())
                     buf.seek(0)
                     imgs.append(Image.open(buf, "png").convert("RGBA"))
     composite = reduce(lambda i1, i2: Image.alpha_composite(i1, i2), imgs)
     self.sat_cache[radar] = (datetime.now(), composite)
     buf = io.BytesIO()
     composite.save(fp=buf, format="png")
     await ctx.embed(image=buf)
Exemplo n.º 9
0
    async def roles(self, ctx: aoi.AoiContext):
        def _(s):
            return s if len(s) < 11 else f"{s[:10]}..."

        if ("audit" in ctx.flags or "full" in ctx.flags) and not ctx.author.guild_permissions.manage_roles:
            ctx.flags = {}

        if "audit" in ctx.flags and "full" in ctx.flags:
            return await ctx.send_error("Both `--audit` and `--full` cannot be passed")
        if "full" in ctx.flags:
            r: discord.Role
            await ctx.paginate([f"{r.position:>3} "
                                f"{'M' if r.mentionable else '-'}"
                                f"{'C' if r.color.to_rgb() != (0, 0, 0) else '-'}"
                                f"{'H' if r.hoist else '-'}|"
                                f"{'@' if r.permissions.mention_everyone or r.permissions.administrator else '-'}"
                                f"{'E' if r.permissions.manage_emojis or r.permissions.administrator else '-'}"
                                f"{'c' if r.permissions.manage_channels or r.permissions.administrator else '-'}"
                                f"{'G' if r.permissions.manage_guild or r.permissions.administrator else '-'}"
                                f"{'N' if r.permissions.manage_nicknames or r.permissions.administrator else '-'}"
                                f"{'W' if r.permissions.manage_webhooks or r.permissions.administrator else '-'}"
                                f"{'I' if r.permissions.view_guild_insights or r.permissions.administrator else '-'}"
                                f"{'R' if r.permissions.manage_roles or r.permissions.administrator else '-'}|"

                                f"{'n' if r.permissions.change_nickname or r.permissions.administrator else '-'}"
                                f"{'i' if r.permissions.create_instant_invite or r.permissions.administrator else '-'}"
                                f"{'r' if r.permissions.add_reactions or r.permissions.administrator else '-'}"

                                f"{'#' if r.permissions.external_emojis or r.permissions.administrator else '-'}?"
                                f"{'F' if r.permissions.attach_files or r.permissions.administrator else '-'}"
                                f"{'L' if r.permissions.embed_links or r.permissions.administrator else '-'}"
                                f"{'h' if r.permissions.read_message_history or r.permissions.administrator else '-'}"
                                f"{'r' if r.permissions.read_messages or r.permissions.administrator else '-'}"
                                f"{'$' if r.permissions.send_messages or r.permissions.administrator else '-'}"
                                f"{'T' if r.permissions.send_tts_messages or r.permissions.administrator else '-'}|"

                                f"{'L' if r.permissions.view_audit_log or r.permissions.administrator else '-'}"
                                f"{'K' if r.permissions.kick_members or r.permissions.administrator else '-'}"
                                f"{'!' if r.permissions.manage_messages or r.permissions.administrator else '-'}"
                                f"{'B' if r.permissions.ban_members or r.permissions.administrator else '-'}|"

                                f"{'P' if r.permissions.priority_speaker or r.permissions.administrator else '-'}"
                                f"{'s' if r.permissions.stream or r.permissions.administrator else '-'}"
                                f"{'S' if r.permissions.speak or r.permissions.administrator else '-'}"
                                f"{'=' if r.permissions.connect or r.permissions.administrator else '-'}"
                                f"{'V' if r.permissions.use_voice_activation or r.permissions.administrator else '-'}"
                                f"{'D' if r.permissions.deafen_members or r.permissions.administrator else '-'}"
                                f"{'X' if r.permissions.mute_members or r.permissions.administrator else '-'}"
                                f"{'<' if r.permissions.move_members or r.permissions.administrator else '-'}|"

                                f"{'A' if r.permissions.administrator else '-'} "
                                f"{discord.utils.escape_markdown(r.name)}"
                                for r in ctx.guild.roles[::-1]], 20, "Role list",
                               # THERE ARE MANY NBSP AND ZWSP HERE I PROMISE
                               fmt="```Pos                        Name\n%s```\n"
                                   "M:?Mentionable??????"
                                   "H:?Hoisted??????"
                                   "C:?Colored??????"
                                   "@:?Can?@?everyone??????"
                                   "E:?Manage?emojis??????"
                                   "c:?Manage?channels??????"
                                   "G:?Manage?server??????"
                                   "N:?Manage?nicknames??????"
                                   "W:?Manage?webhooks??????"
                                   "I:?View?insights??????"
                                   "R:?Manage?roles??????"
                                   "n:?Change?nickname??????"
                                   "i:?Create?invite??????"
                                   "r:?Add?reactions??????"
                                   "#:?External?emoji??????"
                                   "F:?Attach?files??????"
                                   "L:?Embed?links??????"
                                   "h:?Message?history??????"
                                   "r:?Read?messages??????"
                                   "$:?Send?messages??????"
                                   "T:?TTS?messages??????"
                                   "L:?Audit?logs??????"
                                   "K:?Can?kick??????"
                                   "!:?Manage?Messages??????"
                                   "B:?Can?ban??????"
                                   "P:?Priority speaker??????"
                                   "s:?Stream??????"
                                   "S:?Speak??????"
                                   "=:?Connect??????"
                                   "V:?Voice?activation??????"
                                   "D:?Deafen?members??????"
                                   "X:?Mute?members??????"
                                   "<:?Move?members??????"
                                   "A:?Administrator")
        elif "audit" in ctx.flags:
            r: discord.Role
            await ctx.paginate([f"{r.position:>3} "
                                f"{'M' if r.mentionable else '-'}"
                                f"{'C' if r.color.to_rgb() != (0, 0, 0) else '-'}"
                                f"{'H' if r.hoist else '-'}"
                                f"{'@' if r.permissions.mention_everyone else '-'}"
                                f"{'K' if r.permissions.kick_members or r.permissions.administrator else '-'}"
                                f"{'B' if r.permissions.ban_members or r.permissions.administrator else '-'}"
                                f"{'A' if r.permissions.administrator else '-'}"
                                f"{'!' if r.permissions.manage_messages or r.permissions.administrator else '-'} "
                                f"{_(discord.utils.escape_markdown(r.name))}"
                                for r in ctx.guild.roles[::-1]], 20, "Role list",
                               fmt="```Pos     Name\n%s```\n"
                                   "M-Mentionable  ?  "
                                   "H-Hoisted  ?  "
                                   "C-Colored  ?  "
                                   "@-Can @everyone  ?  "
                                   "K-Can kick  ?  "
                                   "B-Can ban  ?  "
                                   "A-Administrator  ?  "
                                   "!-Manage Messages")
        else:
            await ctx.paginate([f"{r.position:>3} "
                                f"{'M' if r.mentionable else '-'}"
                                f"{'C' if r.color.to_rgb() != (0, 0, 0) else '-'}"
                                f"{'H' if r.hoist else '-'} "
                                f"{discord.utils.escape_markdown(r.name)}"
                                for r in ctx.guild.roles[::-1]], 20, "Role list",
                               fmt="```Pos     Name\n%s```\n"
                                   "M-Mentionable  ?  "
                                   "H-Hoisted  ?  "
                                   "C-Colored")