blankie
/
redditbot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Compare commits

base: blankie:3371eba487fbd6c04404b8110305e413a6ce5c7c
Branches Tags
blankie:master
..
compare: blankie:ad17d5a4a5e6e5d321dd41a3e2964898aa464f06
Branches Tags
blankie:master

2 Commits
3371eba487 ... ad17d5a4a5

Author SHA1 Message Date
blank X ad17d5a4a5
Check if extension is .htm or .html, instead of if it starts with .htm 2021-04-29 16:33:29 +07:00
blank X 1330b8c3cd
Add og:video support 2021-04-29 16:32:53 +07:00
1 changed files with 8 additions and 8 deletions
Show Stats Expand all files Collapse all files

16
redditbot.py
View File

@ -365,7 +365,7 @@ async def main():
url = urlunparse(urlparse(url, 'https')) url = urlunparse(urlparse(url, 'https'))
await _download_file(filename, url) await _download_file(filename, url)
ext = await _get_file_ext(filename) ext = await _get_file_ext(filename)
if ext.startswith('.htm'): if ext in ('.htm', '.html'):
with open(filename) as file: with open(filename) as file:
soup = BeautifulSoup(file.read()) soup = BeautifulSoup(file.read())
ptitle = soup.find(lambda tag: tag.name == 'meta' and tag.attrs.get('property') == 'og:title' and tag.attrs.get('content')) or soup.find('title') ptitle = soup.find(lambda tag: tag.name == 'meta' and tag.attrs.get('property') == 'og:title' and tag.attrs.get('content')) or soup.find('title')
@ -374,21 +374,21 @@ async def main():
pdesc = soup.find(lambda tag: tag.name == 'meta' and tag.attrs.get('property') == 'og:description' and tag.attrs.get('content')) or soup.find(lambda tag: tag.name == 'meta' and tag.attrs.get('name') == 'description' and tag.attrs.get('content')) pdesc = soup.find(lambda tag: tag.name == 'meta' and tag.attrs.get('property') == 'og:description' and tag.attrs.get('content')) or soup.find(lambda tag: tag.name == 'meta' and tag.attrs.get('name') == 'description' and tag.attrs.get('content'))
if pdesc: if pdesc:
pdesc = pdesc.attrs.get('content', pdesc.text).strip() pdesc = pdesc.attrs.get('content', pdesc.text).strip()
pimg = soup.find(lambda tag: tag.name == 'meta' and tag.attrs.get('property') == 'og:image' and tag.attrs.get('content')) pmedia = soup.find(lambda tag: tag.name == 'meta' and tag.attrs.get('property') == 'og:video' and tag.attrs.get('content')) or soup.find(lambda tag: tag.name == 'meta' and tag.attrs.get('property') == 'og:image' and tag.attrs.get('content'))
if pimg: if pmedia:
pimg = pimg.attrs.get('content', '').strip() pmedia = pmedia.attrs.get('content', '').strip()
tat = f'{text}\n\nURL: ' tat = f'{text}\n\nURL: '
if ptitle: if ptitle:
tat += f'<a href="{url}">{html.escape(ptitle)}</a>' tat += f'<a href="{url}">{html.escape(ptitle)}</a>'
else: else:
tat += url tat += url
files = [] files = []
if pimg: if pmedia:
pimg = urlunparse(urlparse(pimg, 'https')) pmedia = urlunparse(urlparse(pmedia, 'https'))
await _download_file(filename, pimg) await _download_file(filename, pmedia)
files.append(filename) files.append(filename)
if pdesc: if pdesc:
caplength = 1023 if pimg else 4095 caplength = 1023 if pmedia else 4095
caplength -= len(client.parse_mode.parse(tat)[0]) caplength -= len(client.parse_mode.parse(tat)[0])
captext = pdesc[:caplength] captext = pdesc[:caplength]
if len(captext) >= caplength: if len(captext) >= caplength: