3
e�$g; � @ s" d dl Z d dlZG dd� d�ZdS )� Nc @ s� e Zd ZG dd� de�Zd dd�Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd dd �ZdS ) �ClPwdc @ s e Zd Zdd� ZdS )zClPwd.NoSuchUserExceptionc C s t j| d|f � d S )NzNo such user (%s))� Exception�__init__)�self�user� r �/usr/lib/python3.6/clpwd.pyr s z"ClPwd.NoSuchUserException.__init__N)�__name__�
__module__� __qualname__r r r r r �NoSuchUserException s r Nc C s8 i | _ i | _i | _i | _|d kr.| jd�| _n|| _d S )Ni� )�
_user_key_map� _uid_key_map�_user_full_map�
_uid_full_map�get_sys_min_uid�_min_uid)r Zmin_uidr r r r s zClPwd.__init__c C s | j � | jS )N)�_load_passwd_databaser
)r r r r �
get_user_dict s zClPwd.get_user_dictc C s | j � | jS )N)r r )r r r r � get_uid_dict s zClPwd.get_uid_dictc C s | j � | jS )N)r r )r r r r �get_user_full_dict s zClPwd.get_user_full_dictc C s | j � | jS )N)r r )r r r r �get_uid_full_dict s zClPwd.get_uid_full_dictc C s0 y | j � | S tk
r* tj|��Y nX dS )z*
Return pw_entry for user
N)r �KeyErrorr r )r r r r r �get_pw_by_name# s zClPwd.get_pw_by_namec C s0 y | j � | S tk
r* tj|��Y nX dS )z7
Return list of passwd entries for uid
N)r r r r )r �uidr r r �
get_pw_by_uid, s zClPwd.get_pw_by_uidc C s2 y| j � | jS tk
r, tj|��Y nX dS )z&
Returns uid for user
N)r �pw_uidr r r )r r r r r �get_uid5 s z
ClPwd.get_uidc C s2 y| j � | jS tk
r, tj|��Y nX dS )z`
Returns homedir for a user
@param user: string
@return: string
N)r �pw_dirr r r )r r r r r � get_homedir> s zClPwd.get_homedirc C s� | j s�x�tj� D ]|}|| j|j< |j| j kr8g | j |j< | j |j j|� |j| jkr|| j|j< |j| j krzg | j |j< | j |j j|� qW dS )zZ
Loads the passwd database and fills user_to_uid and user_to_homedir maps
N)
r �pwdZgetpwallr �pw_namer �appendr r
r )r �entryr r r r I s zClPwd._load_passwd_databasec C s> y| j � | }W n tk
r. tj|��Y nX dd� |D �S )zs
Return names of users with uid specified
@param uid: int
@return: list of strings
c S s g | ]
}|j �qS r )r! )�.0r# r r r �
<listcomp>d s z#ClPwd.get_names.<locals>.<listcomp>)r r r r )r r Zentriesr r r � get_namesY s
zClPwd.get_names�� c C sn d}t jj|�rjt|d��J}xB|D ]:}|jd�r"yt|jd�d j� �S tk
rZ Y q"X q"W W dQ R X |S )z�
Return system defined MIN_UID from /etc/login.def or def_min_uid
@param def_min_uid: int
@return: MIN_UID: int
z/etc/login.defs�rZUID_MIN� N) �os�path�exists�open�
startswith�int�split�strip�
ValueError)r Z def_min_uidZLOGIN_DEF_FILE�lines�liner r r r g s
zClPwd.get_sys_min_uid)N)r' )r r
r r r r r r r r r r r r r r& r r r r r r s
r )r r* r r r r r �<module> s