#!/usr/bin/python3

import os
import traceback
import logging
from time import sleep
from dotenv import load_dotenv
from seleniumwire import webdriver
from seleniumwire.webdriver import FirefoxOptions

try:
	os.chdir(os.path.dirname(os.path.abspath(__file__)))
	load_dotenv()

	opts = FirefoxOptions()
	opts.add_argument("--headless")
	browser = webdriver.Firefox(options=opts,executable_path="/usr/local/bin/geckodriver")
	browser.set_window_size(3000, 1120)

	def interceptor(request):
    	request.headers['Authorization'] = 'Basic '+os.environ["ES_TOKEN"]
	browser.request_interceptor = interceptor

	for duration in ["24h", "7d", "1M", "1y"]:
		browser.get("https://mon.gnous.fr/app/dashboards#/view/380099d0-4944-11ec-81a9-95d7ffa23468?embed=true&_g=(filters%3A!()%2CrefreshInterval%3A(pause%3A!t%2Cvalue%3A0)%2Ctime%3A(from%3Anow-"+duration+"%2Cto%3Anow))&hide-filter-bar=true")
		sleep(30)
		if duration == "24h":
			sleep(15)
		browser.get_screenshot_as_file("html/ressources/stats/mm"+duration+".png")

	for duration in ["24h", "7d", "1M", "1y"]:
		browser.get("https://mon.gnous.fr/app/dashboards#/view/ef70de60-4947-11ec-81a9-95d7ffa23468?embed=true&_g=(filters%3A!()%2CrefreshInterval%3A(pause%3A!t%2Cvalue%3A0)%2Ctime%3A(from%3Anow-"+duration+"%2Cto%3Anow))&hide-filter-bar=true")
		sleep(30)
		browser.get_screenshot_as_file("html/ressources/stats/nc"+duration+".png")
except Exception as e:
	logging.error(traceback.format_exc())