main_website/print_stats.py

#!/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())