whatspam.py

import time
import pickle
import random
import sys
import string
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select, WebDriverWait
from selenium.common.exceptions import InvalidArgumentException, NoSuchElementException

import requests

# variable to store the name of the contact
contact = "?asd"
#variable to store message
message = "Hi, bin heute fertig"
timeForFor = 20
answer = "Ich bin Tom"
url = "https://web.whatsapp.com"
rightInfos = False



#Wait for the QR Code
def wait_for_code():
    try:
        element = driver.find_element_by_xpath('//*[@id="side"]/div[1]/div/label/div/div[2]')
    except NoSuchElementException:
        return False
    if(element.is_displayed() == True):
        return True
    else:
        return False

def check_contact_info():
    try:
        while True:
            answer = input('\n'+'\n'+"Are these info's correct: " + '\n' + "Contact: " + contact+ '\n' + "Message: " + message + '\n' + "Time to spam: " + str(timeForFor) + '\n'+ "y/n: ")
            if answer[0] == "y":
                return True
            else:
                if answer[0] == "n":
                    return False
    except IndexError:
        return False



def set_contact():
    return input("Please input contact's name: ")

def set_message():
    return input("Please input message to spam: ")

def set_times():
    try:
        return int(input("Please input time to spam: "))
    except ValueError:
        return 20


#finding search bar
def find_contact(contact):
    inp_xpath_search = '//*[@id="side"]/div[1]/div/label/div/div[2]'
    inp_xpath_cancel = '//*[@id="side"]/div[1]/div/button/div[2]'
    input_box_search = WebDriverWait(driver, 50).until(lambda driver: driver.find_element_by_xpath(inp_xpath_search))
    cancel = driver.find_element_by_xpath(inp_xpath_cancel)
    input_box_search.click() #clicks on search bar
    time.sleep(0)
    input_box_search.send_keys(contact) #enters name of contact
    time.sleep(0.2)

    try:
        #finds contact
        selected_contact = driver.find_element_by_xpath("//span[@title='"+contact+"']")
        selected_contact.click() # select contact
        time.sleep(0.2)
        cancel.click()
    except NoSuchElementException:
        time.sleep(0.2)
        cancel.click()

#find message box
def send_message(message):
    inp_xpath = '//*[@id="main"]/footer/div[1]/div[2]/div/div[2]'
    input_box = driver.find_element_by_xpath(inp_xpath)
    time.sleep(0.2)
    input_box.send_keys(message + Keys.ENTER) #enters message
    time.sleep(0)

def exist_contact(contact):
    inp_xpath_search = '//*[@id="side"]/div[1]/div/label/div/div[2]'
    inp_xpath_cancel = '//*[@id="side"]/div[1]/div/button/div[2]'
    input_box_search = WebDriverWait(driver, 50).until(lambda driver: driver.find_element_by_xpath(inp_xpath_search))
    cancel = driver.find_element_by_xpath(inp_xpath_cancel)
    input_box_search.click() #clicks on search bar
    time.sleep(0)
    input_box_search.send_keys(contact) #enters name of contact
    time.sleep(0.2)

    try:
        #finds contact
        selected_contact = driver.find_element_by_xpath("//span[@title='"+contact+"']")
        time.sleep(0.2)
        selected_contact.click() # select contact
        return True
    except NoSuchElementException:
        time.sleep(0.2)
        cancel.click()
        return False

def aria_labelChecker(aria_label):
    print(aria_label)


#open a Whatsapp Web interface which automatically asks you to scan the QR code
options = webdriver.ChromeOptions()
options.add_argument("user-data-dir=/Users/flo/Library/Application Support/Google/Chrome/Default")
options.add_argument("--app=https://web.whatsapp.com")
print("Please wait, while loading")
try:
    driver = webdriver.Chrome(executable_path="./chromedriver", options=options)
except InvalidArgumentException:
    print('\n'+ '\n'+ "I wanted to start Chrome, but you started it already." + '\n'+ "Please close it, so I can create a fresh browser webpage!"+ '\n'+ '\n')
    sys.exit()


try:



    while wait_for_code() == False:
        time.sleep(0.5)

    while True:
        while rightInfos == False:
            contact = set_contact()
            message = set_message()
            try:
                timeForFor = set_times()
            except ValueError:
                timeForFor = 20
            rightInfos = check_contact_info()

        if(exist_contact(contact) == True):
            rightInfos = True
            break

    find_contact(contact)

    #prints the automated message multiple times
    for count in range(timeForFor):
        send_message(f'{message} {count+1}/{timeForFor}')

        all_children_by_xpath = driver.find_elements_by_xpath('.//*[@id="main"]/div[3]/div/div/div[3]')
        print('len(all_children_by_xpath): ' + str(len(all_children_by_xpath)))


    aria_label = btn.find_element_by_css_selector('span').get_attribute("aria-label")

    while aria_label == Pending:
        time.sleep(0.5)
        aria_label = btn.find_element_by_css_selector('span').get_attribute("aria-label")

    time.sleep(10)

    driver.close()
finally:

    print(" <- Thats a Ctrl+C ? I'll stop for you! Please don't press Ctrl+C until the programm is closed.")
    print("Waiting for messages, that didn't want to go to the Server...")

#    btn = driver.find_element(By.XPATH, '//*[@id="main"]/div[3]/div/div/div[3]/div[94]/div/div/div/div[2]/div/div/span')
#    aria_label = btn.find_element_by_css_selector('span').get_attribute("aria-label")

#    while aria_label == Pending:
#        time.sleep(0.5)
#        aria_label = btn.find_element_by_css_selector('span').get_attribute("aria-label")

    print("Bye, I hope you come back and don't forget me!")
    time.sleep(0.5)
    driver.close()