from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
import json
import time

# 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/Sanyangyishi-Fresh-Instant-Rice-Noodles/2156?category=instant01'
#url = 'https://www.sayweee.com/en/product/Nissin-Beef-Flavor-Cup-Noodles/102519?category=instant01'
#url = 'https://www.sayweee.com/en/product/Formosa-Pork-Sung-/23528?category=canned04'

driver.delete_all_cookies
driver.get(url)
time.sleep(2)

# 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

def callback(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(1)
            # 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(1)
        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(f"{zip_button:}")
        zip_button.click()
        time.sleep(1)

        zip_input = driver.find_element(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)

        check_button = driver.find_element(By.XPATH, "//button[text() = 'Confirm']")
        # print(f"{check_button:}")
        check_button.click()
        time.sleep(2)

        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

            # 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 = {sold_out_notice}")
            notice_close.click()
        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()