@@ -130,6 +130,7 @@ def typeMessage(self, user, message):
130
130
self .__random_sleep__ ()
131
131
132
132
if self .__wait_for_element__ (self .selectors ['send' ], "xpath" ):
133
+ self .__remove_browser_unsupported_banner_if_exists (self , element ):
133
134
self .__get_element__ (self .selectors ['send' ], "xpath" ).click ()
134
135
self .__random_sleep__ (3 , 5 )
135
136
print ('Message sent successfully' )
@@ -269,6 +270,8 @@ def __get_element__(self, element_tag, locator):
269
270
return WebDriverWait (dr , 15 ).until (lambda d : dr .find_element_by_xpath (element_tag ))
270
271
elif locator == 'CSS' and self .is_element_present (By .CSS_SELECTOR , element_tag ):
271
272
return WebDriverWait (dr , 15 ).until (lambda d : dr .find_element_by_css_selector (element_tag ))
273
+ elif locator == 'CLASS' and self .is_element_present (By .CLASS_NAME , element_tag ):
274
+ return WebDriverWait (dr , 15 ).until (lambda d : dr .find_element_by_class_name (element_tag ))
272
275
else :
273
276
logging .info (f"Error: Incorrect locator = { locator } " )
274
277
except Exception as e :
@@ -342,3 +345,17 @@ def __scrolldown__(self):
342
345
def teardown (self ):
343
346
self .driver .close ()
344
347
self .driver .quit ()
348
+
349
+ def __remove_browser_unsupported_banner_if_exists (self , element ):
350
+ element = self .__get_element__ ('rh7Wz' , 'CLASS' )
351
+ if element is not None :
352
+ self .driver .execute_script ("""
353
+ var element = arguments[0];
354
+ element.parentNode.removeChild(element);
355
+ """ , element )
356
+ element = self .__get_element__ ('vohlx' , 'CLASS' )
357
+ if element is not None :
358
+ self .driver .execute_script ("""
359
+ var element = arguments[0];
360
+ element.parentNode.removeChild(element);
361
+ """ , element )
0 commit comments