main.py

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.webdriver.support.ui import WebDriverWait
import time, json
import requests
from termcolor import colored, cprint  ### SUCCESS=green, TRYING=yellow, ERROR/FAIL=red, SYSTEM=magenta ###
import colorama

colorama.init()  # Bugfixing colours on Windows

# Launching chrome drivers
chrome_options = Options()
#chrome_options.add_argument("--headless")  # remove the "#" at the start of the line to launch Chrome in "headless" mode.
#prefs = {"profile.managed_default_content_settings.images": 2}  # disables loading images 
#chrome_options.add_experimental_option("prefs", prefs)  # adds "disable img" option to chrome's options
# CAN'T ADD IMAGELESS BECAUSE OF CAPTCHA

driver = webdriver.Chrome(executable_path=r'./chromedriver.exe', options=chrome_options)


# URLs that we will be using
MAIN_URL = 'https://www.supremenewyork.com'
SHOP_URL = 'https://www.supremenewyork.com/shop'
CHECKOUT_URL = 'https://www.supremenewyork.com/checkout'


# Defining functions
def visit_url(url):
    """Visits a url/link/webpage."""
    try:
        driver.get(url)
        cprint(f'Loaded "{url}".', 'green')
    except Exception as e:
        cprint(f'Website "{url}" can not be reached!.', 'red')
        print(e)


def search_item(category, keywords):
    """Find the value which includes the most of keywords."""
    try: 
        keyword_list = keywords.split(", ")

        item_url = requests.get(f'https://www.supremenewyork.com/shop.json')
        data_obj = item_url.json()

        old_counter = 0
        for value in data_obj['products_and_categories'][category]:  # For every value in a category:
            counter = 0  # Counter resets for every value in the category.
            for word in keyword_list:  # Iterate every keyword that user has entered.
                if word.lower() in value['name'].lower():  # If keyword inside value:  
                    counter += 1  # counter +1 if the value has a keyword in it.
                    # do this for every word in the value.
            if counter > old_counter:  # If the latest value has the most keywords compared to previous value:
                old_counter = counter  
                best_value = value  # Save the value
        #print(best_value)  # print the found json object
        return best_value['id']  # return the id of the item

    except Exception as e:
        cprint(f'Item not found!', 'red')
        print(e)


def select_item_colour(colour_input, id):
    """Selects item colour.

    - [X] Switch to online database.
    - [ ] Priority list (Grey, Purple...)
    """
    try:
        colour_list = colour_input.split(", ")

        item_url = requests.get(f'https://www.supremenewyork.com/shop/{id}.json')
        data_obj = item_url.json()
        
        old_counter = 0
        for value in data_obj['styles']:  # For every value in a style:
            counter = 0  # Counter resets for every value in the category.
            for word in colour_list:  # Iterate every keyword that user has entered.
                if word.lower() in value['name'].lower():  # If keyword inside value:  
                    counter += 1  # counter +1 if the value has a keyword in it.
                    # do this for every word in the value.
            if counter > old_counter:  # If the latest value has the most keywords compared to previous value:
                old_counter = counter  
                best_value = value  # Save the value
                
        driver.find_element_by_xpath(f'//*[@id="details"]/ul/li//a[@data-style-name="{best_value["name"]}"][@data-style-id="{best_value["id"]}"]').click()
        cprint(f'Colour "{best_value["name"]}" selected.', 'green')
    except Exception as e:
        cprint(f'Colour "{best_value["name"]}" not found!', 'red')
        print(e)


def select_item_size(size_input):
    """Selects the size of the item.
    
    Add:
    - [X] Priority list (S,XL,L,M)
    """

    size_list = size_input.split(", ")

    for i in range (0,len(size_list)):
        if size_list[i] == 'S':
            size_list[i] = 'Small'
        elif size_list[i] == 'M':
            size_list[i] = 'Medium'
        elif size_list[i] == 'L':
            size_list[i] = 'Large'
        elif size_list[i] == 'XL':
            size_list[i] = 'XLarge'
        try:
            Select(driver.find_element_by_xpath('//*[@id="size"]')).select_by_visible_text(size_list[i])
            cprint(f'Size "{size_list[i]}" selected.', 'green')
            break
        except Exception as e:
            cprint(f'Size "{size_list[i]}" out of stock!', 'red')
            try:
                cprint(f'Checking "{size_list[i+1]}"...', 'yellow')
            except IndexError:
                cprint(f'Ending program...', 'magenta', attrs=['bold'])
                break


def add_to_basket():
    """Clicks "add to basket" button when on an item's webpage."""
    try:
        WebDriverWait(driver, 3).until(EC.presence_of_element_located((By.XPATH, f'//*[@id="add-remove-buttons"]/input')))
        driver.find_element_by_xpath('//*[@id="add-remove-buttons"]/input').click()
        cprint(f'Added item to basket.', 'green')
    except Exception as e:
        cprint(f'Item out of stock!', 'red')
        print(e)


def checkout():
    """Clicks "checkout" button and proceeds to /checkout.
    
    For Error Handling:
     - [ ] If clicking checkout now fails, directly go to the checkout link.
    """
    try: 
        WebDriverWait(driver, 3).until(EC.presence_of_element_located((By.XPATH, f'/html/body/div[2]/div/div[1]/div/a[2]')))
        time.sleep(0.2) # stops and waits "checkout" button to appear. This is under development
        driver.find_element_by_xpath('/html/body/div[2]/div/div[1]/div/a[2]').click()
        cprint('Clicked "checkout now" button.', 'green')
    except Exception as e:
        cprint('Could not click "checkout now" button!', 'red')
        print(e)


def enter_shipping_info(ship_config):
    """Enters the shipping info/details of the given profile/config.

    :param ship_config: A specific "id" of which shipping configuration/profile we want to load.
    :type  ship_config: int

    :Example:

    >>> ship_config = 1
    open(f'./data/shipping/shipping_1.json')
    """
    try:
        WebDriverWait(driver, 3).until(EC.presence_of_element_located((By.XPATH, f'//*[@id="order_billing_name"]'))) # Waits until shipping form to get loaded.
        with open(f'./data/shipping/shipping_{ship_config}.json') as shipping_info:
            data = json.load(shipping_info)
            WebDriverWait(driver, 3).until(EC.presence_of_element_located((By.XPATH, f'//*[@id="order_billing_name"]'))) 
            driver.find_element_by_xpath('//*[@id="order_billing_name"]').send_keys(data['fullName'])
            driver.find_element_by_xpath('//*[@id="order_email"]').send_keys(data['email'])
            driver.find_element_by_xpath('//*[@id="order_tel"]').send_keys(data['fullName'])
            driver.find_element_by_xpath('//*[@id="bo"]').send_keys(data['address_1'])
            if data['address_2'] == '':
                driver.find_element_by_xpath('//*[@id="oba3"]').send_keys(data['address_2'])
            else:
                pass
            driver.find_element_by_xpath('//*[@id="order_billing_address_3"]').send_keys(data['address_3'])
            driver.find_element_by_xpath('//*[@id="order_billing_city"]').send_keys(data['city'])
            driver.find_element_by_xpath('//*[@id="order_billing_zip"]').send_keys(data['postcode'])
            Select(driver.find_element_by_xpath('//*[@id="order_billing_country"]')).select_by_visible_text(data['country'])
        cprint('Entered shipping details.', 'green')
    except Exception as e:
        cprint('Could not enter shipping details!', 'red')
        print(e)


def enter_card_info(card_config):
    """Enters the card info/details of the given profile/config.

    :param card_config: A specific "id" of which card configuration/profile we want to load.
    :type  card_config: int

    :Example:

    >>> card_config = 1
    open(f'./data/shipping/card_1.json')
    """
    try: 
        with open(f'./data/card/card_{card_config}.json') as card_info:
            data = json.load(card_info)
            Select(driver.find_element_by_xpath('//*[@id="credit_card_type"]')).select_by_visible_text(data['cardType'])
            driver.find_element_by_xpath('//*[@placeholder="number"]').send_keys(data['cardNumber'])
            Select(driver.find_element_by_xpath('//*[@id="credit_card_month"]')).select_by_visible_text(data['cardMonth'])
            Select(driver.find_element_by_xpath('//*[@id="credit_card_year"]')).select_by_visible_text(data['cardYear'])
            driver.find_element_by_xpath('//*[@placeholder="CVV"]').send_keys(data['cardCVV'])
            driver.find_element_by_xpath('//*[@id="cart-cc"]/fieldset/p/label/div/ins').click()  # terms and conditions
        cprint('Entered card details.', 'green')
    except Exception as e:
        cprint('Could not enter card details!', 'red')
        print(e)


def process_payment():
    """Clicks "process payment" button.
    
    For Error handling:
    - [ ] Scrape data after checkout and see if it really checked out. 
    """
    try:
        driver.find_element_by_xpath('//*[@id="pay"]/input').click()
        cprint(f'Clicked "process payment" button.', 'green')
    except Exception as e:
        cprint(f'Could not click "process payment" button!', 'red')
        print(e)



visit_url(SHOP_URL)

SHIPPING_CONFIG = input('Shipping config "id" [1|2|3...]: ')
CARD_CONFIG = input('Card config "id" [1|2|3...]: ')

CATEGORY = input('Select a category [Skate|Accessories|Tops/Sweaters|Pants|Jackets|Sweatshirts|Shirts|T-Shirts|Hats|Bags|new]: ')
KEYWORD_INPUT = input('Enter the keywords of the item (i.e  box, logo, sweatshirt): ')

COLOUR = input('Enter which colour you want [Purple|Black|Navy...]: ')
SIZES = input('Enter which size(s) you want in decreasing prirotiy (i.e  S, XL...): ')

input("Press ENTER to start")
start_time = time.time()  # Start timer 
ITEM_ID = search_item(CATEGORY, KEYWORD_INPUT)
visit_url(SHOP_URL + f'/{ITEM_ID}')

select_item_colour(COLOUR, ITEM_ID)
select_item_size(SIZES)
add_to_basket()
checkout()
enter_shipping_info(SHIPPING_CONFIG)
enter_card_info(CARD_CONFIG)
#process_payment()
elapsed_time = time.time() - start_time # End timer
cprint(f'Total processing time: {float(round(elapsed_time,2))} seconds', 'magenta', attrs=['bold'])


time.sleep(3000) # Wait for 50 minutes before manually closing Chrome.