70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
from xml.etree import ElementTree as etree
|
|
from urllib.request import Request, urlopen
|
|
from urllib.error import HTTPError
|
|
from json import loads, dumps
|
|
from re import sub
|
|
|
|
class RSS:
|
|
def __init__(self, state):
|
|
self.name = "RSS"
|
|
self.bot = state["bot"]
|
|
self.alias = state["alias"]
|
|
self.use = state["use"]
|
|
self.source = state["source"]
|
|
self.channels = state["channels"]
|
|
self.memory = state.get("memory", {
|
|
"initialized": False,
|
|
"known": list()
|
|
})
|
|
|
|
def start(self):
|
|
if not self.memory["initialized"]:
|
|
self.memory["initialized"] = True
|
|
self.fetch(self.cache)
|
|
return self.run()
|
|
|
|
def run(self):
|
|
self.fetch(self.mirror)
|
|
return {
|
|
"bot": self.bot,
|
|
"alias": self.alias,
|
|
"use": self.use,
|
|
"source": self.source,
|
|
"channels": self.channels,
|
|
"memory": self.memory
|
|
}
|
|
|
|
def cache(self, item):
|
|
guid = item.findtext("guid", None)
|
|
if guid is not None:
|
|
self.memory["known"].append(guid)
|
|
|
|
def mirror(self, item):
|
|
guid = item.findtext("guid", None)
|
|
if guid is None:
|
|
return
|
|
|
|
if guid in self.memory["known"]:
|
|
return
|
|
|
|
self.memory["known"].append(guid)
|
|
|
|
use = sub(r"(<\/?[^>]+>)|\n", "", item.findtext(self.use, ""))
|
|
user = item.findtext("author", "").split("@")[0]
|
|
post = "{} (posted by {}) <{}>".format(use, user, guid)
|
|
response = "[{}] {}".format(self.alias, post)
|
|
for channel in self.channels:
|
|
self.bot.send_message(channel, response)
|
|
|
|
def fetch(self, callback):
|
|
req = Request(self.source)
|
|
try:
|
|
response = urlopen(req).read()
|
|
except HTTPError:
|
|
return
|
|
|
|
feed = etree.fromstring(response)
|
|
items = feed.findall("channel/item")
|
|
for item in items:
|
|
callback(item)
|