microblog/app/utils/privacy_replace.py

40 lines
974 B
Python

from urllib.parse import urlparse
from bs4 import BeautifulSoup # type: ignore
from loguru import logger
from app.config import PRIVACY_REPLACE
def replace_content(content: str) -> str:
if not PRIVACY_REPLACE:
return content
soup = BeautifulSoup(content, "html5lib")
links = list(soup.find_all("a", href=True))
if not links:
return content
for link in links:
link.attrs["href"] = replace_url(link.attrs["href"])
return soup.find("body").decode_contents()
def replace_url(u: str) -> str:
if not PRIVACY_REPLACE:
return u
try:
parsed_href = urlparse(u)
if not parsed_href.hostname:
raise ValueError("Missing hostname")
except Exception:
logger.warning(f"Failed to parse url={u}")
return u
if new_netloc := PRIVACY_REPLACE.get(parsed_href.hostname.removeprefix("www.")):
return parsed_href._replace(netloc=new_netloc).geturl()
return u