added Astronomy Picture of the Day module

* upload astronomy picture of the day to the room
* cache latest response
* use own api key by setting environment variable
  APOD_API_KEY (default: DEMO_KEY)
This commit is contained in:
Frank Becker 2020-03-01 20:40:31 +01:00
parent 1a74091e04
commit 967ab33528
3 changed files with 154 additions and 0 deletions

View File

@ -218,6 +218,21 @@ Example:
* !cmd add df "df -h" * !cmd add df "df -h"
* !cmd add whoami "echo You are $MATRIX_USER in room $MATRIX_ROOM." * !cmd add whoami "echo You are $MATRIX_USER in room $MATRIX_ROOM."
### Astronomy Picture of the Day
Upload and send latest astronomy picture of the day to the room.
See https://apod.nasa.gov/apod/astropix.html
Command:
* !apod - Sends latest Astronomy Picture of the Day to the room
API Key:
The module uses a demo API Key which can be replaced by your own api key by setting the environment variable `APOD_API_KEY`.
You can create one at https://api.nasa.gov/#signUp
## Bot setup ## Bot setup
* Create a Matrix user * Create a Matrix user

15
bot.py
View File

@ -86,6 +86,21 @@ class Bot:
} }
await self.client.room_send(room.room_id, 'm.room.message', msg) await self.client.room_send(room.room_id, 'm.room.message', msg)
async def send_image(self, room, url, body):
"""
:param room: A MatrixRoom the image should be send to
:param url: A MXC-Uri https://matrix.org/docs/spec/client_server/r0.6.0#mxc-uri
:param body: A textual representation of the image
:return:
"""
msg = {
"url": url,
"body": body,
"msgtype": "m.image"
}
await self.client.room_send(room.room_id, 'm.room.message', msg)
def remove_callback(self, callback): def remove_callback(self, callback):
for cb_object in self.client.event_callbacks: for cb_object in self.client.event_callbacks:
if cb_object.func == callback: if cb_object.func == callback:

124
modules/apod.py Normal file
View File

@ -0,0 +1,124 @@
import json
import os
import requests
from dateutil import parser
from nio import AsyncClient, UploadError
from nio import UploadResponse
from modules.common.module import BotModule
class Apod:
def __init__(self, title, explanation, date, hdurl, media_type, url):
self.hdurl = hdurl
self.title = title
self.explanation = explanation
self.date = date
self.media_type = media_type
self.url = url
@staticmethod
def create_from_json(json):
return Apod(json.get("title"), json.get("explanation"), json.get("date"), json.get("hdurl"),
json.get("media_type"), json.get("url"))
def __str__(self):
return "title: {} || explanation: {} || date: {} || original-url: {}".format(self.title,
self.explanation,
self.date,
self.hdurl)
class MatrixModule(BotModule):
def __init__(self, name):
super().__init__(name)
self.api_key = os.getenv("APOD_API_KEY", "DEMO_KEY")
self.apod_api_url = f"https://api.nasa.gov/planetary/apod?api_key={self.api_key}&hd=true"
self.last_update = None
self.last_apod = None
self.last_matrix_uri = None
async def matrix_message(self, bot, room, event):
self.logger.debug(f"room: {room.name} sender: {event.sender} wants latest astronomy picture of the day")
await self.send_apod(bot, room)
async def send_apod(self, bot, room):
response = requests.get(self.apod_api_url)
if response.status_code == 200:
apod = Apod.create_from_json(response.json())
current_date = parser.parse(apod.date).date()
self.logger.debug(apod)
if (self.last_apod is None) or (current_date > self.last_update):
self.last_apod = apod
self.last_update = current_date
if apod.media_type == "image":
await self.upload_and_send_image(room, bot, apod)
else:
await self.send_unknown_mediatype(room, bot, apod)
else:
self.logger.debug("latest apod already requested. sending last response: %s", self.last_apod)
if self.last_apod.media_type == "image":
await bot.send_image(room, self.last_matrix_uri, self.last_apod.__str__())
else:
await self.send_unknown_mediatype(room, bot, self.last_apod)
else:
self.logger.error("unable to request apod api. response: %s", response.text)
await bot.send_text(room, "sorry. something went wrong accessing the api :(")
async def send_unknown_mediatype(self, room, bot, apod):
await bot.send_text(room, f"unsupported media_type: {apod.media_type}")
await bot.send_text(room, f"title: {apod.title}\n\nexplanation:\n{apod.explanation}\n\nurl: {apod.url}")
async def upload_and_send_image(self, room, bot, apod):
url = apod.hdurl if apod.hdurl is not None else apod.url
matrix_uri = await self.upload_image(bot, url)
if matrix_uri is not None:
self.last_matrix_uri = matrix_uri
bot.save_settings()
await bot.send_image(room, matrix_uri, apod.__str__())
else:
await bot.send_text(room, "sorry. something went wrong uploading the image to matrix server :(")
async def upload_image(self, bot, url):
self.client: AsyncClient
response: UploadResponse
url_response = requests.get(url)
if url_response.status_code == 200:
content_type = url_response.headers.get("content-type")
(response, alist) = await bot.client.upload(lambda a, b: url_response.content, content_type)
if isinstance(response, UploadResponse):
self.logger.debug("uploaded file to %s", response.content_uri)
return response.content_uri
else:
response: UploadError
self.logger.error("unable to upload file. msg: %s", response.message)
else:
self.logger.error("unable to request url: %s", url_response)
return None
def get_settings(self):
data = super().get_settings()
data["last_update"] = self.last_update.__str__()
data["last_matrix_uri"] = self.last_matrix_uri
data["last_apod"] = json.dumps(self.last_apod.__dict__)
return data
def set_settings(self, data):
super().set_settings(data)
if data.get("last_update"):
self.last_update = parser.parse(data["last_update"]).date()
if data.get("last_matrix_uri"):
self.last_matrix_uri = data["last_matrix_uri"]
if data.get("last_apod"):
self.last_apod = Apod.create_from_json(json.loads(data["last_apod"]))
def help(self):
return 'Sends latest Astronomy Picture of the Day to the room. (https://apod.nasa.gov/apod/astropix.html)'