mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-02-18 22:23:56 +08:00
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
import os
|
|
import platform
|
|
import shutil
|
|
import uuid
|
|
|
|
|
|
from openpilot.common.params import Params
|
|
from openpilot.system.hardware import PC
|
|
from openpilot.system.hardware.hw import Paths
|
|
from openpilot.system.hardware.hw import DEFAULT_DOWNLOAD_CACHE_ROOT
|
|
|
|
class OpenpilotPrefix:
|
|
def __init__(self, prefix: str | None = None, create_dirs_on_enter: bool = True, clean_dirs_on_exit: bool = True, shared_download_cache: bool = False):
|
|
self.prefix = prefix if prefix else str(uuid.uuid4().hex[0:15])
|
|
shm_path = "/tmp" if platform.system() == "Darwin" else "/dev/shm"
|
|
self.msgq_path = os.path.join(shm_path, "msgq_" + self.prefix)
|
|
self.create_dirs_on_enter = create_dirs_on_enter
|
|
self.clean_dirs_on_exit = clean_dirs_on_exit
|
|
self.shared_download_cache = shared_download_cache
|
|
|
|
def __enter__(self):
|
|
self.original_prefix = os.environ.get('OPENPILOT_PREFIX', None)
|
|
os.environ['OPENPILOT_PREFIX'] = self.prefix
|
|
|
|
if self.create_dirs_on_enter:
|
|
self.create_dirs()
|
|
|
|
if self.shared_download_cache:
|
|
os.environ["COMMA_CACHE"] = DEFAULT_DOWNLOAD_CACHE_ROOT
|
|
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_obj, exc_tb):
|
|
if self.clean_dirs_on_exit:
|
|
self.clean_dirs()
|
|
try:
|
|
del os.environ['OPENPILOT_PREFIX']
|
|
if self.original_prefix is not None:
|
|
os.environ['OPENPILOT_PREFIX'] = self.original_prefix
|
|
except KeyError:
|
|
pass
|
|
return False
|
|
|
|
def create_dirs(self):
|
|
try:
|
|
os.mkdir(self.msgq_path)
|
|
except FileExistsError:
|
|
pass
|
|
os.makedirs(Paths.log_root(), exist_ok=True)
|
|
|
|
def clean_dirs(self):
|
|
symlink_path = Params().get_param_path()
|
|
if os.path.exists(symlink_path):
|
|
shutil.rmtree(os.path.realpath(symlink_path), ignore_errors=True)
|
|
os.remove(symlink_path)
|
|
shutil.rmtree(self.msgq_path, ignore_errors=True)
|
|
if PC:
|
|
shutil.rmtree(Paths.log_root(), ignore_errors=True)
|
|
if not os.environ.get("COMMA_CACHE", False):
|
|
shutil.rmtree(Paths.download_cache_root(), ignore_errors=True)
|
|
shutil.rmtree(Paths.comma_home(), ignore_errors=True)
|