Server : Apache System : Linux host44.registrar-servers.com 4.18.0-513.18.1.lve.2.el8.x86_64 #1 SMP Sat Mar 30 15:36:11 UTC 2024 x86_64 User : vapecompany ( 2719) PHP Version : 7.4.33 Disable Function : NONE Directory :  /opt/cloudlinux/venv/lib/python3.11/site-packages/clcommon/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

#!/usr/bin/python
# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT

# pylint: disable=consider-using-with
# pylint: disable=unspecified-encoding

import contextlib
import fcntl
import os


def setup_lock(lockfile_path):
    lock_file_dir = os.path.dirname(lockfile_path)
    if lock_file_dir and not os.path.exists(lock_file_dir):
        os.makedirs(lock_file_dir)

    fd = open(lockfile_path, 'a+')

    # waiting till lock is acquired
    fcntl.flock(fd, fcntl.LOCK_EX)

    fd.seek(0, 0)
    fd.truncate()
    fd.write(str(os.getpid()))
    fd.flush()

    return fd


def release_lock(fd):
    fcntl.flock(fd.fileno(), fcntl.LOCK_UN)
    fd.close()


@contextlib.contextmanager
def acquire_lock(lockfile_path):
    """
    Usage:
    with acquire_lock():
       ... do something ...
    """
    fd = setup_lock(lockfile_path)
    try:
        yield
    finally:
        release_lock(fd)