From 39f3f9c5db8d97ec4be15633b7842a4d0a39ae93 Mon Sep 17 00:00:00 2001 From: Matheus Albino Date: Tue, 3 Oct 2023 22:42:55 -0300 Subject: [PATCH] Add usage of urrlib module --- main.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index 2b4c9b4..e85eb52 100644 --- a/main.py +++ b/main.py @@ -1,3 +1,5 @@ +from urllib.parse import urlparse + from dotenv import dotenv_values from telethon import TelegramClient, events @@ -26,14 +28,18 @@ async def edit_message_url(event): 'furaffinity.net': 'vxfuraffinity.net', } + parsed_url = urlparse(event.raw_text) + for old_url, new_url in replacements.items(): - if old_url in event.raw_text: - updated_text = event.raw_text.replace(old_url, new_url) - updated_text = updated_text.split('?')[0] + if old_url == parsed_url.netloc: + parsed_url = parsed_url._replace(netloc=new_url) + parsed_url = parsed_url._replace(query='') + await client.edit_message( entity=await event.get_chat(), message=event.message, - text=updated_text + text=parsed_url.geturl(), + link_preview=True ) break