3diy
/
moonraker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

utils: add method to load system level modules 

Signed-off-by: Eric Callahan <arksine.code@gmail.com>
This commit is contained in:
Eric Callahan 2021-11-15 06:41:38 -05:00
parent adb88fd8cf
commit dfa11f0f3e
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
moonraker/utils.py
View File

@ -8,6 +8,8 @@ from __future__ import annotations
import logging import logging
import logging.handlers import logging.handlers
import os import os
import glob
import importlib
import pathlib import pathlib
import sys import sys
import subprocess import subprocess
@ -18,6 +20,7 @@ from queue import SimpleQueue as Queue
# Annotation imports # Annotation imports
from typing import ( from typing import (
TYPE_CHECKING,
List, List,
Optional, Optional,
ClassVar, ClassVar,
@ -26,7 +29,12 @@ from typing import (
Any, Any,
) )
if TYPE_CHECKING:
from types import ModuleType
MOONRAKER_PATH = os.path.join(os.path.dirname(__file__), '..') MOONRAKER_PATH = os.path.join(os.path.dirname(__file__), '..')
SYS_MOD_PATHS = glob.glob("/usr/lib/python3*/dist-packages")
SYS_MOD_PATHS += glob.glob("/usr/lib/python3*/site-packages")
class ServerError(Exception): class ServerError(Exception):
def __init__(self, message: str, status_code: int = 400) -> None: def __init__(self, message: str, status_code: int = 400) -> None:
@ -174,3 +182,19 @@ def verify_source(path: str = MOONRAKER_PATH) -> Optional[Tuple[str, bool]]:
ign_exts = rinfo['ignored_exts'] ign_exts = rinfo['ignored_exts']
checksum = hash_directory(path, ign_exts, ign_dirs) checksum = hash_directory(path, ign_exts, ign_dirs)
return checksum, checksum == orig_chksum return checksum, checksum == orig_chksum
def load_system_module(name: str) -> ModuleType:
for module_path in SYS_MOD_PATHS:
sys.path.insert(0, module_path)
try:
module = importlib.import_module(name)
except ImportError as e:
if not isinstance(e, ModuleNotFoundError):
logging.exception(f"Failed to load {name} module")
sys.path.pop(0)
else:
sys.path.pop(0)
break
else:
raise ServerError(f"Unable to import module {name}")
return module