Spaces:
Sleeping
Sleeping
| import time | |
| from selenium.webdriver.common.by import By | |
| from selenium.common.exceptions import NoSuchElementException | |
| def scroll_and_collect_profiles(driver, log_path="profile_links.log"): | |
| profile_links = set() | |
| last_height = driver.execute_script("return document.body.scrollHeight") | |
| while True: | |
| time.sleep(2) | |
| cards = driver.find_elements(By.XPATH, "//a[contains(@href, '/in/') and @data-test-app-aware-link]") | |
| for card in cards: | |
| href = card.get_attribute("href") | |
| if href and "/in/" in href: | |
| profile_links.add(href) | |
| driver.execute_script("window.scrollBy(0, document.body.scrollHeight);") | |
| time.sleep(2) | |
| new_height = driver.execute_script("return document.body.scrollHeight") | |
| if new_height == last_height: | |
| break | |
| last_height = new_height | |
| with open(log_path, "w", encoding="utf-8") as f: | |
| for link in profile_links: | |
| f.write(link + "\n") | |
| print(f"✅ Collected {len(profile_links)} profile links.") | |
| return list(profile_links) | |
| def visit_profiles_and_connect(driver, profile_links): | |
| for link in profile_links: | |
| try: | |
| driver.get(link) | |
| time.sleep(6) | |
| # Try direct Connect button first | |
| connected = driver.execute_script(""" | |
| const connectBtn = [...document.querySelectorAll('button')] | |
| .find(b => b.textContent.trim() === 'Connect'); | |
| if (connectBtn) { | |
| connectBtn.click(); | |
| return true; | |
| } | |
| return false; | |
| """) | |
| if connected: | |
| time.sleep(3) | |
| driver.execute_script(""" | |
| const sendBtn = [...document.querySelectorAll('button')] | |
| .find(b => b.textContent.trim() === 'Send without a note'); | |
| if (sendBtn) sendBtn.click(); | |
| """) | |
| print(f"✅ Connected directly: {link}") | |
| time.sleep(4) | |
| continue | |
| # Try via More > Connect | |
| used_more = driver.execute_script(""" | |
| const btn = document.querySelector('button[id$="-profile-overflow-action"]'); | |
| if (btn) { | |
| btn.click(); | |
| return true; | |
| } | |
| return false; | |
| """) | |
| if used_more: | |
| time.sleep(3) | |
| found_connect = driver.execute_script(""" | |
| const connectBtn = Array.from(document.querySelectorAll('span.display-flex.t-normal.flex-1')) | |
| .find(el => el.textContent.trim() === 'Connect'); | |
| if (connectBtn) { | |
| connectBtn.click(); | |
| return true; | |
| } | |
| return false; | |
| """) | |
| if found_connect: | |
| time.sleep(3) | |
| driver.execute_script(""" | |
| const sendWithoutNoteBtn = document.querySelector('button[aria-label="Send without a note"]'); | |
| if (sendWithoutNoteBtn) sendWithoutNoteBtn.click(); | |
| """) | |
| print(f"✅ Connected via More menu: {link}") | |
| time.sleep(4) | |
| else: | |
| print(f"❌ Connect option not found in More menu: {link}") | |
| else: | |
| print(f"❌ No Connect or More button: {link}") | |
| except Exception as e: | |
| print(f"⚠️ Error visiting profile {link}: {e}") | |
| time.sleep(3) | |