diff --git a/src/change_url/change_url.py b/src/change_url/change_url.py index a83d854..aa2a1ab 100644 --- a/src/change_url/change_url.py +++ b/src/change_url/change_url.py @@ -91,7 +91,7 @@ def get_url_mapping(urls_with_context, template_type): def extract_urls(text): - pattern = r'(?:\n|^)\s*([^\n]*?)\s*(https?://[\w\.-]+)' + pattern = r'(?:\n|^)\s*([^\n]*?)\s*(https?://[\w\.-]+(?::\d+)?)' # pattern = r'(?:\n|^)\s*([^\n]*?)\s*(https?://[^\s]+)' url_list = re.findall(pattern, text) url_type, is_agent = get_url_type(text, url_list)