mirror of https://github.com/zulip/zulip.git
bots: Add weather reporting bot.
This commit is contained in:
parent
7126f6f30c
commit
695c71d2cc
|
@ -0,0 +1,2 @@
|
|||
[weather-config]
|
||||
key=XXXXXXXX
|
Binary file not shown.
After Width: | Height: | Size: 67 KiB |
Binary file not shown.
After Width: | Height: | Size: 24 KiB |
|
@ -0,0 +1,16 @@
|
|||
# WeatherBot
|
||||
|
||||
* This is a bot that sends weather information to a selected stream on
|
||||
request.
|
||||
|
||||
* Weather information is brought to the website using an
|
||||
OpenWeatherMap API. The bot posts the weather information to the
|
||||
stream from which the user inputs the message. If the user inputs a
|
||||
city that does not exist, the bot displays a "Sorry, city not found"
|
||||
message.
|
||||
|
||||
* Before using this bot, you have to generate an OpenWeatherMap API
|
||||
key and replace the dummy value in .weather_config.
|
||||
|
||||
![Example Usage](assets/screen1.png)
|
||||
![Wrong City](assets/screen2.png)
|
|
@ -0,0 +1,75 @@
|
|||
# See readme.md for instructions on running this code.
|
||||
from __future__ import print_function
|
||||
import requests
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
from six.moves.configparser import SafeConfigParser
|
||||
|
||||
|
||||
class WeatherHandler(object):
|
||||
def __init__(self):
|
||||
self.directory = os.path.dirname(os.path.realpath(__file__)) + '/'
|
||||
self.config_name = '.weather_config'
|
||||
self.response_pattern = 'Weather in {}, {}:\n{} F / {} C\n{}'
|
||||
if not os.path.exists(self.directory + self.config_name):
|
||||
print('Weather bot config file not found, please set it up in {} file in this bot main directory'
|
||||
'\n\nUsing format:\n\n[weather-config]\nkey=<OpenWeatherMap API key here>\n\n'.format(self.config_name))
|
||||
sys.exit(1)
|
||||
super(WeatherHandler, self).__init__()
|
||||
|
||||
def usage(self):
|
||||
return '''
|
||||
This plugin will give info about weather in a specified city
|
||||
'''
|
||||
|
||||
def handle_message(self, message, client, state_handler):
|
||||
help_content = '''
|
||||
This bot returns weather info for specified city.
|
||||
You specify city in the following format:
|
||||
city, state/country
|
||||
state and country parameter is optional(useful when there are many cities with the same name)
|
||||
For example:
|
||||
@**Weather Bot** Portland
|
||||
@**Weather Bot** Portland, Me
|
||||
'''.strip()
|
||||
|
||||
if (message['content'] == 'help') or (message['content'] == ''):
|
||||
response = help_content
|
||||
else:
|
||||
url = 'http://api.openweathermap.org/data/2.5/weather?q=' + message['content'] + '&APPID='
|
||||
r = requests.get(url + get_weather_api_key_from_config(self.directory, self.config_name))
|
||||
if "city not found" in r.text:
|
||||
response = "Sorry, city not found"
|
||||
else:
|
||||
response = format_response(r.text, message['content'], self.response_pattern)
|
||||
|
||||
client.send_reply(message, response)
|
||||
|
||||
|
||||
def format_response(text, city, response_pattern):
|
||||
j = json.loads(text)
|
||||
city = j['name']
|
||||
country = j['sys']['country']
|
||||
fahrenheit = to_fahrenheit(j['main']['temp'])
|
||||
celsius = to_celsius(j['main']['temp'])
|
||||
description = j['weather'][0]['description'].title()
|
||||
|
||||
return response_pattern.format(city, country, fahrenheit, celsius, description)
|
||||
|
||||
|
||||
def to_celsius(temp_kelvin):
|
||||
return int(temp_kelvin) - 273.15
|
||||
|
||||
|
||||
def to_fahrenheit(temp_kelvin):
|
||||
return int(temp_kelvin) * 9 / 5 - 459.67
|
||||
|
||||
|
||||
def get_weather_api_key_from_config(directory, config_name):
|
||||
config = SafeConfigParser()
|
||||
with open(directory + config_name, 'r') as config_file:
|
||||
config.readfp(config_file)
|
||||
return config.get("weather-config", "key")
|
||||
|
||||
handler_class = WeatherHandler
|
Loading…
Reference in New Issue