added extract_toot function to functions.py
This commit is contained in:
parent
85fec32c83
commit
de3449ae56
23
functions.py
23
functions.py
|
@ -60,3 +60,26 @@ def make_toot_markov(query = None):
|
||||||
"toot": toot,
|
"toot": toot,
|
||||||
"media": None
|
"media": None
|
||||||
}
|
}
|
||||||
|
|
||||||
|
def extract_toot(toot):
|
||||||
|
soup = BeautifulSoup(toot, "html.parser")
|
||||||
|
for lb in soup.select("br"): #replace <br> with linebreak
|
||||||
|
lb.insert_after("\n")
|
||||||
|
lb.decompose()
|
||||||
|
|
||||||
|
for p in soup.select("p"): #ditto for <p>
|
||||||
|
p.insert_after("\n")
|
||||||
|
p.unwrap()
|
||||||
|
|
||||||
|
for ht in soup.select("a.hashtag"): #make hashtags no longer links, just text
|
||||||
|
ht.unwrap()
|
||||||
|
|
||||||
|
for link in soup.select("a"): #ocnvert <a href='https://example.com>example.com</a> to just https://example.com
|
||||||
|
link.insert_after(link["href"])
|
||||||
|
link.decompose()
|
||||||
|
|
||||||
|
toot = soup.get_text()
|
||||||
|
text = re.sub("https://([^/]+)/(@[^ ]+)", r"\2@\1", text) #put mastodon-style mentions back in
|
||||||
|
text = re.sub("https://([^/]+)/users/([^ ]+)", r"@\2@\1", text) #put pleroma-style mentions back in
|
||||||
|
text = text.rstrip("\n") #remove trailing newline
|
||||||
|
return text
|
Reference in New Issue