AlkantarClanX12
Current Path : /opt/cloudlinux/venv/lib64/python3.11/site-packages/clcommon/ |
Current File : //opt/cloudlinux/venv/lib64/python3.11/site-packages/clcommon/clcaptain.py |
# -*- coding: utf-8 -*- # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2018 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT # import os import errno from subprocess import Popen, PIPE from clcommon.utils import ExternalProgramFailed CAPTAIN_PATH = '/usr/share/python-cllib/cl.captain' def captain(do_try, cmd, stdin=None): try: do_try() except (OSError, IOError) as e: if e.errno == errno.EDQUOT and os.geteuid() > 0: command = [CAPTAIN_PATH] + [str(s) for s in cmd] with Popen( command, stdin=None if stdin is None else PIPE, stdout=PIPE, stderr=PIPE, text=True, ) as proc: if stdin is None: std_out, std_err = proc.communicate() else: std_out, std_err = proc.communicate(stdin) if proc.returncode != 0: raise ExternalProgramFailed( std_err or f'output of "{command}": {std_out}' ) from e else: raise def mkdir(path, mode=0o755, recursive=False): umask = os.umask(0) if recursive: collector = "/" for subdir in path.split("/"): collector = os.path.join(collector, subdir) if os.path.exists(collector): continue captain(lambda: os.mkdir(collector, mode), ['-d', collector, '-m', f'{mode:o}']) else: captain(lambda: os.mkdir(path, mode), ['-d', path, '-m', f'{mode:o}']) os.umask(umask) def write(path, content=''): def do_try(): with open(path, 'w', encoding='utf-8') as fp: fp.write(content) fp.flush() os.fsync(fp.fileno()) captain(do_try, ['-f', path], content) def symlink(src, dst): captain(lambda: os.symlink(src, dst), ['-l', dst, '-s', src])