-
Notifications
You must be signed in to change notification settings - Fork 10
/
WhatsBlaster.py
60 lines (53 loc) · 2.5 KB
/
WhatsBlaster.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import WebDriverWait
import time
class WhatsBlaster:
def __init__(self, chromedriver_path):
options = webdriver.ChromeOptions()
options.add_argument("--force-device-scale-factor=1")
self.driver = webdriver.Chrome(chromedriver_path, chrome_options=options)
self.driver.get("https://web.whatsapp.com")
print "Login, and wait for app to load."
raw_input("Then, hit enter to continue. ")
def close(self):
self.driver.close()
def send_message(self, contact_name, message, timeout=10):
driver = self.driver
# Find contact
search = driver.find_element_by_css_selector("input[title='Search or start new chat']")
search.send_keys(contact_name)
try:
element = WebDriverWait(driver, 10).until(
lambda driver: driver.find_element_by_css_selector(
"span[title='{}']".format(contact_name)
))
except:
return "Failure: could not find contact {}.".format(contact_name)
# Load chat
time.sleep(4)
driver.find_element_by_css_selector(
"#pane-side span[title='{}']".format(contact_name)
).click()
try:
element = WebDriverWait(driver, 10).until(
lambda driver: driver.find_element_by_css_selector(
"#main [title='{}']".format(contact_name)
))
except:
return "Failure: could not load chat with contact {}.".format(contact_name)
# Send message + wait until successful
messages_0 = driver.find_elements_by_css_selector("#main span.selectable-text.invisible-space.copyable-text")
input_element = driver.find_element_by_css_selector("footer div.copyable-text.selectable-text")
input_element.send_keys(message)
input_element.send_keys(Keys.ENTER)
for t in range(timeout):
messages_1 = driver.find_elements_by_css_selector("#main span.selectable-text.invisible-space.copyable-text")
if len(messages_0) != len(messages_1):
return "Success: sent to {}.".format(contact_name)
time.sleep(1)
return "Message to {} may not have been sent.".format(contact_name)
if __name__ == '__main__':
W = WhatsBlaster('./chromedriver')
print W.send_message("Mickey Mouse", "Hello!")
W.close()