from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
import json
import time
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# Configure the Selenium driver options
options = Options()
options.add_argument('--disable-gpu')
driver = webdriver.Chrome(options=options)

# Define the zip codes for which to extract the price
zip_codes = ['85305', '30313', '21230', '14127','28202','60605','45202','44114','76011','80204','48226','54304','77054','46225','32202','64129','89118','90301','33056','55415','02035','70112','07073','19148','15212','95054','98134','33607','37213','20785']

# Load the page and extract the __NEXT_DATA__ data
#url = 'https://www.sayweee.com/en/product/Want-Want-Senbei-Rice-Crackers/17119'
url = 'https://www.sayweee.com/en/product/Lays-Potato-Chips-Cucumber-Flavor/51261'
#url = 'https://www.sayweee.com/en/product/Glico-Pocky-Cookies-and-Cream-Family-Pack/79467'
#url = 'https://www.sayweee.com/en/product/Xiang-Na-Lan-Thai-Jasmine-Rice/19566?category=dried01'
#url = 'https://www.sayweee.com/en/product/Sekka-Premium-Japanese-Medium-Grain-Rice/81224'
#url = 'https://www.sayweee.com/en/product/Big-Green-Spring-Farm-Organic-Millet/90991?category=dried02'
#url = 'https://www.sayweee.com/en/product/Liuzhou-Guangxi-Specialty-LuoSiFen-Hot-Spicy-Noodles/95442?category=instant01'
#url = 'https://www.sayweee.com/en/product/House-Foods-Firm-Tofu-/85175?trace_id=c93f3d1a-80a4-4e1a-8d41-a402ae43186f'
#url = 'https://www.sayweee.com/en/product/Lactaid-Lactose-Free-2--Reduced-Fat-Milk/95353?trace_id=7847dd6e-c511-48e2-a76e-a283399967b1'
#url = 'https://www.sayweee.com/en/product/Washington-Fuji-Apples/94817?category=fruits03'

driver.delete_all_cookies
driver.get(url)

# Close the cookies setting window
try:  
    cookie_button = WebDriverWait(driver, 2).until(EC.presence_of_element_located((By.XPATH, "/html/body/div[3]/div[2]/div/div/div[3]/button")))
    print(f"cookie_button = {cookie_button}")
    cookie_button.click()
except:
    pass

# Clear any modal dialog
try:    
    modal_close = WebDriverWait(driver, 2).until(EC.presence_of_element_located((By.XPATH, "//*[contains(concat(' ', normalize-space(@class), ' '), ' icon-icon-close-s ')]")))
    print(f"modal_close = {modal_close}")
    modal_close.click()
except:
    pass

def callback(zip_code):
    try:
        try:
            # Clear any modal dialog
            modal_close = WebDriverWait(driver, 2).until(EC.presence_of_element_located((By.XPATH, "//*[contains(concat(' ', normalize-space(@class), ' '), ' icon-icon-close-s ')]")))
            # print(f"modal_close = {modal_close}")
            modal_close.click()
        except:
            pass


        try:
            # Clear any notice
            notice_close = WebDriverWait(driver, 5).until(EC.presence_of_element_located((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()
        except:
            pass

        try:
            # Clear any modal dialog
            modal_close = WebDriverWait(driver, 2).until(EC.presence_of_element_located((By.XPATH, "//*[contains(concat(' ', normalize-space(@class), ' '), ' icon-icon-close-s ')]")))
            # print(f"modal_close = {modal_close}")
            modal_close.click()
        except:
            pass

        # Enter zip code in the input field and click the "Check" button
        zip_button = WebDriverWait(driver, 5).until(EC.presence_of_element_located((By.XPATH, '/html/body/div[2]/div/header/div[3]/div/div/div[2]/div[1]')))
        # print(f"{zip_button:}")
        zip_button.click()

        zip_input = WebDriverWait(driver, 5).until(EC.presence_of_element_located((By.XPATH, '//div/label/following-sibling::input')))
        # print(f"{zip_input:}")
        zip_input.click()
        # print("clearing zip_input")
        zip_input.send_keys("\b\b\b\b\b")
        # print("sending zip code")
        zip_input.send_keys(zip_code)

        try:
            check_button = WebDriverWait(driver, 5).until(EC.presence_of_element_located((By.XPATH, "//button[text() = 'Confirm']")))
            # print(f"{check_button:}")
        except:
            check_button = WebDriverWait(driver, 5).until(EC.presence_of_element_located((By.XPATH, "/html/body/div[5]/div/div[2]/div/div[2]/div[2]/div/div[1]/form/div/div/div[1]/div/div/div[3]/div/button")))
        finally:
            check_button.click()

        try:
            # Clear any notice
            notice_close = WebDriverWait(driver, 5).until(EC.presence_of_element_located((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()
        except:
            pass

        try:
            # Clear any modal dialog
            modal_close = WebDriverWait(driver, 2).until(EC.presence_of_element_located((By.XPATH, "//*[contains(concat(' ', normalize-space(@class), ' '), ' icon-icon-close-s ')]")))
            # print(f"modal_close = {modal_close}")
            modal_close.click()
        except:
            pass

            # Extract the price of rice crackers for this zip code
        try:
            sold_out_notice = driver.find_element(By.XPATH, '//*[@id="soldStatusNotice"]')
            print(f"No such product in this area = {zip_code, sold_out_notice}")
        except:
            print(f"The product is available in this area {zip_code}")
    
    except Exception as e:
        print(f"Caught exception: {e}")
        print(f"Price not found for zip code {zip_code}")

for zip_code in zip_codes:
    callback(zip_code)

driver.quit()