You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
764 B
27 lines
764 B
3 years ago
|
import os
|
||
|
|
import shutil
|
||
|
|
import uuid
|
||
|
|
|
||
|
|
from common.params import Params
|
||
|
|
|
||
|
|
class OpenpilotPrefix(object):
|
||
|
|
def __init__(self, prefix: str = None) -> None:
|
||
|
|
self.prefix = prefix if prefix else str(uuid.uuid4())
|
||
|
|
self.msgq_path = os.path.join('/dev/shm', self.prefix)
|
||
|
|
|
||
|
|
def __enter__(self):
|
||
|
|
os.environ['OPENPILOT_PREFIX'] = self.prefix
|
||
|
|
try:
|
||
|
|
os.mkdir(self.msgq_path)
|
||
|
|
except FileExistsError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def __exit__(self, exc_type, exc_obj, exc_tb):
|
||
|
|
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)
|
||
|
|
del os.environ['OPENPILOT_PREFIX']
|
||
|
|
return True
|