tineye.py

from seleniumwire import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from selenium.common.exceptions import NoSuchElementException, TimeoutException
import chromedriver_autoinstaller
import os
import re
import random
import json
from os.path import isfile
import sys
from clear_screen import clear

def load_settings():
    if not isfile("proxy.txt"):
        with open("proxy.txt", 'w') as proxy_file:
            proxy_file.write("DEFAULT_PROXY")
        print("Proxy file created.")
    return

def load_proxies(file_path):
    with open(file_path, 'r') as proxy_file:
        proxies = proxy_file.readlines()
    return proxies

def initialize_driver(proxy):
    chromedriver_autoinstaller.install()
    options = webdriver.ChromeOptions()
    options.add_argument('--window-size=1920,1080')
    options.add_argument('--headless')
    options.add_argument('--no-sandbox')
    options.add_argument('--disable-gpu')
    options.add_argument('--disable-extensions')
    options.add_argument('--disable-dev-shm-usage')
    options.add_argument('--disable-blink-features=AutomationControlled')
    options.add_argument('--ignore-certificate-errors')
    options.add_argument('--ignore-ssl-errors')
    options.add_argument('--disable-dev-shm-usage')
    options.add_argument("--disable-logging")
    options.add_argument("--log-level=3")
    options.add_experimental_option("excludeSwitches", ["enable-logging"])
    options.add_argument("--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3")
    options.add_argument("--start-maximized")
    options.add_argument("--disable-blink-features")
    options.add_argument("--disable-blink-features=AutomationControlled")
    proxy_parts = proxy.split(':')
    formatted_proxy = f'{proxy_parts[2]}:{proxy_parts[3]}@{proxy_parts[0]}:{proxy_parts[1]}'
    proxy_options = {
        'proxy': {
            'https': f'https://{formatted_proxy}',
        }
    }
    driver = webdriver.Chrome(seleniumwire_options=proxy_options, options=options)
    clear()
    return driver

def remove_proxy(file_path, proxy):
    with open(file_path, 'r') as f:
        proxies = f.readlines()
    with open(file_path, 'w') as f:
        for p in proxies:
            if p.strip() != proxy:
                f.write(p)
                
driver = None
if __name__ == "__main__":
    try:
        if len(sys.argv) > 1:
            path = sys.argv[1]
        else:
            path = str(input("Please enter a filename: "))
        if not os.path.isfile(os.path.join(os.path.dirname(__file__), path)):
            print("File not found in the current directory!")
            os._exit(1)
        settings = load_settings()
        proxies = load_proxies("proxy.txt")
        proxy = random.choice(proxies).strip()
        if(proxy == "DEFAULT_PROXY"):
            print("No proxy found in proxy.txt! Please add a proxy.")
            os._exit(1)
        driver = initialize_driver(proxy)
        wait = WebDriverWait(driver, 60)
        driver.get("https://tineye.com")

        try:
            upload = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="upload_box"]')))
            upload.send_keys(os.path.join(os.path.dirname(__file__), path))
        except TimeoutException:
            print("Upload button not found!")
            driver.quit()
            os._exit(1)

        try:
            results = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="result_count"]')))
            results_text = results.text.replace(',', '')  # Remove commas.
            num_results = re.search(r'\d+', results_text).group()
            print("Proxy: ", proxy)
            print("Results: ", num_results)
            print("Search url: ", driver.current_url)
        except TimeoutException:
            print("Results not found")
            driver.quit()

        try:
            os.remove(os.path.join(os.path.dirname(__file__), path))
        except:
            print("Error deleting file!")
            driver.quit()
            os._exit(1)
            
    except Exception as e:
        print(f"An error occurred: {e}!")
    finally:
        if driver is not None:
            driver.quit()