michael/pyfedi
1
0
Fork 0
mirror of https://codeberg.org/rimu/pyfedi synced 2025-01-23 19:36:56 -08:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor remove_tracking_from_link to preserve the t parameter in youtube urls, without regex #34

Browse source
This commit is contained in:
rimu 2024-03-08 22:09:54 +13:00
parent 6914a47206
commit 01423586b5
1 changed files with 19 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
app/utils.py
View file

@ -10,7 +10,7 @@ from typing import List, Literal, Union
import markdown2 import markdown2
import math import math
from urllib.parse import urlparse from urllib.parse import urlparse, parse_qs, urlencode
from functools import wraps from functools import wraps
import flask import flask
from bs4 import BeautifulSoup, NavigableString from bs4 import BeautifulSoup, NavigableString
@ -751,11 +751,25 @@ def sha256_digest(input_string):
def remove_tracking_from_link(url): def remove_tracking_from_link(url):
# strip ?si=abcDEFgh from youtu.be links parsed_url = urlparse(url)
clean = re.search(r"(https://youtu.be/\w+)", url)
if clean is not None: if parsed_url.netloc == 'youtu.be':
return clean.group(1) # Extract video ID
video_id = parsed_url.path[1:] # Remove leading slash
# Preserve 't' parameter if it exists
query_params = parse_qs(parsed_url.query)
if 't' in query_params:
new_query_params = {'t': query_params['t']}
new_query_string = urlencode(new_query_params, doseq=True)
else:
new_query_string = ''
cleaned_url = f"https://youtu.be/{video_id}"
if new_query_string:
cleaned_url += f"?{new_query_string}"
return cleaned_url
else: else:
return url return url