def delete_image_card(self) -> None: self.driver.execute_script(''' document.querySelector(`{}`).click() '''.format(self.DELETE)) waits.wait(self.driver).until( expected_conditions.presence_of_element_located( (By.CSS_SELECTOR, self.RESTORE)))
def choose(self, album_name: str): self.dropdown.click() transfer_target = self.get_transfer_target(album_name) transfer_target.click() self.start_transfer() waits.wait(self.driver).until( expected_conditions.presence_of_element_located((By.CSS_SELECTOR, self.ACKNOWLEDGEMENT)) )
def disable_edit(self): back_button_wrapper: List[WebElement] = self.driver.find_elements_by_css_selector(self.BACK) if len(back_button_wrapper) == 0: return self.back_button.click() waits.wait(self.driver) \ .until( expected_conditions.presence_of_element_located((By.CSS_SELECTOR, self.EDIT)) )
def request_main_photo(self, new_main_photo: ImageCard) -> ConfirmMakeMainModal: modal: ConfirmMakeMainModal = new_main_photo.make_main() main_photo_change = self.create_main_photo_change_waiter(new_main_photo.id) modal.on_confirm = lambda: waits.wait(self.driver).until(main_photo_change) return modal
def drag_and_drop(self, lhs: ImageCard, rhs: ImageCard): lhs.take_position(rhs) waits.wait(self.driver).until_not( expected_conditions.presence_of_element_located((By.CSS_SELECTOR, self.DRAG_AND_DROP_TRACKER)) )
def wait_uploading(self, count=1) -> None: waits.wait(self.driver).until( waits.number_of_elements_located((By.XPATH, self.UPLOADED_PHOTOS), count) )
def get_transfer_target(self, album_name) -> WebElement: transfer_target = self.TRANSFER_TARGET_TEMPLATE.format(album_name) waits.wait(self.driver).until(expected_conditions.presence_of_element_located((By.XPATH, transfer_target))) return self.driver.find_element_by_xpath(transfer_target)