plane-notify/defSS.py

import json
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager
import time
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By

def get_adsbx_screenshot(file_path, url_params, enable_labels=False, enable_track_labels=False, overrides={}):
    chrome_options = webdriver.ChromeOptions()
    chrome_options.headless = True
    chrome_options.add_argument('window-size=800,800')
    chrome_options.add_argument('ignore-certificate-errors')
    #Plane images issue loading when in headless setting agent fixes. 
    chrome_options.add_argument("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.54 Safari/537.36")
    import os
    import platform
    if platform.system() == "Linux" and os.geteuid()==0:
        chrome_options.add_argument('--no-sandbox') # required when running as root user. otherwise you would get no sandbox errors.
    browser = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options)
    url = f"https://globe.adsbexchange.com/?{url_params}"
    print(url)
    browser.set_page_load_timeout(80)
    browser.get(url)
    WebDriverWait(browser, 40).until(lambda d: d.execute_script("return jQuery.active == 0"))
    remove_id_elements = ["show_trace", "credits", 'infoblock_close', 'selected_photo_link', "history_collapse"]
    for element in remove_id_elements:
        try:
            element = browser.find_element_by_id(element)
            browser.execute_script("""var element = arguments[0];    element.parentNode.removeChild(element); """, element)
        except:
            print("issue removing", element, "from map")
    #Remove watermark on data
    try:
        browser.execute_script("document.getElementById('selected_infoblock').className = 'none';")
    except:
        print("Couldn't remove watermark from map")
    #Disable slidebar
    try:
        browser.execute_script("$('#infoblock-container').css('overflow', 'hidden');")
    except:
        print("Couldn't disable sidebar on map")
    #Remove share
    # try:
    #     element = browser.find_element_by_xpath("//*[contains(text(), 'Copy Link')]")
    #     browser.execute_script("""var element = arguments[0];    element.parentNode.removeChild(element); """, element)
    # except Exception as e:
    #     print("Couldn't remove share button from map", e)
    #browser.execute_script("toggleFollow()")
    if enable_labels:
        browser.find_element_by_tag_name('body').send_keys('l')
    if enable_track_labels:
        browser.find_element_by_tag_name('body').send_keys('k')
    from selenium.webdriver.support import expected_conditions as EC
    time.sleep(15)

    if 'reg' in overrides.keys():
        element = browser.find_element_by_id("selected_registration")
        browser.execute_script(f"arguments[0].innerText = '* {overrides['reg']}'", element)
        reg = overrides['reg']
    else:
        try: 
            reg = browser.find_element_by_id("selected_registration").get_attribute("innerHTML")
            print(reg)
        except Exception as e:
            print("Couldn't find reg in tar1090", e)
            reg = None 
    if reg is not None:
        try:
            try:
                myElem = WebDriverWait(browser, 150).until(EC.presence_of_element_located((By.ID, 'silhouette')))
                photo_box = browser.find_element_by_id("silhouette")
            except:
                photo_box = browser.find_element_by_id("airplanePhoto")
            finally:
                import requests, json
                photo_list = json.loads(requests.get("https://raw.githubusercontent.com/Jxck-S/aircraft-photos/main/photo-list.json").text)
                if reg in photo_list.keys():
                    browser.execute_script("arguments[0].id = 'airplanePhoto';", photo_box) 
                    browser.execute_script(f"arguments[0].src = 'https://raw.githubusercontent.com/Jxck-S/aircraft-photos/main/images/{reg}.jpg';", photo_box) 
                    copyright = browser.find_element_by_id("copyrightInfo")
                    browser.execute_script("arguments[0].id = 'copyrightInfoFreeze';", copyright) 
                    browser.execute_script("$('#copyrightInfoFreeze').css('font-size', '12px');")
                    #browser.execute_script("""var element = arguments[0]; 
                    #element.parentNode.removeChild(element.firstChild);""", copyright)
                    browser.execute_script(f"arguments[0].appendChild(document.createTextNode('Image © {photo_list[reg]['photographer']}'))", copyright)

        except Exception as e:
            print("Error on changing photo", e)
    if 'type' in overrides.keys():
        element = browser.find_element_by_id("selected_icaotype")
        browser.execute_script(f"arguments[0].innerText = '* {overrides['type']}'", element)
    if 'typelong' in overrides.keys():
        element = browser.find_element_by_id("selected_typelong")
        browser.execute_script(f"arguments[0].innerText = '* {overrides['typelong']}'", element)
    time.sleep(5)
    browser.save_screenshot(file_path)
    browser.quit()
def generate_adsbx_screenshot_time_params(timestamp):
    from datetime import datetime
    from datetime import timedelta
    timestamp_dt = datetime.utcfromtimestamp(timestamp)
    print(timestamp_dt)
    start_time = timestamp_dt - timedelta(minutes=1)
    time_params = "&showTrace=" + timestamp_dt.strftime("%Y-%m-%d")  + "&startTime=" + start_time.strftime("%H:%M:%S") + "&endTime=" + timestamp_dt.strftime("%H:%M:%S")
    return time_params