bitbucket webdriver

bucketdrive.py

import os, time, subprocess, random

from functools import wraps

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.select import Select

from pyvirtualdisplay import Display

"""
https://seleniumhq.github.io/selenium/docs/api/py/webdriver/selenium.webdriver.common.action_chains.html
"""


class WebdriverChauffuer(object):

    def __init__(self, username=None, password=None, start_url=None):
        self.username = username
        self.password = password
        self.start_url = start_url

    def quit_driver(self):
        self.driver.quit()

    def restart_driver(self):
        self.driver.quit()
        self.start_driver()

    def sleep_random(self, start, end):
        time.sleep(self.get_random_time(start, end))

    def get_random_time(self, start, end):
        diff = end - start
        random_num = int((random.SystemRandom(random.seed()).random()) * diff)
        actual_num = (random_num + start)
        return self.convert_num_to_secs(actual_num)

    def convert_num_to_secs(self, num):
        """takes a number like 89 and converts it to 0.89 seconds"""
        return (float(num)/float(100))

    def locate_element(self, search_text, xpaths=None):
        if not xpaths:
            xpaths = ["//button[normalize-space(text())='{text}']", "//input[@value='{text}']",
                      "//a[child::span[normalize-space(text())='{text}']]", "//a[normalize-space(text())='{text}']"]
        try:
            return self.driver.find_element_by_id(search_text)
        except:
            try:
                return self.driver.find_element_by_name(search_text)
            except:
                try:
                    return self.driver.find_element_by_class_name(search_text)
                except:
                    for path in xpaths:
                        try:
                            return self.driver.find_element_by_xpath(path.format(text=search_text))
                        except:
                            pass
        return None

    def get(self, url=None):
        self.driver.get(url)

    @property
    def source_code(self):
        return self.driver.page_source or None
        
    def get_active_element(self):
        return self.driver.switch_to_active_element()

    def activate_hidden_element(self, search_text=None):
        action = webdriver.ActionChains(self.driver)
        element = self.locate_element(search_text)
        action.move_to_element(element).click().perform()
        return element

    def access_link(self, search_text=None):
        element = self.locate_element(search_text)
        link  = element.get_attribute('href')
        self.get(link)

    def submit_form(self, search_text=None):
        form = self.locate_element(search_text)
        form.submit()

    def click_button(self,search_text=None):
        element = self.locate_element(search_text)
        element.click()

    def find_box_and_fill(self, search_text=None, value=None):
        box = self.locate_element(search_text)
        box.send_keys(value)

    def find_box_and_fill_as_person(self, search_text=None, value=None):
        box = self.locate_element(search_text)
        for character in value:
            box.send_keys(character)
            self.sleep_random(7, 15)

    def pick_select_box(self, search_text=None, value=None):
        element = self.locate_element(search_text)
        element.select_by_value(value)

    def pick_expiration_year_for_card(self, expiration_year=None, html_name=None):
        """ensure your constant for card year is 4 digits like 2017"""
        abbreviated_year = EXPIRATION_YEAR[2:]
        try:
            self.pick_select_box(html_name=html_name, value=expiration_year)
        except:
            self.pick_select_box(html_name=html_name, value=abbreviated_year)


class ChromeDriver(WebdriverChauffuer):

    def __init__(self, username=None, password=None, start_url=None, driver=None):
        super(ChromeDriver, self).__init__(username=username, password=password, start_url=start_url)
        if driver:
            self.driver = driver
        else:
            self.start_driver()

    def start_driver(self):
        self.driver = webdriver.Chrome()


class FirefoxDriver(WebdriverChauffuer):

    def __init__(self, username=None, password=None, start_url=None, driver=None):
        super(FirefoxDriver, self).__init__(username=username, password=password, start_url=start_url)
        if driver:
            self.driver = driver
        else:
            self.start_driver()

    def start_driver(self):
        self.driver = webdriver.Firefox()
        
        
class BitbucketDriver(FirefoxDriver):

    def __init__(self, username="codyc54321", password="mypw"):
        super(BitbucketDriver, self).__init__(username=username, password=password, start_url="https://bitbucket.org/account/signin/")
        self.main_url  = "https://bitbucket.org/"

    def login(self):
        self.get(self.start_url)
        time.sleep(2)
        self.find_box_and_fill(search_text="username", value=self.username)
        self.find_box_and_fill(search_text="password", value=self.password)
        self.submit_form(search_text="Log in") # causes "Error fetching XRDS document: <urlopen error no host given>" in bitbucket browser
        self.click_button(search_text="Log in")  # causes ElementNotVisibleException
        self.activate_hidden_element(search_text="Log in") # causes ElementNotVisibleException