Current File : //opt/alt/python37/lib/python3.7/site-packages/lvestats/lib/__pycache__/uidconverter.cpython-37.pyc
B
��1bm � @ s d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d d lmZmZ d d
lmZ d d lmZmZ d Zd
Zdadadd� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Z!d d � Z"d d � Z#d d!� Z$d"d#� Z%d$d%� Z&d&d'� Z'd(d)� Z(dS )*� )�absolute_importN)�cpapi)�Engine)� sessionmaker)�
NoResultFound)� LIMIT_LVP_ID)�NameMap)�Optional�Dict)�user)�deserialize_lve_id�serialize_lve_idZshaman)�uid_to_username�username_to_uidc C s t d krt� } | �� | a t S )N)�_name_map_cacher Z
link_xml_node)Zname_map� r �J/opt/alt/python37/lib/python3.7/site-packages/lvestats/lib/uidconverter.py�
_get_name_map s
r c C s& y t �| �jS tk
r d S X d S )N)�pwd�getpwuid�pw_name�KeyError)�uidr r r �_uid_to_username_local( s r c C sF t � �| �}|d krByt�| �j}W n tk
r@ t| �}Y nX |S )N)r �get_namer r r r � _get_reseller_name_from_panel)r �resellerr r r �_uid_to_reseller_local0 s r c C sN t d kr@ydd� t�� �� D �a W n tjk
r> t� a Y nX t �| �}|S )Nc S s i | ]\}}||�qS r r )�.0�k�vr r r �
<dictcomp>A s z1_get_reseller_name_from_panel.<locals>.<dictcomp>)�_id_reseller_map_cacher Zget_reseller_id_pairs�itemsZ NotSupported�dict�get)r r r r r r ; s
r c C s t � �| �S )N)r Zget_id)�namer r r �_reseller_to_uid_localH s r' c C s t | �}|d k rtt|�S d S )N)r' r
r )r& �resultr r r �reseller_to_uidM s r) c C s& y t �| �jS tk
r d S X d S )N)r �getpwnam�pw_uidr )�usernamer r r �_username_to_uid_localS s r- c C s^ t |d�� }z@y&|�t��tj| ktj|k��� }W n tk
rH d S X W d |�� X |j S )N)�bind)
r �queryr �filterr � server_id�oner �close� user_name)r r1 � db_engine�session� user_instr r r �_uid_to_username_db[ s
r8 c C s t | ||�S )N)r8 )r r1 r5 r r r �_uid_to_reseller_dbi s r9 c C s^ t |d�� }z@y&|�t��tj| ktj|k��� }W n tk
rH d S X W d |�� X |j S )N)r. )
r r/ r r0 r4 r1 r2 r r3 r )r, r1 r5 r6 r7 r r r �_username_to_uid_dbo s
r: c C s` d | |||fkrt d��t| �\} }|r,tnt}|r8tnt}||krT|| �}|rT|S || ||�S )NzEAll parameters should be specified and not None for uid_to_username())�
ValueErrorr r r r9 r8 )r �local_server_idr1 r5 � is_resellerZuid_to_name_local_funcZuid_to_name_database_funcr( r r r r } s r c C s< d| |||fkrt d��||kr0t| �}|r0|S t| ||�S )z�
Lookups for uid in local user database (/etc/passwd) and global one (table lve_stats2_user).
If local_server_id equals to server_id then local user database has priority.
NzEAll parameters should be specified and not None for username_to_uid())r; r- r: )r, r< r1 r5 r( r r r r � s r c C s" t | �\}}|r| S tt| �S d S )N)r r
r )Zany_idZdummyr= r r r �_convert_id_to_lvp_id� s r> c C s t | �}|r|S dS )N���)r) )r& Zlvp_idr r r �_convert_name_to_lvp_id� s r@ ))Z
__future__r r Zclcommonr Zsqlalchemy.enginer Zsqlalchemy.ormr Zsqlalchemy.orm.excr Zclcommon.clprocr Zlveapir �typingr r
Z lvestats.ormr Zlvestats.lib.commons.funcr r
�
__author__�__all__r r" r r r r r' r) r- r8 r9 r: r r r> r@ r r r r �<module> s8
Mini Shell