mathdatech/tootbot
1
0
Fork 0
mirror of https://github.com/cquest/tootbot.git synced 2024-11-23 20:11:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #20 from hmlendea/master

Browse Source 
Support for Reddit
This commit is contained in:
Christian Quest 2022-12-08 11:00:21 +01:00 committed by GitHub
commit 89ae4479dd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tootbot.py
View File

@ -100,6 +100,7 @@ if source[:4] == 'http':
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...
if 'summary' in t: if 'summary' in t:
for p in re.finditer(r"https://pbs.twimg.com/[^ \xa0\"]*", t.summary): for p in re.finditer(r"https://pbs.twimg.com/[^ \xa0\"]*", t.summary):
media = requests.get(p.group(0)) media = requests.get(p.group(0))
@ -107,9 +108,19 @@ if source[:4] == 'http':
media.content, mime_type=media.headers.get('content-type')) media.content, mime_type=media.headers.get('content-type'))
toot_media.append(media_posted['id']) toot_media.append(media_posted['id'])
for p in re.finditer(r"https://i.redd.it/[a-zA-Z0-9]*.(gif/jpg/mp4/png|webp)", t.summary):
mediaUrl = p.group(0)
try:
media = requests.get(mediaUrl)
media_posted = mastodon_api.media_post(
media.content, mime_type=media.headers.get('content-type'))
toot_media.append(media_posted['id'])
except:
print('Could not upload media to Mastodon! ' + mediaUrl)
if 'links' in t: if 'links' in t:
for l in t.links: for l in t.links:
if l.type in ('image/jpg', 'image/png'): if l.type in ('image/gif', 'image/jpg', 'image/png', 'image/webp'):
media = requests.get(l.url) media = requests.get(l.url)
media_posted = mastodon_api.media_post( media_posted = mastodon_api.media_post(
media.content, mime_type=media.headers.get('content-type')) media.content, mime_type=media.headers.get('content-type'))
@ -119,9 +130,12 @@ if source[:4] == 'http':
m = re.search(r"http[^ \xa0]*", c) m = re.search(r"http[^ \xa0]*", c)
if m is not None: if m is not None:
l = m.group(0) l = m.group(0)
r = requests.get(l, allow_redirects=False) try:
if r.status_code in {301, 302}: r = requests.get(l, allow_redirects=False)
c = c.replace(l, r.headers.get('Location')) if r.status_code in {301, 302}:
c = c.replace(l, r.headers.get('Location'))
except:
print('Cannot resolve link redirect: ' + l)
# remove ellipsis # remove ellipsis
c = c.replace('\xa0…', ' ') c = c.replace('\xa0…', ' ')
@ -130,6 +144,10 @@ if source[:4] == 'http':
c = c + '\nSource: ' + t.authors[0].name c = c + '\nSource: ' + t.authors[0].name
c = c + '\n\n' + t.link c = c + '\n\n' + t.link
# replace links to reddit by libreddit ones
c = c.replace('old.reddit.com', 'libreddit.net')
c = c.replace('reddit.com', 'libreddit.net')
if tags: if tags:
c = c + '\n' + tags c = c + '\n' + tags