This commit is contained in:
Austin-Olacsi 2024-05-03 18:28:48 +02:00 committed by GitHub
commit c1bf17af67
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -33,7 +33,9 @@ def on_result(_request, _search, result):
# (only) on the 'parsed_url'
if not replacement:
return False
result[parsed] = result[parsed]._replace(netloc=pattern.sub(replacement, result[parsed].netloc))
subdomain = pattern.search(result[parsed].netloc).group(1) or ''
new_netloc = replacement.replace('(*)', subdomain)
result[parsed] = result[parsed]._replace(netloc=pattern.sub(new_netloc, result[parsed].netloc))
result['url'] = urlunparse(result[parsed])
for url_field in _url_fields:
@ -43,7 +45,9 @@ def on_result(_request, _search, result):
if not replacement:
del result[url_field]
else:
url_src = url_src._replace(netloc=pattern.sub(replacement, url_src.netloc))
subdomain = pattern.search(url_src.netloc).group(1) or ''
new_netloc = replacement.replace('(*)', subdomain)
url_src = url_src._replace(netloc=pattern.sub(new_netloc, url_src.netloc))
result[url_field] = urlunparse(url_src)
return True