Mirrored_Repos/plex-prerolls
1
0
Fork 0
mirror of https://github.com/nwithan8/plex-prerolls synced 2024-08-30 16:52:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

logging, code format change

Browse Source
This commit is contained in:
Brian Lindner 2022-10-08 21:07:14 -04:00
parent bf36f91dfb
commit b94b6f0360
No known key found for this signature in database
1 changed files with 49 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

97
plexutil.py
View File

@ -5,20 +5,22 @@ Raises:
FileNotFoundError: [description] FileNotFoundError: [description]
KeyError: [description] KeyError: [description]
""" """
import os
import sys
import logging import logging
import logging.config import logging.config
from typing import Optional, Dict, List import os
import sys
from configparser import ConfigParser from configparser import ConfigParser
from plexapi.server import PlexServer, CONFIG from typing import Dict, Optional
from plexapi.server import CONFIG # type: ignore
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
filename = os.path.basename(sys.argv[0]) filename = os.path.basename(sys.argv[0])
SCRIPT_NAME = os.path.splitext(filename)[0] SCRIPT_NAME = os.path.splitext(filename)[0]
def plex_config(config_file: Optional[str]='') -> Dict[str,str]:
def plex_config(config_file: Optional[str] = "") -> Dict[str, str]:
"""Return Plex Config paramaters for connection info {PLEX_URL, PLEX_TOKEN}\n """Return Plex Config paramaters for connection info {PLEX_URL, PLEX_TOKEN}\n
Attempts to use one of either:\n Attempts to use one of either:\n
* supplier path/to/config file (INI Format) * supplier path/to/config file (INI Format)
@ -36,58 +38,56 @@ def plex_config(config_file: Optional[str]='') -> Dict[str,str]:
dict: Dict of config params {PLEX_URL, PLEX_TOKEN} dict: Dict of config params {PLEX_URL, PLEX_TOKEN}
""" """
cfg = {} # type: Dict[str, str] cfg: dict[str, str] = {}
plex_url = '' plex_url = ""
plex_token = '' plex_token = ""
filename = '' filename = ""
use_local_config = False use_local_config = False
use_plexapi_config = False use_plexapi_config = False
# Look for a local Config.ini file, use settings if present # Look for a local Config.ini file, use settings if present
local_config = ConfigParser() local_config = ConfigParser()
if config_file == None or config_file == '': if config_file == None or config_file == "":
filename = 'config.ini' filename = "config.ini"
else: else:
filename = str(config_file) filename = str(config_file)
#try reading a local file # try reading a local file
local_config.read(filename) local_config.read(filename)
if len(local_config.sections()) > 0: #len(found_config) > 0: if len(local_config.sections()) > 0: # len(found_config) > 0:
# if local config.ini file found, try to use local first # if local config.ini file found, try to use local first
if local_config.has_section('auth'): if local_config.has_section("auth"):
try: try:
server = local_config['auth'] server = local_config["auth"]
plex_url = server['server_baseurl'] plex_url = server["server_baseurl"]
plex_token = server['server_token'] plex_token = server["server_token"]
if len(plex_url) > 1 and len(plex_token) > 1: if len(plex_url) > 1 and len(plex_token) > 1:
use_local_config = True use_local_config = True
except KeyError as e: except KeyError as e:
msg = 'Key Value not found {}' logger.error("Key Value not found", exc_info=e)
logger.error(msg, exc_info=e)
raise e raise e
else: else:
msg = '[auth] section not found in LOCAL config.ini file' msg = "[auth] section not found in LOCAL config.ini file"
logger.error(msg) logger.error(msg)
raise KeyError(msg) raise KeyError(msg)
if not use_local_config and len(CONFIG.sections()) > 0: if not use_local_config and len(CONFIG.sections()) > 0: # type: ignore
# use PlexAPI Default ~/.config/plexapi/config.ini OR from PLEXAPI_CONFIG_PATH # use PlexAPI Default ~/.config/plexapi/config.ini OR from PLEXAPI_CONFIG_PATH
# IF not manually set locally in local Config.ini above # IF not manually set locally in local Config.ini above
# See https://python-plexapi.readthedocs.io/en/latest/configuration.html # See https://python-plexapi.readthedocs.io/en/latest/configuration.html
if CONFIG.has_section('auth'): if CONFIG.has_section("auth"): # type: ignore
try: try:
server = CONFIG.data['auth'] server = CONFIG.data["auth"] # type: ignore
plex_url = server.get('server_baseurl') plex_url: str = server.get("server_baseurl") # type: ignore
plex_token = server.get('server_token') plex_token: str = server.get("server_token") # type: ignore
if len(plex_url) > 1 and len(plex_token) > 1: if len(plex_url) > 1 and len(plex_token) > 1:
use_plexapi_config = True use_plexapi_config = True
except KeyError as e: except KeyError as e:
msg = 'Key Value not found' logger.error("Key Value not found", exc_info=e)
logger.error(msg, exc_info=e)
raise e raise e
else: else:
msg = "[auth] section not found in PlexAPI MAIN config.ini file" msg = "[auth] section not found in PlexAPI MAIN config.ini file"
@ -95,15 +95,16 @@ def plex_config(config_file: Optional[str]='') -> Dict[str,str]:
raise KeyError(msg) raise KeyError(msg)
if not use_local_config and not use_plexapi_config: if not use_local_config and not use_plexapi_config:
msg = 'ConfigFile Error: No Plex config information found {server_baseurl, server_token}' msg = "ConfigFile Error: No Plex config information found [server_baseurl, server_token]"
logger.error(msg) logger.error(msg)
raise FileNotFoundError(msg) raise FileNotFoundError(msg)
cfg['PLEX_URL'] = plex_url cfg["PLEX_URL"] = plex_url
cfg['PLEX_TOKEN']= plex_token cfg["PLEX_TOKEN"] = plex_token
return cfg return cfg
def init_logger(log_config: str) -> None: def init_logger(log_config: str) -> None:
"""load and configure a program logger using a supplier logging configuration file \n """load and configure a program logger using a supplier logging configuration file \n
if possible the program will attempt to create log folders if not already existing if possible the program will attempt to create log folders if not already existing
@ -122,18 +123,16 @@ def init_logger(log_config: str) -> None:
except FileNotFoundError as e_fnf: except FileNotFoundError as e_fnf:
# Assume this is related to a missing Log Folder # Assume this is related to a missing Log Folder
# Try to create # Try to create
if e_fnf.filename and e_fnf.filename[-3:] == 'log': if e_fnf.filename and e_fnf.filename[-3:] == "log":
logfile = e_fnf.filename logfile = e_fnf.filename
logdir = os.path.dirname(logfile) logdir = os.path.dirname(logfile)
if not os.path.exists(logdir): if not os.path.exists(logdir):
try: try:
msg = 'Creating log folder "{}"'.format(logdir) logger.debug('Creating log folder "%s"', logdir)
logger.debug(msg)
os.makedirs(logdir, exist_ok=True) os.makedirs(logdir, exist_ok=True)
except Exception as e: except Exception as e:
msg = 'Error creating log folder "{}"'.format(logdir) logger.error('Error creating log folder "%s"', logdir, exc_info=e)
logger.error(msg, exc_info=e)
raise e raise e
elif logger.handlers: elif logger.handlers:
# if logger config loaded, but some file error happened # if logger config loaded, but some file error happened
@ -144,13 +143,11 @@ def init_logger(log_config: str) -> None:
if not os.path.exists(logdir): if not os.path.exists(logdir):
try: try:
msg = 'Creating log folder "{}"'.format(logdir) logger.debug('Creating log folder "%s"', logdir)
logger.debug(msg)
os.makedirs(logdir, exist_ok=True) os.makedirs(logdir, exist_ok=True)
except Exception as e: except Exception as e:
msg = 'Error creating log folder "{}"'.format(logdir) logger.error('Error creating log folder "%s"', logdir, exc_info=e)
logger.error(msg, exc_info=e)
raise e raise e
else: else:
# not sure the issue, raise the exception # not sure the issue, raise the exception
@ -160,11 +157,15 @@ def init_logger(log_config: str) -> None:
logging.config.fileConfig(log_config, disable_existing_loggers=False) logging.config.fileConfig(log_config, disable_existing_loggers=False)
else: else:
msg = 'Logging Config file "{}" not available, will be using defaults'.format(log_config) logger.debug('Logging Config file "%s" not available, will be using defaults', log_config)
logger.debug(msg)
if __name__ == '__main__':
msg = 'Script not meant to be run directly, please import into other scripts.\n\n' + \ if __name__ == "__main__":
'usage:\nimport {}'.format(SCRIPT_NAME) + '\n' + \ msg = (
'cfg = {}.getPlexConfig()'.format(SCRIPT_NAME) + '\n' "Script not meant to be run directly, please import into other scripts.\n\n"
logger.error(msg) + f"usage:\nimport {SCRIPT_NAME}"
+ "\n"
+ f"cfg = {SCRIPT_NAME}.getPlexConfig()"
+ "\n"
)
logger.error(msg)