loading clearance and user agent from env automatically

Author: Shinga13

src/datafunctions.py

@@ -45,6 +45,7 @@ def finish_backend_init():
 
     enter_splash(self)
     load_build_file(self, self.config['autosave_filename'], update_ui=False)
+    self.downloader.default_session_from_env()
     exec_in_thread(
             self, populate_cache, self, finished=finish_backend_init,
             update_splash=lambda new_text: splash_text(self, new_text))

src/downloader.py

@@ -1,4 +1,5 @@
 from json import loads as json__loads, JSONDecodeError
+from os import getenv as os__getenv
 from pathlib import Path
 from requests import Session
 from requests.exceptions import Timeout
@@ -54,6 +55,21 @@ def configure_default_session(self, cookies: dict[str, str] = {}, headers: dict[
         self._session.headers.clear()
         self._session.headers.update(self._headers)
 
+    def default_session_from_env(self):
+        """
+        Configures default session from env variables.
+        """
+        cf_clearance = os__getenv('SETS_CF_CLEARANCE')
+        if cf_clearance is not None:
+            self._cookies['cf_clearance'] = cf_clearance
+        user_agent = os__getenv('SETS_USER_AGENT')
+        if user_agent is not None:
+            self._headers['User-Agent'] = user_agent
+        self._session.cookies.clear()
+        self._session.cookies.update(self._cookies)
+        self._session.headers.clear()
+        self._session.headers.update(self._headers)
+
     def fetch_json(self, url: str) -> dict | list | None:
         """
         Fetches json from url and returns parsed object. Returns `None` if retrieveing the json