feat(yt-dl): automatically gather youtube cookies

Contents/Code/__init__.py

@@ -2,7 +2,9 @@
 
 # standard imports
 import inspect
+import os
 import re
+import sys
 
 # plex debugging
 try:
@@ -40,6 +42,19 @@
 except Exception as e:
     Log.Exception("Failed to bypass RestrictedPython: {}".format(e))
 
+# change priority of Contents/Libraries/Shared to be first
+# this is required to import versions of libraries that we specify in our requirements.txt file
+# instead of what Plex provides
+try:
+    # sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'Libraries', 'Shared'))
+    import selenium
+    getattr(selenium, "__file__")
+    # todo - determine the best way to move our libraries to the top of the path
+    for path in sys.path:
+        Log.Debug("sys.path: {}".format(path))
+except Exception as e:
+    Log.Exception("Failed to add Contents/Libraries/Shared to sys.path: {}".format(e))
+
 # local imports
 from default_prefs import default_prefs
 from constants import contributes_to, version

Contents/Code/general_helper.py

@@ -16,6 +16,10 @@
     from plexhints.log_kit import Log  # log kit
     from plexhints.prefs_kit import Prefs  # prefs kit
 
+# imports from Libraries\Shared
+from plexapi.base import PlexPartialObject
+import requests
+from typing import Optional
 
 # local imports
 from constants import metadata_base_directory, metadata_type_map, themerr_data_directory
@@ -27,15 +31,15 @@
 
 
 def get_media_upload_path(item, media_type):
-    # type: (any, str) -> str
+    # type: (PlexPartialObject, str) -> str
     """
     Get the path to the theme upload directory.
 
     Get the hashed path of the theme upload directory for the item specified by the ``item``.
 
     Parameters
     ----------
-    item : any
+    item : PlexPartialObject
         The item to get the theme upload path for.
     media_type : str
         The media type to get the theme upload path for. Must be one of 'art', 'posters', or 'themes'.
@@ -75,15 +79,15 @@ def get_media_upload_path(item, media_type):
 
 
 def get_themerr_json_path(item):
-    # type: (any) -> str
+    # type: (PlexPartialObject) -> str
     """
     Get the path to the Themerr data file.
 
     Get the path to the Themerr data file for the item specified by the ``item``.
 
     Parameters
     ----------
-    item : any
+    item : PlexPartialObject
         The item to get the Themerr data file path for.
 
     Returns
@@ -102,7 +106,7 @@ def get_themerr_json_path(item):
 
 
 def get_themerr_json_data(item):
-    # type: (any) -> dict
+    # type: (PlexPartialObject) -> dict
     """
     Get the Themerr data for the specified item.
 
@@ -111,7 +115,7 @@ def get_themerr_json_data(item):
 
     Parameters
     ----------
-    item : any
+    item : PlexPartialObject
         The item to get the Themerr data for.
 
     Returns
@@ -153,16 +157,102 @@ def get_themerr_settings_hash():
     return settings_hash
 
 
+def get_user_country_code(ip_address=None):
+    # type: (Optional[str]) -> str
+    """
+    Get the country code for the user with the given IP address.
+
+    Parameters
+    ----------
+    ip_address : Optional[str]
+        The IP address of the user.
+
+    Returns
+    -------
+    str
+        The `ALPHA-2 <https://www.iban.com/country-codes>`__ country code for the user with the given IP address.
+
+    Examples
+    --------
+    >>> get_user_country_code()
+    'US'
+    """
+    api_url = 'https://ipinfo.io/json'
+    if ip_address:
+        api_url = 'https://ipinfo.io/{}/json'.format(ip_address)
+
+    try:
+        response = requests.get(api_url)
+        data = response.json()
+        return data.get('country').encode('utf-8')
+    except requests.RequestException as e:
+        Log.Error("Could not determine user country: {}".format(e))
+
+
+def is_user_in_eu(ip_address=None):
+    # type: (Optional[str]) -> bool
+    """
+    Check if the user with the given IP address is in the European Union.
+
+    Parameters
+    ----------
+    ip_address : Optional[str]
+        The IP address of the user.
+
+    Returns
+    -------
+    bool
+        True if the user with the given IP address is in the European Union, False otherwise.
+
+    Examples
+    --------
+    >>> is_user_in_eu()
+    False
+    """
+    eu_countries = [
+        'AT',  # Austria
+        'BE',  # Belgium
+        'BG',  # Bulgaria
+        'CY',  # Cyprus
+        'CZ',  # Czech Republic
+        'DE',  # Germany
+        'DK',  # Denmark
+        'EE',  # Estonia
+        'ES',  # Spain
+        'FI',  # Finland
+        'FR',  # France
+        'GR',  # Greece
+        'HR',  # Croatia
+        'HU',  # Hungary
+        'IE',  # Ireland
+        'IT',  # Italy
+        'LT',  # Lithuania
+        'LU',  # Luxembourg
+        'LV',  # Latvia
+        'MT',  # Malta
+        'NL',  # Netherlands
+        'PL',  # Poland
+        'PT',  # Portugal
+        'RO',  # Romania
+        'SE',  # Sweden
+        'SI',  # Slovenia
+        'SK',  # Slovakia
+    ]
+
+    country_code = get_user_country_code(ip_address=ip_address)
+    return country_code in eu_countries
+
+
 def remove_uploaded_media(item, media_type):
-    # type: (any, str) -> None
+    # type: (PlexPartialObject, str) -> None
     """
     Remove themes for the specified item.
 
     Deletes the themes upload directory for the item specified by the ``item``.
 
     Parameters
     ----------
-    item : any
+    item : PlexPartialObject
         The item to remove the themes from.
     media_type : str
         The media type to remove the themes from. Must be one of 'art', 'posters', or 'themes'.
@@ -193,7 +283,7 @@ def remove_uploaded_media_error_handler(func, path, exc_info):
     ----------
     func : any
         The function that caused the error.
-    path : str
+    path : any
         The path that caused the error.
     exc_info : any
         The exception information.
@@ -202,15 +292,15 @@ def remove_uploaded_media_error_handler(func, path, exc_info):
 
 
 def update_themerr_data_file(item, new_themerr_data):
-    # type: (any, dict) -> None
+    # type: (PlexPartialObject, dict) -> None
     """
     Update the Themerr data file for the specified item.
 
     This updates the themerr data file after uploading media to the Plex server.
 
     Parameters
     ----------
-    item : any
+    item : PlexPartialObject
         The item to update the Themerr data file for.
     new_themerr_data : dict
         The Themerr data to update the Themerr data file with.

Contents/Code/platform_helper.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+
+# standard imports
+import os
+import platform
+
+# plex debugging
+try:
+    import plexhints  # noqa: F401
+except ImportError:
+    pass
+else:  # the code is running outside of Plex
+    from plexhints.log_kit import Log  # log kit
+
+
+def get_os_architecture():
+    # Getting architecture using platform module
+    machine = platform.machine()
+
+    # For more detailed check, especially for Windows OS
+    if os.name == 'nt':
+        # Possible values: '32bit', '64bit'
+        # This will tell us if the OS is 64-bit or 32-bit
+        architecture = platform.architecture()
+
+        if architecture[0] == '64bit':
+            return 'x86_64'
+        elif architecture[0] == '32bit':
+            return 'x86'
+        else:
+            return 'Unknown architecture'
+    else:
+        # For Unix/Linux systems, we can rely more on platform.machine()
+        if machine in ['x86_64', 'AMD64']:
+            return 'x86_64'
+        elif machine in ['i386', 'i686', 'x86']:
+            return 'x86'
+        elif machine in ['aarch64', 'arm64']:
+            return 'aarch64'
+        else:
+            return 'Unknown architecture'
+
+
+# constants
+architecture = get_os_architecture()
+os_system = platform.system().lower()

Contents/Code/scheduled_tasks.py

@@ -21,6 +21,7 @@
 # local imports
 from constants import plugin_identifier
 from plex_api_helper import scheduled_update
+from selenium_helper import install_driver
 from webapp import cache_data
 
 # setup logging for schedule
@@ -109,6 +110,10 @@ def setup_scheduling():
     plex_api_helper.scheduled_update : Scheduled function to update the themes.
     """
     if Prefs['bool_auto_update_items']:
+        schedule.every(max(1, int(Prefs['int_update_browser_driver_interval']))).hours.do(
+            job_func=run_threaded,
+            target=install_driver
+        )
         schedule.every(max(15, int(Prefs['int_update_themes_interval']))).minutes.do(
             job_func=run_threaded,
             target=scheduled_update