wttr.py

from .. import loader, utils  # pylint: disable=relative-beyond-top-level
import logging
import requests

logger = logging.getLogger(__name__)


def register(cb):
    cb(WWTrMod())


@loader.tds
class WWWTrMod(loader.Module):
    """Get weather using wttr.in"""

    strings = {"name": "wttr", "processing": "Getting weather..."}

    @loader.unrestricted
    async def wttrcmd(self, message):
        """.wttr <city> for weather
            can also work with multiple cities separated with spaces"""
        city = utils.get_args(message)
        if city:
            msg_text = []
            msg = await utils.answer(message, self.strings("processing", message))
            for i in city:
                r = requests.get("https://wttr.in/" + i + "?format=%l:+%c+%t,+%w+%m&m")
                msg_text.append(r.text)
            await msg[0].edit("\n".join(msg_text))
        else:
            msg = await utils.answer(message, self.strings("processing", message))
            r = requests.get("https://wttr.in/?format=%l:+%c+%t,+%w+%m&m")
            await msg[0].edit(r.text)