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:
parent
1a74091e04
commit
967ab33528
15
README.md
15
README.md
|
@ -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
15
bot.py
|
@ -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:
|
||||||
|
|
|
@ -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)'
|
Loading…
Reference in New Issue