#!/usr/bin/env python3

from selenium import webdriver

browser = webdriver.Firefox()
TRP_CONSTRUCT = "https://DOMAIN.translate.goog/PATH/?_x_tr_sl=auto&_x_tr_tl=en&_x_tr_hl=en-US&_x_tr_pto=wapp"

url = input("Enter URL you wish to translate: ")
remote_domain = url.split('://')[1].split('/')[0]
remote_path = '/'.join(url.split('/')[3:])
rDUri = remote_domain.replace('.', '-')
if '?' in remote_path:
    remote_path = remote_path.split('?')[0]
    remote_URL_query = ''.join(url.split('?')[1:])
    rUrl = TRP_CONSTRUCT.replace('DOMAIN', rDUri).replace('PATH', remote_path) + '&' + remote_URL_query
else:
    rUrl = TRP_CONSTRUCT.replace('DOMAIN', rDUri).replace('PATH', remote_path)

print("Executing :" + rUrl + " \nwithin DOM.")
browser.get(rUrl)
browser.execute_script('''document.querySelector('iframe').style.display = "none";document.querySelector('iframe').remove()''')