def open_notes(self): open_menu_btn = Lib.simple_wait_element_css( self.driver, self.OPEN_MENU_CSS) open_menu_btn.click() notes_menu = Lib.simple_wait_element_css( self.driver, self.NOTES_BTN_CSS) notes_menu.click()
def save(self): btn_confirm = Lib.simple_wait_element_css(self.driver, self.BTN_CONFIRM_CSS) btn_confirm.click() btn_close = Lib.simple_wait_element_css(self.driver, self.BTN_CLOSE_CSS) btn_close.click()
def back_to_start_gender(self): self.open_profile() current_id = self.get_current_gender() _id = 1 if current_id == 1: _id = 2 Lib.simple_wait_element_css( self.driver, "input[id='field_gender_%d']" % _id).click()
def delete_album(self): self.select_created_album() self.set_edit_album() delete_link = Lib.simple_wait_element_css(self.driver, self.DELETE_ALBUM_CSS) delete_link.click() delete_btn = Lib.simple_wait_element_css(self.driver, self.DELETE_CONFIRM_BTN) delete_btn.click()
def unfollow(self): url = Lib.simple_wait_element(self.driver, self.URL_CSS) self.driver.get(url.get_attribute('href')) unfollow_arrow_btn = Lib.simple_wait_element_css( self.driver, self.ARROW_UNFOLLOW_BTN_CSS) unfollow_arrow_btn.click() unfollow_btn = Lib.simple_wait_element_css(self.driver, self.UNFOLLOW_CSS) unfollow_btn.click()
def get_current_gender(self): self.click_edit() Lib.simple_wait_element_css(self.driver, '#field_gender_1') isChecked = self.driver.execute_script( 'return document.getElementById("field_gender_1").checked') cur_gender = -1 if isChecked: cur_gender = 1 else: cur_gender = 2 return cur_gender
def like_first_found_post(self): like_btn = Lib.simple_wait_element_css(self.driver, self.LIKE_BTN_CSS) self.DATA_ID = like_btn.get_attribute(self.LIKE_BTN_ATTRIBUTE) self.likes_count = like_btn.get_attribute(self.DATA_COUNT) if self.likes_count == None: self.likes_count = 0 self.jsClick(like_btn) # Здесь он нужен
def get_pending_friends(self): self.USER_URL = Lib.simple_wait_element_css( self.driver, self.CURRENT_USER_CSS).get_attribute('href') self.driver.get(self.USER_URL + self.PENDING_PREFIX) self.scroll_to_new() ids = [] for f in self.PENDING_FRIENDS: ids.append(f.get_attribute("href").split("/")[-1]) return self.FRIEND_ID in ids
def change_gender(self): new_gender = -1 if self.START_GENDER == 1: new_gender = 2 else: new_gender = 1 _id = "field_gender_%d" % new_gender gender_radio = Lib.simple_wait_element_css(self.driver, "input[id='%s']" % _id) gender_radio.click()
def scroll_to_new(self): self.OUTGOING_FRIENDS_COUNT = int( Lib.simple_wait_element_css(self.driver, self.OUTGOING_FRIENDS_COUNT_CSS).text) self.PENDING_FRIENDS = Lib.simple_get_elements_css( self.driver, self.PENDING_FRIENDS_CSS) while len(self.PENDING_FRIENDS) < self.OUTGOING_FRIENDS_COUNT: self.driver.execute_script( "window.scrollTo(0, document.body.scrollHeight)") self.PENDING_FRIENDS = Lib.simple_get_elements_css( self.driver, self.PENDING_FRIENDS_CSS)
def upload_note(self): el = Lib.simple_wait_element_css(self.driver, self.UPLOAD_NOTE_CSS) el.click()
def focus_note(self): el = Lib.simple_wait_element_css(self.driver, self.FOCUS_NOTE_CSS) el.click()
def set_note_text(self): el = Lib.simple_wait_element_css( self.driver, self.SET_NOTE_TEXT_CSS) el.send_keys(self.DEFAULT_NOTE_TEXT)
def open(self): changeBtn = Lib.simple_wait_element_css(self.driver, self.LANGUAGE_ELEMENT_CSS) self.jsClick(changeBtn) # Здесь он нужен
def add_to_friends(self): add_btn = Lib.simple_wait_element_css(self.driver, self.ADD_BTN_CSS) add_btn.click()
def select_created_album(self): album = Lib.simple_wait_element_css(self.driver, self.ALBUM_LINK_CSS) album.click()
def set_edit_album(self): edit_btn = Lib.simple_wait_element_css(self.driver, self.EDIT_ALBUM_CSS) edit_btn.click()
def send_message(self): el = Lib.simple_wait_element_css(self.driver, self.SEND_BTN_CSS) el.click()
def create_album(self): create_album_btn = Lib.simple_wait_element_css( self.driver, self.CREATE_ALBUM_BTN_CSS) create_album_btn.click() save_btn = Lib.simple_wait_element_css(self.driver, self.SAVE_BTN_CSS) save_btn.click()
def search(self): search_btn = Lib.simple_wait_element_css(self.driver, self.SEARCH_BTN_CSS) search_btn.click()
def get_last_post(self): last_post = Lib.simple_wait_element_css( self.driver, self.LAST_POST_CSS) return last_post.text
def get_likes_from_btn_by_owner(self): like_btn = Lib.simple_wait_element_css( self.driver, self.LIKE_BTN_OWNER % self.DATA_ID) return like_btn.get_attribute(self.DATA_COUNT)
def selectFirstDialog(self): el = Lib.simple_wait_element_css(self.driver, self.FIRST_DIALOG_CSS) el.click() self.dialog_id = el.get_attribute('href').split("/")[-1]
def click_edit(self): el = Lib.simple_wait_element_css(self.driver, self.EDIT_BTN_CSS) el.click()
def change(self): el = Lib.simple_wait_element_css(self.driver, self.INACTIVE_LANGUAGE) el.click()
def get_active_language(self): el = Lib.simple_wait_element_css(self.driver, self.ACTIVE_LANGUAGE) return el.text
def open_photos_page(self): el = Lib.simple_wait_element_css(self.driver, self.PHOTOS_URL_CSS) el.click()
def writeMessage(self): message_field = Lib.simple_wait_element_css( self.driver, self.MESSAGE_FIELD_CSS) message_field.send_keys(self.DEFAULT_MESSAGE)
def remove_like(self): like_btn = Lib.simple_wait_element_css( self.driver, self.LIKE_BTN_OWNER % self.DATA_ID) like_btn.click()
def apply(self): el = Lib.simple_wait_element_css(self.driver, self.CONFIRM_BTN_CSS) el.click()