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()
