Selenium - O que você deveria saber - Parte 2

Brincando com formulários

Quantas vezes você já não preencheu um formulário na web?

Hoje vamos aprender como fazer isso, vamos visualizar o exemplo abaixo que procura por um termo no google.

from selenium import webdriver
from selenium.webdriver.common.keys import Keys

firefox = webdriver.Firefox()
firefox.get('http://google.com.br/')

# pegar o campo de busca onde podemos digitar algum termo
campo_busca = firefox.find_element_by_name('q')

# Digitar "Python Club" no campo de busca
campo_busca.send_keys('Python Club')

# Simular que o enter seja precisonado
campo_busca.send_keys(Keys.ENTER)

Foi bem simples, encontramos o campo_busca e invocamos o metódo send_keys com o texto que desejamos digitar e depois simulamos o pressionamento do botão "Enter".

Nota: Sempre que houver a necessidade de utilizar uma tecla especial podemos encontrá-la na classe Keys.

E se for um campo select, como eu faço?

Digamos que exista o seguinte código:

<select name="estados">
  <option value="mg">MG</option>
  <option value="rj">RJ</option>
  <option value="sp">SP</option>
</select>

Existe uma classe só para facilitar o trabalho com selects, ela possui metódos para selecionar uma opção pelo seu texto ou valor, entre outras.

# Importar a classe Select
>> from selenium.webdriver.support.ui import Select

# É preciso passar o elemento para a classe
>> estados = Select(firefox.find_element_by_name('estados'))

# Selecionar a opção MG
>> estados.select_by_visible_text('MG')

# Selecionar a opção SP
>> estados.select_by_value('sp')

Agora você já sabe o básico para manipular um formulário, você pode digitar nos campos, selecionar valores, clicar em botões.

Trabalhando com múltiplas janelas

E quando se abre uma nova janela? ou um alerta? O que fazemos?

>> firefox.window_handles
[u'{7fd12d82-4fb3-48a4-a8b9-e1e460c00236}']

A instância firefox possui um atributo chamado window_handles, que é uma lista contendo um ID para cada janela aberta.

Nota: O ID é criado para nova janela aberta e não para uma ABA ou um alerta javascript.

Quando você abrir uma nova janela, poderá perceber que é criado automaticamente um novo ID.

>> firefox.window_handles
[u'{7fd12d82-4fb3-48a4-a8b9-e1e460c00236}', u'{2ce4de19-0902-48e3-a1cc-50f6378afd79}']

Para alternar entre janelas, basta chamar o metódo switch_to_window passando o ID da janela como parâmetro.

Imagine que temos uma janela aberta na página do Google e a outra na página da Python Club.

>> firefox.window_handles
[u'{7fd12d82-4fb3-48a4-a8b9-e1e460c00236}', u'{2ce4de19-0902-48e3-a1cc-50f6378afd79}']

# Título da janela atual
>> firefox.title
u'Google'

# Trocar para a última janela da lista
>> firefox.switch_to_window_handles(firefox.window_handles[-1])
>> firefox.title
u'PythonClub //'

# Fechar a janela atual
>> firefox.close()

# Voltar para a janela do Google
>> firefox.switch_to_window_handles(firefox.window_handles[-1])
>> firefox.title
u'Google'

E como saber o ID da janela atual? Simples!

>> firefox.current_window_handle
u'{7fd12d82-4fb3-48a4-a8b9-e1e460c00236}'

Fácil não?!

E se abrir um alerta javascript, como fazemos???

Existe uma função para tratar alertas javascript, a função switch_to_alert irá permitir que manipule eles sem problemas.

>> alerta = firefox.switch_to_alert()
>> alerta.  # Tab para autocomplete
alerta.accept     alerta.dismiss    alerta.driver     alerta.send_keys  alerta.text

# Se for um `confirm`, você pode aceitar ou cancelar.
>> alerta.accept()  # Aceitar, ou clicar em "OK"
>> alerta.dismiss() # Cancelar, ou clicar em "Cancel"/"Cancelar"

# Se for uma caixa de texto e você quiser digitar algo
>> alerta.send_keys('Digitar esse texto')

# Se você quiser visualizar o texto que está presente no alerta
>> alerta.text
u'Texto do alerta'

Trabalhando com frames

Não existe muita diferença entre manipulação de frames e janelas, o princípio é o mesmo.

• Encontrar o elemento (metódo find_element)
• Mudar para ele (metódo switch_to_frame)
• Realizar as ações

# Vamos tentar localizar o botão que está dentro do frame e será gerado uma Exception
>> firefox.find_element_by_id('<botao_id>')
NoSuchElementException: Message: u'Unable to locate element: {"method":"id","selector":"<botao_id>"}'

# Precisamos antes encontrar o frame
>> frame = firefox.find_element_by_id('buttons')

# Vamos alterar para ele
>> firefox.switch_to_frame(frame)

# Agora podemos encontrar o elemento
>> botao = firefox.find_element_by_id('<botao_id>')

# E podemos manipulá-lo :)
>> botao.click()

E se eu quiser esperar?!

Quando você tenta localizar um elemento, o Selenium irá consultar a página e se não encontrar será gerado uma exception de imediato.

Mas e se o elemento demorar um pouco para aparecer, pode ser que ele faça parte de uma animação, um consulta ajax ou qualquer coisa do tipo.

Então precisamos dizer ao Selenium para esperar.

Existe uma classe chamada WebDriverWait que pode ser utilizada para para esperar por determinadas ações.

Hoje será apresentado o básico sobre ela e voltaremos a falar sobre na Parte 4 deste tutorial.

Veja o exemplo abaixo:

>> elemento = firefox.find_element_by_id('<elemento_id>')

Nesse caso se o elemento não existe, será gerado uma exception.

Mas e se soubermos que ele pode demorar um tempo para aparecer?

# Importar a classe WebDriverWait
>> from selenium.webdriver.support.ui import WebDriverWait

def esperar_pelo_elemento(firefox):
  return firefox.find_element_by_id('<elemento_id>')

>> elemento = WebDriverWait(firefox, 5).until(esperar_pelo_elemento)

O que fizemos? Nós pedimos para o firefox esperar por 5 segundos até que o resultado da função esperar_pelo_elemento seja True. Caso passe esse tempo e ele não encontre o elemento, então será gerada uma exception.

Essa foi nossa introdução ao WebDriverWait, basicamente você precisa passar uma função que aceite como parâmetro a instância do navegador e lá executar o código para encontrar o elemento.

Por hoje é só! Nos vemos na próxima, espero que tenha aprendido algo hoje :)