Added better xpath selectors for the UI elements
This commit is contained in:
@@ -0,0 +1,129 @@
|
||||
from google.cloud import pubsub_v1
|
||||
from selenium import webdriver
|
||||
from selenium.webdriver.chrome.options import Options
|
||||
from selenium.webdriver.common.by import By
|
||||
import json
|
||||
import time
|
||||
import os
|
||||
from concurrent.futures import TimeoutError
|
||||
|
||||
# os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "C:/Users/Karen/AppData/Roaming/gcloud/application_default_credentials.json"
|
||||
|
||||
# Configure the Selenium driver options
|
||||
options = Options()
|
||||
options.add_argument('--disable-gpu')
|
||||
driver = webdriver.Chrome(options=options)
|
||||
|
||||
# Set up Pub/Sub subscription
|
||||
project_id = "fp-20230425"
|
||||
subscription_name = "fp-subscription-NFL-zipcodes"
|
||||
subscriber = pubsub_v1.SubscriberClient()
|
||||
subscription_path = subscriber.subscription_path(project_id, subscription_name)
|
||||
timeout = 5
|
||||
|
||||
# Load the product page and extract the __NEXT_DATA__ data
|
||||
url = 'https://www.sayweee.com/en/product/Want-Want-Senbei-Rice-Crackers/17119'
|
||||
driver.delete_all_cookies
|
||||
driver.get(url)
|
||||
# time.sleep(5)
|
||||
|
||||
time.sleep(5)
|
||||
# Close the cookies setting window
|
||||
try:
|
||||
cookie_button = driver.find_element(By.XPATH, "/html/body/div[3]/div[2]/div/div/div[3]/button")
|
||||
print(f"cookie_button = {cookie_button}")
|
||||
cookie_button.click()
|
||||
except:
|
||||
pass
|
||||
|
||||
time.sleep(5)
|
||||
# Clear any modal dialog
|
||||
try:
|
||||
modal_close = driver.find_element(By.XPATH, "//*[contains(concat(' ', normalize-space(@class), ' '), ' icon-icon-close-s ')]")
|
||||
print(f"modal_close = {modal_close}")
|
||||
modal_close.click()
|
||||
time.sleep(2)
|
||||
except:
|
||||
pass
|
||||
|
||||
# Define callback function for Pub/Sub messages
|
||||
def callback(message):
|
||||
# Extract zip code from message data
|
||||
zip_code = message.data.decode('utf-8')
|
||||
print(f"Received zip code: {zip_code}")
|
||||
|
||||
try:
|
||||
try:
|
||||
time.sleep(5)
|
||||
# Clear any modal dialog
|
||||
modal_close = driver.find_element(By.XPATH, "//*[contains(concat(' ', normalize-space(@class), ' '), ' icon-icon-close-s ')]")
|
||||
print(f"modal_close = {modal_close}")
|
||||
modal_close.click()
|
||||
time.sleep(2)
|
||||
except:
|
||||
pass
|
||||
|
||||
|
||||
try:
|
||||
time.sleep(5)
|
||||
# Clear any notice
|
||||
notice_close = driver.find_element(By.XPATH, "/html/body/div[3]/div/div[2]/div/div[2]/div/div/div[1]/div[2]/button")
|
||||
print(f"notice_close = {notice_close}")
|
||||
notice_close.click()
|
||||
time.sleep(2)
|
||||
except:
|
||||
pass
|
||||
|
||||
# Enter zip code in the input field and click the "Check" button
|
||||
zip_button = driver.find_element(By.XPATH, '/html/body/div[2]/div/header/div[3]/div/div/div[2]/div[1]')
|
||||
print(zip_button)
|
||||
zip_button.click()
|
||||
time.sleep(2)
|
||||
|
||||
zip_input = driver.find_element(By.XPATH, '//div/label/following-sibling::input')
|
||||
zip_input.clear()
|
||||
zip_input.send_keys(zip_code)
|
||||
|
||||
check_button = driver.find_element(By.XPATH, "//button[text() = 'Confirm']")
|
||||
check_button.click()
|
||||
time.sleep(2)
|
||||
|
||||
# Extract the price of rice crackers for this zip code
|
||||
|
||||
'''
|
||||
price_element = driver.find_element(By.XPATH, '/html/body/div[2]/div/div[1]/div[1]/div/div/div[2]/div[2]/div[1]/div[4]/div[1]/div')
|
||||
price = price_element.text.strip()[1:]
|
||||
print(price)
|
||||
'''
|
||||
next_data_element = driver.find_element(By.XPATH, "//script[@type='application/ld+json']")
|
||||
next_data_text = next_data_element.get_attribute('innerHTML')
|
||||
next_data_start_index = next_data_text.index('{')
|
||||
next_data_end_index = next_data_text.rindex('}') + 1
|
||||
next_data_json = next_data_text[next_data_start_index:next_data_end_index]
|
||||
next_data = json.loads(next_data_json)
|
||||
price = next_data['offers']['price']
|
||||
# Print the price
|
||||
print(price)
|
||||
message.ack() # acknowledge receipt of message
|
||||
except:
|
||||
print(f"Price not found for zip code {zip_code}")
|
||||
price = 'Price not found'
|
||||
|
||||
|
||||
# Start receiving messages
|
||||
streaming_pull_future = subscriber.subscribe(subscription_path, callback=callback)
|
||||
print(f'Started receiving messages from subscription {subscription_path}...')
|
||||
|
||||
# Wrap subscriber in a 'with' block to automatically call close() when done.
|
||||
with subscriber:
|
||||
try:
|
||||
# When `timeout` is not set, result() will block indefinitely,
|
||||
# unless an exception is encountered first.
|
||||
streaming_pull_future.result(timeout=timeout)
|
||||
except TimeoutError:
|
||||
streaming_pull_future.cancel() # Trigger the shutdown.
|
||||
streaming_pull_future.result() # Block until the shutdown is complete.
|
||||
|
||||
# Keep the script running indefinitely to continue receiving messages
|
||||
while True:
|
||||
time.sleep(10)
|
||||
Reference in New Issue
Block a user