관리-도구
편집 파일: environments.cpython-37.pyc
B �v6d D � @ s d dl mZ d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl m Z d dl m Z d dl mZmZmZmZ dd lmZmZ dd lmZmZ d dlZd d lmZ d d lmZ d Z dZ dZ dZ dZ dZ!dZ"G dd� de#�Z$d dd�Z%d dd�Z&d dd�Z'd!d d �Z(dS )"� )�print_function)�absolute_import)�divisionN)�datetime)� get_directory)� check_call� check_output� list_dirs� run_command� )�EXTENSION_PATTERN� ExtensionInfo)� Interpreter� interpreters)�JSONDecodeError)�ClSelectExcept� virtualenvz.virtualenv.backupz /opt/alt/python37/bin/virtualenv�#z /usr/share/l.v.e-manager/utilsZpython_wrapperzset_env_vars.pyc @ s� e Zd Zd4dd�Zdd� Zdd� Zd5dd �Zd6d d �Zd7d d�Zdd� Z d8dd�Z d9dd�Z dd� Z dd� Z dd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd:d*d+�Zd;d,d-�Zd<d.d/�Zd=d0d1�Zd2d3� ZdS )>� EnvironmentNc C s� || _ |r|| _nt�t�� �j| _t�| j�j| _ tj � | j d�| _ |d krVt | _n|| _tj � t| j| j�|�| _ tj � t| jt�| j �| _d | _d | _d | _d| | _ddi| _d S )Nz .pip/pip.log�python�MySQLdbz MySQL-python)�name�user�pwd�getpwuid�os�getuid�pw_name�getpwnam�pw_dir�homepath�path�join� pip_logfile�DEFAULT_PREFIX�prefix� _abs_prefix� BACKUP_PREFIXZ backup_pathZ _requirements� _interpreter�_pip�interpreter_name�_extension_remap)�selfr r r% � r- �Y/opt/alt/python37/lib/python3.7/site-packages/clselect/clselectpythonuser/environments.py�__init__& s" zEnvironment.__init__c C s d| j j| j j| j| j| jf S )Nz(%s.%s(name='%s', user='%s', prefix='%s'))� __class__� __module__�__name__r r r% )r, r- r- r. �__repr__< s zEnvironment.__repr__c s t �� j��� �fdd�}|S )Nc s4 t ��j� t ��j� � jt jd< �jt jd< d S )N�USER�HOME)r �setgid�pw_gid�setuid�pw_uidr �environr r- )r, �user_pwdr- r. �funcD s z!Environment._demote.<locals>.func)r r r )r, r< r- )r, r; r. �_demoteA s zEnvironment._demotec C s2 | j | �� | �� d�}|r.||= t| |�|iS |S )N)r � interpreter� extensions)r r> r? �getattr)r, �key�er- r- r. �as_dictL s zEnvironment.as_dictTc C sF | j | �� �� d�}|r*|�d| �� i� |rB||= t| |�|iS |S )N)r r> r? )r r> rC �updater? r@ )r, rA Zwith_extensionsrB r- r- r. � as_deepdictW s zEnvironment.as_deepdictc C s< |st | jd�}| j}|r(tj�||�}d�ttj�| j��| j �}t d|d|j |g}| � � | j |d�}yt||� W n� tjk �r. } z�t|�}d } d|kr�d} njd|k�r |�d |�d �d �} |d | � �d �} | d ks�| d kr�|} n |d | � } || d d � } n|} tj| | d ��W d d }~X Y nX | �� d S )N)� target_userz({}:{})z--promptz--python)� preexec_fn�cwd�waitzDisk quota exceededzJDisk quota exceeded. Contact system administrator to increase disk quota.� Traceback�:r � ���)�message�details)r r r! r r" �formatr �basenamer% r �VIRTUALENV_BIN�binaryr= r r r �ExternalProgramFailed�str�find�rfind�configure_environment) r, r> �versionrI r! �prompt�args�kwargs�errZ err_traceZerr_textZerr_charZ err_trace_endr- r- r. �createe s> zEnvironment.createc C sX d| j �d�d | j g}x:|D ]2}tj�||�}tj�|�rLt�|��d�r |S q W d S )Nr �.r z/opt/alt/python)r* �splitr r! r" �islink�readlink� startswith)r, �bin_pathZfiles_to_check�filer! r- r- r. �detect_python_binary� s z Environment.detect_python_binaryFc C s� t j�| jd�}t j�|| j�d }| �|�}|dkr:dS t j�|�rPt �|� t �||� y&t j�|�s�t �t j�t t �|� W n* t t fk r� |r�t �||� � Y nX t j�t j�|t ��s�t �t j�t t �t j�|t �� dS )z� Configures environment: 1. Rename binary to pythonX.Y_bin 2. Makes symlink from python binary to python_wrapper �binZ_binN)r r! r" r* rf �exists�remove�rename�symlink� WRAPPERS_PATH�PYTHON_WRAPPER�IOError�OSError�SET_ENV_VARS_SCRIPT)r, � auto_restorerd Znew_interpreter_path�interpreter_pathr- r- r. rX � s$ z!Environment.configure_environmentc C s> | j }|rtj �||�}tj �|�r:tddd|| �� d� d S )Nz/bin/rmz-rz--interactive=never)rG )r! r r" rh r r= )r, rY r! r- r- r. �destroy� s zEnvironment.destroyc C s || j kr| j | S |S dS )z� Returns extensions name considering extension remap table :param extension_name: Input extension name :return: Result extension name N)r+ )r, �extension_namer- r- r. �_get_extension_name� s z Environment._get_extension_namec C s� t j�| j|�}t j�| j|d��s(dS td|� | �|� | �|�}g }t j�|�rpt|d�}|� � }|� � | j |d� | j |dd� t|d�}|� |� |� � t| j|d�dd | �|�� dS ) zX Recreate python virtual environment with requirements :return: )rY Nz Re-create python virtualenv:�rT)rY rI �w�installz-r)r r! r" rh �pip�print� _pip_freeze�pip_requirements�open� readlines�closers r^ � writelinesr )r, rY Zenv_path�requirements_path� requirementsZ reqs_filer- r- r. � _recreate� s& zEnvironment._recreatec C s& x t dd��� D ]}| �|� qW d S )NrY )rA )r �keysr� )r, rY r- r- r. �recreate� s zEnvironment.recreatec C s t j�| j�S )N)r r! rh )r, r- r- r. rh � s zEnvironment.existsc C s | j st| j| jd�| _ | j S )N)r% rF )r( r r! r )r, r- r- r. r> � s zEnvironment.interpreterc C s� | � |�}t�| j�}|�t�}|d |dd � p4d }}ddd| �� d | j d | f}|r�|d }ddd| �� d | j d | d | f}t�|||�r�t d | ��t || � � | j d � | � � d S ) Nr r � z /bin/bashz-lz-cz --log-file=z install z==z6Extension '%s' install is prohibited. System extension)r[ rG rH )ru r �get_locked_extensionsr* r` �VERSION_DELIMITERry r# �is_extensions_locked� ValueErrorr r= r r{ )r, rt �locked_extensions�t� extensionrY �commandr- r- r. �extension_install� s $ z Environment.extension_installc C sD ddd| � � d | j d�|� f}t|| �� | jd� | �� d S )Nz /bin/bashz-lz-cz --log-file=z install -r {})r[ rG rH )ry r# rP r r= r r{ )r, r� r� r- r- r. � extension_install_requirements s z*Environment.extension_install_requirementsc C s2 t | �� d| j dd|| �� | jd� | �� d S )Nz --log-file=rx z --upgrade)rG rH )r ry r# r= r r{ )r, r� r- r- r. �extension_update s z Environment.extension_updatec C s� t �| j�}|�t�}|d |dd � p*d }}|r<|d }t �|||�rVtd| ��tj| � � d| j d|f| � � tj tj tj | j dd�}|�d �\}}|jr�t|p�|��| �� d S ) Nr r r� z$Extension '%s' removal is prohibitedz --log-file=� uninstallT)rG �stdin�stderr�stdoutrH �text�y)r r� r* r` r� r� r� � subprocess�Popenry r# r= �PIPEr � communicate� returncode� Exceptionr{ )r, r� r� r� rY �pr� r� r- r- r. �extension_uninstall s z Environment.extension_uninstallc C sB i }t �| j�}y<t| �� dd| j d| �� | jd�}dd� t� |�D �}W nN t t t t jt jfk r� t| �� dd| j | �� | jd�}t�|�}Y nX dd� |D �}x�t||�D ]�\\}}}t �|||��r,ttd d� |�d �D ��t|�|�� �} | �r:t|�|��d k�r:|d �| �d �||< q�||d �||< q�W |S )N�listz --log-file=z --format=json)rG rH c S s g | ]}|d |d f�qS )r rY r- )�.0�xr- r- r. � <listcomp>- s z*Environment.extensions.<locals>.<listcomp>c s s | ]\}}t � �|�V qd S )N)r � extension_doc)r� r� �_r- r- r. � <genexpr>1 s z)Environment.extensions.<locals>.<genexpr>c S s g | ] }|� � �qS r- )�strip)r� �vr- r- r. r� 5 s �,r z, )�docrY )r r� r* r ry r# r= r �json�loadsr �KeyErrorr� r �FileProcessErrorrT r �findall�zipr� r� �setr` �get�lenr" ) r, �resultr� �outputr? �docsr rY r� Z version_diffr- r- r. r? ( s$ $ "zEnvironment.extensionsc C s< |d k r t j�| j|dd�S | js6t j�| jdd�| _| jS )Nrg ry )r r! r" r) )r, rY r- r- r. ry > s zEnvironment.pipc C s* |d k rt j�| j|d�S t j�| jd�S )Nzrequirement.pip)r r! r" )r, rY r- r- r. r| E s z Environment.pip_requirementsc C s� d}| � � }|r ||j }ntdd�|j }t� }tj�|j�rbt�|j�� d�rb|r^t d� dS |jdt � � � d� } t�|j�} t�|j| � t�| | j| j� y�xx|jD ]n} t�|j|j�r�|s�d}|r�t d| � q�|r�tj�d |j| f � td d |j| fd � |� | � t d � q�W d}W n^ t t fk �r� } z:t dt | � � x|D ]} t� | | � �qVW t�!| � W dd} ~ X Y nX |�s�t�!| � |S )a_ copy binary python from /opt/alt/pythonXY/bin/pythonX.Y to virtualenvdir/bin/pythonX.Y :param backup: make backup old python interpreter :param force: force rewrite python interpreter without check :param verbose: print actions details to stdout :return: True - updating success; False - updating fail FrY )rA z/opt/alt/pythonzVNothing to do, binary in your virtual environment is already symlink to global python!z.orig_%sz%Y-%m-%d_%H-%Mz not need updating; skip '%s'z copy '%s' -> '%s'...z/bin/cpz--force)�cmdZDoneTzFail %sN)"r> rY r r� r r! ra � python_binrb rc rz r �now�strftime�stat�shutil�copy�chown�st_uid�st_gidZ binary_list�filecmp�cmprS �sysr� �writer �appendro rn rU �copyfile�unlink)r, �backup�force�verbose�_alt_interpreters_dict� update_resultr> Zmain_interpreterZ updated_listZ python_backupZstat_Zvirtualenv_python_binrB Zupdated_pythonr- r- r. �update_python_interpreterJ sF z%Environment.update_python_interpreterc C sP t j�| �|��sdS | �|�ddf}t| �|�d�}t|| �� | j|d� dS )zX Output installed packages in requirements format :return: None N�freezez-lrw )r[ rG rH r� ) r r! rh ry r} r| r r= r )r, rY r� �fr- r- r. r{ ~ s zEnvironment._pip_freezec C s( x"t dd��� D ]}| j|d� qW dS )zX Output installed packages in requirements format :return: None rY )rA )rY N)r r� r{ )r, rY r- r- r. � pip_freeze� s zEnvironment.pip_freeze)NN)N)NT)NNN)F)N)N)N)FFTN)N)r2 r1 � __qualname__r/ r3 r= rC rE r^ rf rX rs ru r� r� rh r> r� r� r� r� r? ry r| r� r{ r� r- r- r- r. r $ s0 . ! 4 r c C s@ |st }| r"tj�t�| �j|�S tj�t�t�� �j|�S d S )N) r$ r r! r" r r r r r )r r% r- r- r. r& � s r& c C sT t | |�}y t|�}W n tk r* g S X g }x |D ]}|�t|| |�� q6W |S )N)r& r ro r� r )r r% � venv_path�env_list�envs�env_namer- r- r. � environments� s r� c s t � fdd�t||�D ��S )Nc 3 s" | ]}t |j� d ��� �V qdS ))rA N)r� rC �items)r� rB )rA r- r. r� � s z$environments_dict.<locals>.<genexpr>)�dictr� )rA r r% r- )rA r. �environments_dict� s r� c s t � fdd�t||�D ��S )Nc 3 s" | ]}t |j� d ��� �V qdS ))rA N)r� rE r� )r� rB )rA r- r. r� � s z(environments_deepdict.<locals>.<genexpr>)r� r� )rA r r% r- )rA r. �environments_deepdict� s r� )NN)NN)NN)NN))� __future__r r r r r� r r� r� r� r Zclselect.clselectctlr �clselect.utilsr r r r r? r r r r � simplejsonr� r �clselect.clselectexceptr r$ r' rR r� rl rm rp �objectr r& r� r� r� r- r- r- r. �<module> s< q