145 lines
4.8 KiB
Python
145 lines
4.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""Main module."""
|
|
|
|
import logging
|
|
import os
|
|
import time
|
|
import typing as T
|
|
|
|
os.environ['PATH'] = os.environ['PATH'] + \
|
|
':' + os.path.join(os.path.abspath(os.path.curdir), 'bin')
|
|
|
|
from selenium import webdriver as wd
|
|
from selenium.common.exceptions import WebDriverException, NoSuchElementException
|
|
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(levelname)s: [%(name)s] -> %(message)s'
|
|
)
|
|
|
|
|
|
def safely(f: T.Callable) -> T.Callable:
|
|
def _protection(self, *args, **kwargs):
|
|
try:
|
|
f(self, *args, **kwargs)
|
|
except WebDriverException as e:
|
|
self.logger.error("Something went wrong: %s", e)
|
|
|
|
return _protection
|
|
|
|
|
|
class Operator(wd.Firefox):
|
|
def __init__(
|
|
self,
|
|
base_uri: str,
|
|
name: str = None,
|
|
proxy: T.Optional[T.Tuple[str, int]] = None,
|
|
headless: bool = True,
|
|
debug: bool = False,
|
|
*args, **kwargs) -> None:
|
|
"""
|
|
Adds some configuration to Firefox.
|
|
"""
|
|
self.profile = wd.FirefoxProfile()
|
|
# Do not send telemetry
|
|
self.profile.set_preference('datareporting.policy.dataSubmissionEnabled', False)
|
|
self.profile.set_preference('datareporting.healthreport.service.enabled', False)
|
|
self.profile.set_preference('datareporting.healthreport.uploadEnabled', False)
|
|
|
|
self.opts = wd.firefox.options.Options()
|
|
self.opts.headless = headless
|
|
|
|
self.debug = debug
|
|
self.base_uri = base_uri
|
|
|
|
if proxy:
|
|
self.profile.set_preference('network.proxy.type', 1)
|
|
self.profile.set_preference('network.proxy.http', proxy[0])
|
|
self.profile.set_preference('network.proxy.http_port', proxy[1])
|
|
self.profile.set_preference('network.proxy.ssl', proxy[0])
|
|
self.profile.set_preference('network.proxy.ssl_port', proxy[1])
|
|
|
|
super().__init__(firefox_profile=self.profile, options=self.opts, *args, **kwargs)
|
|
|
|
self.z_name = name if name is not None else __name__
|
|
self.logger = logging.getLogger("{}.{}".format(__name__, self.name))
|
|
if debug:
|
|
self.logger.setLevel(logging.DEBUG)
|
|
self.logger.debug("Debug level")
|
|
self._logged_in = False
|
|
self._checked_in = False
|
|
|
|
@safely
|
|
def do_login(self, user: str, password: str) -> None:
|
|
"""
|
|
Do the login and proceed.
|
|
"""
|
|
# Retrieve login page
|
|
#self.get('{}/jsp/login.jsp'.format(self.base_uri))
|
|
self.get(self.base_uri)
|
|
self.logger.debug("After login get: %s", self.current_url)
|
|
time.sleep(3)
|
|
# Username
|
|
user_form = self.find_element_by_name('m_cUserName')
|
|
# Password
|
|
pass_form = self.find_element_by_name('m_cPassword')
|
|
# Login button
|
|
login_butt = self.find_element_by_xpath('//input[contains(@id, "_Accedi")]')
|
|
# Compile and submit
|
|
user_form.send_keys(user)
|
|
pass_form.send_keys(password)
|
|
do_it = True
|
|
if self.debug:
|
|
_do_it = input("Really do the login? [y/n]").lower()
|
|
do_it = True if _do_it == "y" else False
|
|
if do_it:
|
|
self.logger.debug("Before clicking: %s", self.current_url)
|
|
login_butt.submit()
|
|
self.logger.debug("After clicking: %s", self.current_url)
|
|
self.logger.debug("Login result: %s", self.title)
|
|
if 'Routine window' in self.title:
|
|
self.refresh()
|
|
try:
|
|
self.find_element_by_xpath('//input[contains(@value, "Accedi")]')
|
|
# TODO: reckon a proper timeout mechanism
|
|
# based on cookie expire time
|
|
self._logged_in = True
|
|
self.logger.info("Login success for user: %s", user)
|
|
except NoSuchElementException as e:
|
|
self.logger.error("Login failed: %s", e)
|
|
|
|
@safely
|
|
def check_in(self, force: bool=False) -> None:
|
|
"""
|
|
Click the check in button.
|
|
"""
|
|
# TODO: Check if the page displays you're already
|
|
# checked in.
|
|
if self._checked_in:
|
|
self.logger.warn("Already logged in!")
|
|
if not force:
|
|
return
|
|
# Click the check in button and change
|
|
# self._checked_in state in case of success
|
|
pass
|
|
|
|
@safely
|
|
def check_out(self, force: bool=False) -> None:
|
|
"""
|
|
Click the check out button.
|
|
"""
|
|
# TODO: Check if the page displays you're already
|
|
# checked out.
|
|
if not self._checked_in:
|
|
self.logger.warn("Not yet logged in!")
|
|
if not force:
|
|
return
|
|
# Click the check in button and change
|
|
# self._checked_in state in case of success
|
|
pass
|
|
|
|
def __del__(self):
|
|
self.quit()
|