mirror of https://github.com/miaowware/qrm2.git
129 lines
5.1 KiB
Python
129 lines
5.1 KiB
Python
"""
|
|
Weather extension for qrm
|
|
---
|
|
Copyright (C) 2019 Abigail Gold, 0x5c
|
|
|
|
This file is part of discord-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):
|
|
'''Posts an image of HF Band Conditions.'''
|
|
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:
|
|
embed.description = 'Could not download file...'
|
|
embed.colour = cmn.colours.bad
|
|
else:
|
|
data = io.BytesIO(await resp.read())
|
|
embed.set_image(url=f'attachment://condx.png')
|
|
await ctx.send(embed=embed, file=discord.File(data, 'condx.png'))
|
|
|
|
@commands.group(name="weather", aliases=['wttr'], category=cmn.cat.weather)
|
|
async def _weather_conditions(self, ctx: commands.Context):
|
|
'''Posts an image of 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`
|
|
'''
|
|
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):
|
|
'''Posts an image of Local Weather Conditions for the next three days from [wttr.in](http://wttr.in/).
|
|
See help for weather command for possible location types. Add a `-c` or `-f` to use Celcius or Fahrenheit.'''
|
|
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:
|
|
embed.description = 'Could not download file...'
|
|
embed.colour = cmn.colours.bad
|
|
else:
|
|
data = io.BytesIO(await resp.read())
|
|
embed.set_image(url=f'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):
|
|
'''Posts an image of current Local Weather Conditions from [wttr.in](http://wttr.in/).
|
|
See help for weather command for possible location types. Add a `-c` or `-f` to use Celcius or Fahrenheit.'''
|
|
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:
|
|
embed.description = 'Could not download file...'
|
|
embed.colour = cmn.colours.bad
|
|
else:
|
|
data = io.BytesIO(await resp.read())
|
|
embed.set_image(url=f'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))
|