qrm2/exts/weather.py

"""
Weather extension for qrm
---
Copyright (C) 2019-2020 Abigail Gold, 0x5c

This file is part of qrm2 and is released under the terms of
the GNU General Public License, version 2.
"""


import io
import re

import aiohttp

import discord
import discord.ext.commands as commands

import common as cmn


class WeatherCog(commands.Cog):
    wttr_units_regex = re.compile(r"\B-([cCfF])\b")

    def __init__(self, bot: commands.Bot):
        self.bot = bot
        self.session = aiohttp.ClientSession(connector=bot.qrm.connector)

    @commands.command(name="bandconditions", aliases=["cond", "condx", "conditions"], category=cmn.cat.weather)
    async def _band_conditions(self, ctx: commands.Context):
        """Gets a solar conditions report."""
        async with ctx.typing():
            embed = cmn.embed_factory(ctx)
            embed.title = "Current Solar Conditions"
            embed.colour = cmn.colours.good
            async with self.session.get("http://www.hamqsl.com/solarsun.php") as resp:
                if resp.status != 200:
                    raise cmn.BotHTTPError(resp)
                data = io.BytesIO(await resp.read())
            embed.set_image(url="attachment://condx.png")
            await ctx.send(embed=embed, file=discord.File(data, "condx.png"))

    @commands.group(name="weather", aliases=["wttr"], case_insensitive=True, category=cmn.cat.weather)
    async def _weather_conditions(self, ctx: commands.Context):
        """Gets local weather conditions from [wttr.in](http://wttr.in/).

        *Supported location types:*
        city name: `paris`
        any location: `~Eiffel Tower`
        Unicode name of any location in any language: `Москва`
        airport code (3 letters): `muc`
        domain name `@stackoverflow.com`
        area codes: `12345`
        GPS coordinates: `-78.46,106.79`

        Add a `-c` or `-f` to use Celcius or Fahrenheit: `-c YSC`"""
        if ctx.invoked_subcommand is None:
            await ctx.send_help(ctx.command)

    @_weather_conditions.command(name="forecast", aliases=["fc", "future"], category=cmn.cat.weather)
    async def _weather_conditions_forecast(self, ctx: commands.Context, *, location: str):
        """Gets local weather forecast for the next three days from [wttr.in](http://wttr.in/).
        See help of the `weather` command for possible location types and options."""
        async with ctx.typing():
            try:
                units_arg = re.search(self.wttr_units_regex, location).group(1)
            except AttributeError:
                units_arg = ""
            if units_arg.lower() == "f":
                units = "u"
            elif units_arg.lower() == "c":
                units = "m"
            else:
                units = ""

            loc = self.wttr_units_regex.sub("", location).strip()

            embed = cmn.embed_factory(ctx)
            embed.title = f"Weather Forecast for {loc}"
            embed.description = "Data from [wttr.in](http://wttr.in/)."
            embed.colour = cmn.colours.good

            loc = loc.replace(" ", "+")
            async with self.session.get(f"http://wttr.in/{loc}_{units}pnFQ.png") as resp:
                if resp.status != 200:
                    raise cmn.BotHTTPError(resp)
                data = io.BytesIO(await resp.read())
            embed.set_image(url="attachment://wttr_forecast.png")
            await ctx.send(embed=embed, file=discord.File(data, "wttr_forecast.png"))

    @_weather_conditions.command(name="now", aliases=["n"], category=cmn.cat.weather)
    async def _weather_conditions_now(self, ctx: commands.Context, *, location: str):
        """Gets current local weather conditions from [wttr.in](http://wttr.in/).
        See help of the `weather` command for possible location types and options."""
        async with ctx.typing():
            try:
                units_arg = re.search(self.wttr_units_regex, location).group(1)
            except AttributeError:
                units_arg = ""
            if units_arg.lower() == "f":
                units = "u"
            elif units_arg.lower() == "c":
                units = "m"
            else:
                units = ""

            loc = self.wttr_units_regex.sub("", location).strip()

            embed = cmn.embed_factory(ctx)
            embed.title = f"Current Weather for {loc}"
            embed.description = "Data from [wttr.in](http://wttr.in/)."
            embed.colour = cmn.colours.good

            loc = loc.replace(" ", "+")
            async with self.session.get(f"http://wttr.in/{loc}_0{units}pnFQ.png") as resp:
                if resp.status != 200:
                    raise cmn.BotHTTPError(resp)
                data = io.BytesIO(await resp.read())
            embed.set_image(url="attachment://wttr_now.png")
            await ctx.send(embed=embed, file=discord.File(data, "wttr_now.png"))


def setup(bot: commands.Bot):
    bot.add_cog(WeatherCog(bot))