mathdatech/tootbot
1
0
Fork 0
mirror of https://github.com/cquest/tootbot.git synced 2025-01-18 12:26:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

rss/atom feed support

Browse Source
This commit is contained in:
cquest 2019-04-03 17:47:28 +02:00
parent 91f811ecbb
commit 66f4a1a887
1 changed files with 24 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
tootbot.py
View File

@ -28,13 +28,23 @@ if len(sys.argv) > 5:
else: else:
days = 1 days = 1
twitter = sys.argv[1] if len(sys.argv) > 6:
tags = sys.argv[6]
else:
tags = None
source = sys.argv[1]
mastodon = sys.argv[2] mastodon = sys.argv[2]
passwd = sys.argv[3] passwd = sys.argv[3]
mastodon_api = None mastodon_api = None
d = feedparser.parse('http://twitrss.me/twitter_user_to_rss/?user='+twitter) if source[:4] == 'http':
d = feedparser.parse(source)
twitter = None
else:
d = feedparser.parse('http://twitrss.me/twitter_user_to_rss/?user='+source)
twitter = source
for t in reversed(d.entries): for t in reversed(d.entries):
# check if this tweet has been processed # check if this tweet has been processed
@ -73,9 +83,8 @@ for t in reversed(d.entries):
print("ERROR: First Login Failed!") print("ERROR: First Login Failed!")
sys.exit(1) sys.exit(1)
#h = BeautifulSoup(t.summary_detail.value, "html.parser")
c = t.title c = t.title
if t.author.lower() != ('(@%s)' % twitter).lower(): if twitter and t.author.lower() != ('(@%s)' % twitter).lower():
c = ("RT https://twitter.com/%s\n" % t.author[2:-1]) + c c = ("RT https://twitter.com/%s\n" % t.author[2:-1]) + c
toot_media = [] toot_media = []
# get the pictures... # get the pictures...
@ -84,9 +93,9 @@ for t in reversed(d.entries):
media_posted = mastodon_api.media_post(media.content, mime_type=media.headers.get('content-type')) media_posted = mastodon_api.media_post(media.content, mime_type=media.headers.get('content-type'))
toot_media.append(media_posted['id']) toot_media.append(media_posted['id'])
# replace t.co link by original URL # replace short links by original URL
m = re.search(r"http[^ \xa0]*", c) m = re.search(r"http[^ \xa0]*", c)
if m != None: if m is not None:
l = m.group(0) l = m.group(0)
r = requests.get(l, allow_redirects=False) r = requests.get(l, allow_redirects=False)
if r.status_code in {301, 302}: if r.status_code in {301, 302}:
@ -94,12 +103,18 @@ for t in reversed(d.entries):
# remove pic.twitter.com links # remove pic.twitter.com links
m = re.search(r"pic.twitter.com[^ \xa0]*", c) m = re.search(r"pic.twitter.com[^ \xa0]*", c)
if m != None: if m is not None:
l = m.group(0) l = m.group(0)
c = c.replace(l, ' ') c = c.replace(l, ' ')
# remove ellipsis # remove ellipsis
c = c.replace('\xa0…',' ') c = c.replace('\xa0…', ' ')
if twitter is None:
c = c + '\nSource: '+ t.authors[0].name +'\n\n' + t.link
if tags:
c = c + '\n' + tags
if toot_media is not None: if toot_media is not None:
toot = mastodon_api.status_post(c, in_reply_to_id=None, toot = mastodon_api.status_post(c, in_reply_to_id=None,
@ -109,5 +124,5 @@ for t in reversed(d.entries):
spoiler_text=None) spoiler_text=None)
if "id" in toot: if "id" in toot:
db.execute("INSERT INTO tweets VALUES ( ? , ? , ? , ? , ? )", db.execute("INSERT INTO tweets VALUES ( ? , ? , ? , ? , ? )",
(t.id, toot["id"], twitter, mastodon, instance)) (t.id, toot["id"], source, mastodon, instance))
sql.commit() sql.commit()