simplebot_cubaweather.py

"""
Search weather info using cuba-weather from redcuba.cu
"""

import simplebot
from cuba_weather import RCApiClient

__version__ = '1.0.0'
template = '\n\n'.join((
    '🌐 {}',
    '📅 {}',
    '⛅ Estado: {}',
    '🌡 Temperatura: {}°C',
    '💧 Humedad: {}%',
    'Presión atmosférica: {} hpa',
    '🌬 Vientos:\n{}',
))


@simplebot.command
def cuwtr(payload, replies):
    """Obtén el estado del tiempo en Cuba. Ejemplo: /cuwtr Santiago"""
    if payload:
        weather = RCApiClient().get(payload, suggestion=True)
        replies.add(
            text=template.format(
                weather.city_name,
                weather.timestamp,
                weather.general,
                weather.temperature,
                weather.humidity,
                weather.pressure,
                weather.wind,
            )
        )
    else:
        replies.add(
            text='Envíame una localidad de Cuba, ej.: /cuwtr Santiago')


class TestCuwtr:
    def test_cuwtr(self, mocker):
        msg = mocker.get_one_reply('/cuwtr')
        assert 'Envíame una localidad' in msg.text

        msg = mocker.get_one_reply('/cuwtr Santiago')
        assert 'Temperatura:' in msg.text