Produits Entr'ouvert » Mandaye

#!/usr/bin/env python

import json

import sys

# import time

from selenium.common import exceptions

from selenium.webdriver import Firefox, FirefoxProfile

from selenium.webdriver.common.by import By

from selenium.webdriver.firefox.options import Options

from selenium.webdriver.support import expected_conditions as EC

from selenium.webdriver.support.ui import WebDriverWait

firefox_binary_path = '/opt/firefox/firefox'

gecko_binary_path = '/opt/firefox/geckodriver'  # http://bit.ly/2iGGDeR

def get_driver():

    options = Options()

    # options.set_headless()

    profile = FirefoxProfile()

    profile.accept_untrusted_certs = True

    driver = Firefox(executable_path=gecko_binary_path, firefox_binary=firefox_binary_path,

                     firefox_options=options, firefox_profile=profile)

    return driver

class HeadlessBrowser(object):

    def __init__(self, url):

        self.url = url

    def __enter__(self):

        self.driver = get_driver()

        self.driver.get(self.url)

        return self.driver

    def __exit__(self, *args):

        self.driver.close()

class App(object):

    def __init__(self, data):

        self.data = data

    def check_authentication(self, browser):

        raise NotImplementedError

    def before_authentication(self, browser):

        return browser

    def authenticate(self):

        status = {}

        with HeadlessBrowser(self.data['address']) as browser:

            # wait couple seconds before processing

            try:

                WebDriverWait(browser, 1).until(

                    EC.presence_of_element_located((By.ID, self.data['locators'].keys()[0].strip('#'))))

            except exceptions.TimeoutException:

                return {'result': 'timeout', 'reason': 'element no found'}

            # run prior attempting an authentication

            self.before_authentication(browser)

            for key, value in self.data['locators'].iteritems():

                element = browser.find_element_by_css_selector(key)

                element.send_keys(value)

            element.submit()

            # check authentication

            auth_checked = self.check_authentication(browser)

            status['result'] = 'ok' if auth_checked else 'failure'

            status['url'] = browser.current_url

            status['cookies'] = browser.get_cookies()

            # time.sleep(10)

        return status

class Archimed(App):

    def before_authentication(self, browser):

        browser.find_element_by_css_selector('#dropMenu1').click()

        return browser

    def check_authentication(self, browser):

        try:

            WebDriverWait(browser, 1).until(

                EC.presence_of_element_located((By.CLASS_NAME, 'account_logoff')))

        except exceptions.TimeoutException:

            return False

        return bool(browser.find_elements_by_css_selector('a.account_logoff'))

class Teamnet(App):

    def check_authentication(self, browser):

        try:

            WebDriverWait(browser, 1).until(

                EC.presence_of_element_located((By.CLASS_NAME, 'infoUtilisateur')))

        except exceptions.TimeoutException:

            return False

        return bool(browser.find_elements_by_css_selector(".infoUtilisateur"))

class Redmine(App):

    def check_authentication(self, browser):

        try:

            WebDriverWait(browser, 1).until(

                EC.presence_of_element_located((By.ID, 'loggedas')))

        except exceptions.TimeoutException:

            return False

        return bool(browser.find_elements_by_css_selector("#loggedas"))

if __name__ == "__main__":

    """{

        'address': 'https://dev.entrouvert.org/login',

        'locators': {

            '#username': 'whatever',

            '#password': 'whatever'

        }

    }

    """

    if len(sys.argv) < 2:

        print('python headless.py whatever.json')

    filename = sys.argv.pop()

    data = json.load(open(filename))

    if 'entrouvert' in data['address']:

        app = Redmine(data)

    elif 'montpellier' in data['address']:

        app = Archimed(data)

    else:

        app = Teamnet(data)

    print(app.authenticate())