Skip to content

Commit a203c77

Browse files
committed
Fix a method that finds all unique links on a page
1 parent 87357f2 commit a203c77

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

seleniumbase/fixtures/page_utils.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,11 +166,17 @@ def _get_unique_links(page_url, soup):
166166
elif link.startswith('/'):
167167
link = full_base_url + link
168168
elif link.startswith('./'):
169-
link = full_base_url + link[1:]
169+
f_b_url = full_base_url
170+
if len(simple_url.split('/')) > 1:
171+
f_b_url = full_base_url + "/" + simple_url.split('/')[1]
172+
link = f_b_url + link[1:]
170173
elif link.startswith('#'):
171174
link = full_base_url + link
172175
elif '//' not in link:
173-
link = full_base_url + "/" + link
176+
f_b_url = full_base_url
177+
if len(simple_url.split('/')) > 1:
178+
f_b_url = full_base_url + "/" + simple_url.split('/')[1]
179+
link = f_b_url + "/" + link
174180
else:
175181
pass
176182
unique_links.append(link)

0 commit comments

Comments
 (0)