106 lines
3.5 KiB
Python
106 lines
3.5 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import time
|
|
import logging
|
|
|
|
from selenium import webdriver
|
|
from pyvirtualdisplay import Display
|
|
|
|
logger = logging.getLogger(__file__)
|
|
|
|
class DomainNotFoundError(Exception):
|
|
pass
|
|
|
|
class HpUpdater():
|
|
|
|
def __init__(self):
|
|
logger.debug('Initiating HpUpdater')
|
|
|
|
def _login(self, driver):
|
|
logger.debug('login called')
|
|
|
|
driver.get('https://admin.hostpoint.ch/customer/Auth/Login')
|
|
driver.find_element_by_name('username').send_keys(os.environ.get('HpLogin'))
|
|
driver.find_element_by_name('password').send_keys(os.environ.get('HpPass'))
|
|
driver.find_element_by_name('login').click()
|
|
|
|
def _updateARecord(self, driver, ip):
|
|
logger.debug('updateARecord called')
|
|
|
|
driver.get('https://admin.hostpoint.ch/customer/Domains/Overview')
|
|
|
|
domainTableRows = driver.find_elements_by_class_name('b-table__body-row')
|
|
|
|
for row in domainTableRows:
|
|
if os.environ.get('Domain') == row.find_element_by_class_name('domain-name-value').get_attribute('value'):
|
|
row.find_element_by_name('edit_dns').click()
|
|
logger.debug('domain found')
|
|
break
|
|
else:
|
|
raise DomainNotFoundError(f'Domain {os.environ.get("Domain")} can not be found in domain list.')
|
|
|
|
driver.implicitly_wait(10)
|
|
|
|
tableRows = driver.find_element_by_id('dns-record-list').find_elements_by_class_name('b-record')
|
|
|
|
# The last two entries are my A records
|
|
for e in tableRows[-2:]:
|
|
try:
|
|
e.find_element_by_name('edit').click()
|
|
|
|
# For some reason with each iteration it return one more such element
|
|
editForm = driver.find_elements_by_class_name('dns-record-list-edit')
|
|
editForm[-1].find_element_by_name('ip_address').clear()
|
|
editForm[-1].find_element_by_name('ip_address').send_keys(ip)
|
|
|
|
editForm[-1].find_element_by_name('apply').click()
|
|
|
|
except Exception as e:
|
|
logger.error(f'Exception during update: {str(e)}')
|
|
|
|
logger.info(f'Changed A records ip to {ip}')
|
|
|
|
try:
|
|
|
|
e = driver.find_element_by_name('execute_now')
|
|
driver.execute_script("arguments[0].click();", e)
|
|
|
|
# Give the site some time to save
|
|
time.sleep(10)
|
|
|
|
logger.info('Successfully saved changes')
|
|
|
|
except Exception as e:
|
|
logger.error(f'Exception during save: {str(e)}')
|
|
|
|
def update(self, ip):
|
|
logger.info('Start Hostpoint Record update')
|
|
|
|
# Thanks to https://github.com/dimmg/dockselpy
|
|
# Display is only used for firefox to run properly
|
|
display = Display(visible=0, size=(800, 600))
|
|
display.start()
|
|
|
|
logging.info('Initialized virtual display..')
|
|
|
|
firefox_profile = webdriver.FirefoxProfile()
|
|
firefox_profile.set_preference('browser.download.folderList', 2)
|
|
firefox_profile.set_preference('browser.download.manager.showWhenStarting', False)
|
|
firefox_profile.set_preference('browser.download.dir', os.getcwd())
|
|
firefox_profile.set_preference('browser.helperApps.neverAsk.saveToDisk', 'text/csv')
|
|
|
|
logging.info('Prepared firefox profile..')
|
|
|
|
driver = webdriver.Firefox(firefox_profile=firefox_profile)
|
|
logging.info('Initialized firefox browser..')
|
|
|
|
self._login(driver)
|
|
try:
|
|
self._updateARecord(driver, ip)
|
|
except DomainNotFoundError as e:
|
|
logging.error(e)
|
|
|
|
driver.quit()
|
|
display.stop()
|