diff --git a/whatsapp_api.py b/whatsapp_api.py index d19d204..884fb8d 100644 --- a/whatsapp_api.py +++ b/whatsapp_api.py @@ -16,6 +16,10 @@ class WhatsApp: + + if __name__ == '__main__': + main() + def __init__(self): self.driver = self._setup_driver() self.driver.get(WP_LINK) @@ -92,4 +96,24 @@ def get_last_message(self): all_messages = self.get_all_messages() return all_messages[-1] + def zap_to_list(self, contact_list): + + def search_contact(self, keyword): + '''Write and send message''' + self._click(NEW_CHAT) + self._send_keys(SEARCH_CONTACT, keyword) + sleep(1) + try: + self._click(FIRST_CONTACT) + except Exception as e: + print("Contact not found") + + def send_message(self, message): + '''Write and send message''' + self.write_message(message) + self._click(SEND) + standard_message = input('Type your message: ') + for contact in contact_list: + do = search_contact(self, contact) + msg = send_message(self,standard_message) \ No newline at end of file