For Selenium routine actions

routine.py

# -*- coding: utf-8 -*-
__author__ = 'Mark Allan B. Meriales'

from selenium.common.exceptions import NoSuchElementException       


# Routine for finding, clicking, clearing and sending keys
# to an element
def click_clear_send_keys(elem, keys, **kwargs):
    if kwargs.get('click', False):
        elem.click()
    if kwargs.get('clear', False):
        elem.clear()
    elem.send_keys(keys)   
    return elem

def find_and_populate_element_by_name(driver, elem_name, keys, **kwargs):
    try:
        return click_clear_send_keys(driver.find_element_by_name(elem_name), keys, **kwargs)
    except NoSuchElementException:
        print 'Element not found'
        return None

def find_and_populate_element_by_id(driver, elem_name, keys, **kwargs):
    try:
        return click_clear_send_keys(driver.find_element_by_id(elem_name), keys, **kwargs)
    except NoSuchElementException:
        print 'Element not found'
        return None

def find_and_populate_element_by_xpath(driver, elem_name, keys, **kwargs):
    try:
        return click_clear_send_keys(driver.find_element_by_xpath(elem_name), keys, **kwargs)
    except NoSuchElementException:
        print 'Element not found'
        return None