3
�ft`� � @ s� d dl mZ d dl mZ d dlmZmZmZ d dlmZ d dl m
Z
d dl Z d dl
Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZejd�Zdd � Zd
d � Zd d
� Zdd� Zdd� ZG dd� de �Z dS )� )�absolute_import)�unicode_literals)�ProcessLockError�ThreadLockError� LockError)�_)�miscN�dnfc C s6 t jj� s2tj| jd��j� }tjj t
j � d|�} | S )Nzutf-8Zlocks) r �utilZ am_i_root�hashlibZsha1�encodeZ hexdigest�os�path�joinr Z getCacheDir)Zdir_Zhexdir� r �/usr/lib/python3.6/lock.py�
_fit_lock_dir& s
r c C s t tjjt| �d�d| �S )Nzdownload_lock.pid�cachedir)� ProcessLockr
r r r )r � exit_on_lockr r r �build_download_lock/ s r c C s t tjjt| �d�d| �S )Nzmetadata_lock.pidZmetadata)r r
r r r )r r r r r �build_metadata_lock3 s r c C s t tjjt| �d�d| �S )Nzrpmdb_lock.pidZRPMDB)r r
r r r )Z
persistdirr r r r �build_rpmdb_lock8 s r c C s t tjjt| �d�d| �S )Nz log_lock.pid�log)r r
r r r )Zlogdirr r r r �build_log_lock= s r c @ s> e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )r Fc C s& || _ d| _|| _|| _tj� | _d S )Nr )�blocking�count� description�target� threading�RLock� thread_lock)�selfr r r r r r �__init__C s
zProcessLock.__init__c C s2 | j jdd�s d| j }t|��| jd7 _d S )NF)r z'%s already locked by a different thread� )r! �acquirer r r )r" �msgr r r � _lock_threadJ s
zProcessLock._lock_threadc C s> t j| jt jt jB d�}�zytj|tjtjB � W n4 t k
rh } z|j
t
j krVdS � W Y d d }~X nX t j |d�}t
|�dkr�t j|t|�jd�� |S y t|�}W n* tk
r� td�| j }t|��Y nX ||kr�|S t jd| t j��s*t j|dt j� t j|d� t j|t|�jd�� |S |S t j|� X d S ) Ni� r$ � r zutf-8z�Malformed lock file found: %s.
Ensure no other dnf/yum process is running and remove the lock file manually or run systemd-tmpfiles --remove dnf.conf.z
/proc/%d/stat���)r
�openr �O_CREAT�O_RDWR�fcntlZflockZLOCK_EXZLOCK_NB�OSError�errnoZ EWOULDBLOCK�read�len�write�strr �int�
ValueErrorr r �access�F_OK�lseek�SEEK_SET� ftruncate�close)r" �pid�fd�eZold_pidr& r r r � _try_lockP s6 zProcessLock._try_lockc C s | j d8 _ | jj� d S )Nr$ )r r! �release)r" r r r �_unlock_threadz s zProcessLock._unlock_threadc C s� t jjtjj| j�� | j� d}tj� }| j |�}xp||kr�|dkr�| j
sl| j � d| j |f }t
||��||kr�td�| }tj|� |}tjd� | j |�}q6W d S )Nr$ z%s already locked by %dz*Waiting for process with pid %d to finish.r) r) )r r
Z
ensure_dirr
r �dirnamer r' �getpidr? r rA r r r �logger�info�timeZsleep)r" Zprev_pidZmy_pidr< r& r r r � __enter__~ s"
zProcessLock.__enter__c G s" | j dkrtj| j� | j� d S )Nr$ )r r
�unlinkr rA )r" Zexc_argsr r r �__exit__� s
zProcessLock.__exit__N)F) �__name__�
__module__� __qualname__r# r' r? rA rG rI r r r r r B s
*r ) Z
__future__r r Zdnf.exceptionsr r r Zdnf.i18nr Zdnf.yumr Z dnf.loggingr Zdnf.utilr/ r- r Zloggingr
r rF Z getLoggerrD r r r r r �objectr r r r r �<module> s(