Browse Source

[WIP] Rudimentary Telegram integration

FREEZE-blogpost-2021-01-20
Felix 6 months ago
parent
commit
940df38858
  1. 4
      isso/__init__.py
  2. 75
      isso/ext/notifications.py
  3. 7
      share/isso-dev.conf
  4. 7
      share/isso.conf

4
isso/__init__.py

@ -72,7 +72,7 @@ from isso.wsgi import origin, urlsplit
from isso.utils import http, JSONRequest, html, hash
from isso.views import comments
from isso.ext.notifications import Stdout, SMTP
from isso.ext.notifications import Stdout, SMTP, Telegram
logging.getLogger('werkzeug').setLevel(logging.WARN)
logging.basicConfig(
@ -110,6 +110,8 @@ class Isso(object):
subscribers.append(Stdout(None))
elif backend in ("smtp", "SMTP"):
smtp_backend = True
elif backend == "telegram":
subscribers.append(Telegram(conf))
else:
logger.warn("unknown notification backend '%s'", backend)
if smtp_backend or conf.getboolean("general", "reply-notifications"):

75
isso/ext/notifications.py

@ -5,6 +5,7 @@ from __future__ import unicode_literals
import io
import time
import json
import re
import socket
import smtplib
@ -23,6 +24,11 @@ try:
except ImportError:
uwsgi = None
try:
import requests
except ImportError:
requests = None
from isso import local
from _thread import start_new_thread
@ -196,6 +202,75 @@ class SMTP(object):
break
class Telegram(object):
def __init__(self, conf):
self.conf = conf.section("telegram")
self.chat_id = self.conf.get('chat_id')
token = self.conf.get('token')
self.telegram_api = 'https://api.telegram.org/bot{}/sendMessage'.format(token)
def __iter__(self):
yield "comments.new:new-thread", self._new_thread
yield "comments.new:finish", self._new_comment
yield "comments.edit", self._edit_comment
yield "comments.delete", self._delete_comment
def _sanitize(self, message):
# Need to strip elements Telegram hates
# See https://core.telegram.org/bots/api#html-style
allowed = ["b", "i", "u", "s", "a", "code", "pre"]
sanitized_msg = message
# Preserve some emphasis that was mis-rendered by misaka
sanitized_msg = sanitized_msg.replace(
"<strong>", "<b>").replace("</strong>", "</b>")
for m in re.findall(r"(</?\w+>)", message):
# Only tag content:
stripped = m.lstrip("<").rstrip(">").lstrip("/")
# Links should be ok:
if stripped.startswith("a href"):
continue
if stripped not in allowed:
sanitized_msg = sanitized_msg.replace(m, "")
return sanitized_msg
def _post_tg(self, message):
if not requests:
logger.warn("Need requests package for posting to Telegram API!")
return
r = requests.post(self.telegram_api,
data={
'chat_id': self.chat_id,
'text': message,
'parse_mode': 'HTML'
})
# Response should look like:
# b'{"ok":true,"result":...}
logger.info("Got response from TG API: {}".format(r.content))
r_dict = json.loads(r.content)
if r_dict.get("ok") is not True:
logger.info("Posting to Telegram successful")
else:
logger.warn("Posting to Telegram failed! {}".format(r_dict))
def _new_thread(self, thread):
self._post_tg("New thread! {id}: {title}".format(**thread))
def _new_comment(self, thread, comment):
comment["text"] = self._sanitize(comment["text"])
msg = "New comment!\n<b>Author</b>: {author}\nWebsite: {website}\n<b>Content</b>:\n{text}"
self._post_tg(msg.format(**comment))
def _edit_comment(self, comment):
comment["text"] = self._sanitize(comment["text"])
msg = "Comment edited:\n<b>Author:</b> {author}\n<b>New content:</b>\n {text}"
self._post_tg(msg.format(**comment))
def _delete_comment(self, id):
self._post_tg("Comment deleted: {}".format(id))
class Stdout(object):
def __init__(self, conf):

7
share/isso-dev.conf

@ -8,10 +8,15 @@
dbpath = /var/isso/comments.db
host = http://isso-dev.local/
max-age = 15m
notify = stdout
notify = stdout, telegram
reply-notifications = false
log-file = /var/log/isso.log
[telegram]
token =
chat_id =
[admin]
enabled = true
password = strong_default_password_for_isso_admin

7
share/isso.conf

@ -63,6 +63,13 @@ gravatar-url = https://www.gravatar.com/avatar/{}?d=identicon
# needing to previously know the posts URIs)
latest-enabled = false
# Telegram notifications integration
[telegram]
token =
chat_id =
[admin]
enabled = false

Loading…
Cancel
Save