@@ -73,9 +73,13 @@ def __init__(self, browser, container, mapping=dict(), wait_for_seconds=10):
7373 "search" : Selector (select = ".searchBtn" ),
7474 "delete" : Selector (select = ".deleteBtn" ),
7575 "delete_prompt" : Selector (select = ".deletePrompt" ), # [data-test="body"]
76- "delete_btn" : Selector (select = '[data-test="button"][label="Delete"]' ),
76+ "delete_btn" : Selector (
77+ by = By .XPATH ,
78+ select = '//button[@data-test="button" and .//span[text()="Delete"]]' ,
79+ ),
7780 "delete_cancel" : Selector (
78- select = '[data-test="button"][label="Cancel"]'
81+ by = By .XPATH ,
82+ select = '//button[@data-test="button" and .//span[text()="Cancel"]]' ,
7983 ),
8084 "delete_close" : Selector (select = '[data-test="close"]' ),
8185 "delete_loading" : Selector (select = 'button[data-test="wait-spinner"]' ),
@@ -126,11 +130,7 @@ def get_headers(self):
126130 :return: Generator for Str list The headers in the table
127131 """
128132 headers = []
129- GET_PARENT_ELEMENT = (
130- "var parent = arguments[0].firstChild.firstChild;if(parent.hasChildNodes()){var r='';var C=parent.childNodes;"
131- "for(var n=0;n<C.length;n++){if(C[n].nodeType==Node.TEXT_NODE){r+=' '+C[n].nodeValue}}"
132- "return r.trim()}else{return parent.innerText}"
133- )
133+ GET_PARENT_ELEMENT = "return arguments[0].innerText.trim();"
134134 for each in self .get_elements ("header" ):
135135 parent_text = self .browser .execute_script (GET_PARENT_ELEMENT , each )
136136 headers .append (parent_text )
0 commit comments