File: //home/arjun/projects/env/lib/python3.10/site-packages/honcho/__pycache__/command.cpython-310.pyc
o
we�- � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d dlmZ d d l mZmZ ejd
d ejd � e�e�Zej� e� � �Z e d
d� e d�D ��Z G dd� de!�Z"d�dd�Z#e j$ddd�Z%e#e%dd� e%j&ddd �Z'de'_(d d � Z)e'j*d d d!�Z+e#e+� d"d#� Z,e'j*d$d%d!�Z-e#e-� e-j.d&d'd(e e/d)d*� e-j.d+d,d-d.e/d/d*� e-j.d0d1d2e j0e1d3d*� e-j.d4d5d6e/d7d8� e-j.d9d:d;e/d<� e-j.d=d>d?d@e/dA� e-j.dBdCdDde/d/d*� e-j.dEdFe2e �3� �e/dGdH� e-j.dIdJe/dKd8� dLdM� Z4e'j*dNdOd!�Z5e5j.dPdQdRdS� dTdU� Z6e'j*dVdWd!�Z7e#e7� e7j.dXe j8dYdZ� d[d\� Z9e'j*d]d^d!�Z:e#e:� e:j.d0d1d2e1e j0d3d_� e:j.d4d5d`e/d7d8� e:j.dadbdce/ddd8� e:j.dedfdgdZ� dhdi� Z;e'j*djdkd!�Z<e)e,e4e6e9e;dl�Z=d�dmdn�Z>dodp� Z?dqdr� Z@dsdt� ZAdudv� ZBdwdx� ZCdydz� ZDd{d|� ZEd}d~� ZFdd�� ZGd�d�� ZHed�k�r�e>� dS dS )�� N)� OrderedDict)� defaultdict)�iter_entry_points)� __version__)�Popen)�Manager)�Printer)�compat�environz5%(asctime)s [%(process)d] [%(levelname)s] %(message)sz%Y-%m-%d %H:%M:%S)�format�datefmt�levelc c s � | ]}|j |fV qd S �N)�name)�.0�_export� r �G/home/arjun/projects/env/lib/python3.10/site-packages/honcho/command.py� <genexpr> s � �r �honcho_exportersc @ s e Zd ZdS )� CommandErrorN)�__name__�
__module__� __qualname__r r r r r s r Fc C s� |rd nt j}| jdd|pddd� | jddd|pd d
d � | jd d
|dd� | jdd
|dd� | jddd|dd � | jddddt d� d S )Nz-ez--envz.envz'environment file[,file] (default: .env))�default�helpz-dz
--app-root�DIR�.z procfile directory (default: .))�metavarr r z --no-colour�
store_truezdisable coloured output)�actionr r z --no-prefixzdisable logging prefixz-fz
--procfile�FILEz!procfile path (default: Procfile)z-vz --version�versionz %(prog)s )r r" )�argparse�SUPPRESS� add_argumentr )�parser�
with_defaults�suppressr r r �_add_common_args" s: �����
�r) �honchoz"Manage Procfile-based applications)� descriptionT)r' �tasks�command)�title�destc C s0 t t| jt| ���}t�d�d�|j��� d S )Nz Valid procfile detected ({0})z, ) � _procfile�_procfile_path�app_root�_choose_procfile�log�infor �join� processes)�args�procfiler r r �
command_checkE s r: �checkzvalidate a Procfile)r c C s | j dkr| j �d| j�| _ t| jt| ��}t|�}t| j| j�}t | j
�}t | |�}t j
|j|||d�}t| j �� }|| jd�}| jtj�| j�| j | j| jpS| jd�} t| j� |�|| �D ] }
tj�| j|
j �} t � d| � t | |
j � |
j r�t�!| d� qad S )N� /var/log/APP�APP)� concurrency�env�port)� template_dir)�appr2 r4 �shell�userz Writing '%s'i� )"r4 �replacerB r1 r2 r3 r0 � _read_envr? �_parse_concurrencyr> � _choose_portr
�expand_processesr7 �export_choicesr �loadrA �os�path�abspathrC rD �_mkdir�location�renderr6 r r5 � _write_file�content�
executable�chmod) r8 �
procfile_pathr9 r? r> r@ r7 � export_ctor�export�context�frM r r r �command_exportQ s:
�
�
��r[ rX z#export a Procfile to another formatz-az--appzalternative app namer= )r r �typer z-lz--logz"directory to place process logs inr< r z-pz--portz$starting port number (default: 5000)�Nz-cz
--concurrencyz#number of each process type to run.zprocess=num,process=num)r r\ r z-uz--userz"user the application should run as)r r\ z-sz--shellz)the shell that should run the applicationz/bin/sh)r r r\ z-tz--template-dirz(directory to search for custom templatesr z'format to export to; one of %(choices)s�FORMAT)r �choicesr\ r rP zfolder to export to�LOCATIONc C s, dg}| j r |�| j � t�|d d d� �S )Nz--help���)�task�appendr& �
parse_args)r8 �argvr r r � command_help� s rf r z-describe available tasks or one specific taskrb ztask to show help for�?)r �nargsc C s� t j�t| j| j�� | j}|r |d dkr |dd � }tjr"|}n
d� dd� |D ��}t
|t j t j
dd�}t�tjtj� |�� t �|j� d S ) Nr z--� � c s s � | ]}t �|�V qd S r )�shlex�quote)r �argr r r r � s � z command_run.<locals>.<genexpr>F)�stdout�stderr�start_new_session)rL r
�updaterF r2 r? re r �
ON_WINDOWSr6 r �sysrn ro �signal�SIGINT�SIG_IGN�wait�exit�
returncode)r8 re �cmd�pr r r � command_run� s �r| �runz2run a command using your application's environmentre zcommand to run)rh r c C s t | jt| ��}t|�}t| j�}t| j| j�}t| j �}t
| |�}| j rFt � }| j D ]}z |j | ||< W q+ t
yD td�|���w n|j }tttj| j | j d��} tj|||||d�D ]}
tj�� } | �|
j� | j|
j |
j |
j | d� qa| � � t� | j � d S )Nz-Process type '{0}' does not exist in Procfile)�colour�prefix)r> r? �quietr@ )r� r? )!r1 r2 r3 r0 rG r> rF r? � _parse_quietr� rH r7 r �KeyErrorr r r r rs rn � no_colour� no_prefixr
rI rL �copyrq � add_processr rz �looprx ry ) r8 rV r9 r> r? r� r@ r7 r �managerr{ �er r r �
command_start� s>
���
�
r� �startz-start the application (or a specific PROCESS))r r\ r r z'the number of each process type to run.z-qz--quietz*process names for which to suppress outputzprocess1,process2,process3r7 �*z-process(es) to start (default: all processes)c C s t djtjtd�� d S )Nz{prog} {version})�progr" )�printr r& r� r )r8 r r r �command_version s �r� r" zdisplay honcho version)r; rX r r} r� r" c
C sv | d ur
t �| �}nt �� }z
t� t|j |� W d S ty: } zt�t|�� t �
d� W Y d }~d S d }~ww �Nri ) r& rd �_check_output_encoding�COMMANDSr- r r4 �error�strrs rx )re r8 r� r r r �main s ��r� c C s t j�| |�S r )rL rM r6 )r2 r9 r r r r1 ) s r1 c
C s� zt | �� }|�� }W d � n1 sw Y W n ty% td��w zt�|�}W |S ty@ } ztt|���d }~ww )Nz(Procfile does not exist or is not a file)�open�read�IOErrorr r
�parse_procfile�AssertionErrorr� )�filenamerZ rS r9 r� r r r r0 - s
�� � � ��r0 c C s� dd� |� d�D �}g }|D ]-}z"ttj�| |���}|�|�� � W d � n1 s,w Y W q ty; Y qw t� d�|��S )Nc S s g | ]}|� � �qS r )�strip)r r� r r r �
<listcomp>= s z _read_env.<locals>.<listcomp>�,�
)
�splitr� rL rM r6 rc r� r� r
�parse)r2 r? �filesrS �envfilerZ r r r rF < s �� �rF c C sH t dd� �}| d u r |S | �d�D ]}|�dd�\}}t|�||< q|S )Nc S s dS r� r r r r r �<lambda>J s z$_parse_concurrency.<locals>.<lambda>r� �=ri )r r� �int)�desc�result�item�keyr> r r r rG I s rG c C s g }| d u r|S | � d�}|S )Nr� )r� )r� r� r r r r� S s
r� c C sD t | j| j�}|�dd �}t| d�r| jd ur| jS |d ur |S dS )N�PROCFILEr9 �Procfile)rF r2 r? �pop�hasattrr9 )r8 r? � env_procfiler r r r3 [ s r3 c C sN |� dd �}tj� dd �}t| d�r| jS |d ur t|�S |d ur%t|�S dS )N�PORTr@ i� )r� rL r
r� r@ r� )r8 r? �env_port� os_env_portr r r rH g s
rH c
C sL t j�| �rd S zt �| � W d S ty% } z t�d� t|��d }~ww )Nz!Could not create export directory)rL rM �exists�makedirs�OSErrorr4 r� r )rM r� r r r rO u s
��rO c
C sz t tj�| �� z t| d��}|�|� W d � W d S 1 s w Y W d S ty< } z t�d� t |��d }~ww )N�wz Could not write to export file)
rO rL rM �dirnamer� �writer� r4 r� r )rM rS �fpr� r r r rR s &�
��rR c C sL t jjd u } t�d�}t j�� sd S | s t�t jj�|kr$t�d� d S d S )N�utf8z�Your terminal is not configured to receive UTF-8 encoded text. Please adjust your locale settings or force UTF-8 output by setting PYTHONIOENCODING="utf-8".)rs rn �encoding�codecs�lookup�isattyr4 �warn)� no_encodingr� r r r r� � s
�r� �__main__)Fr )Ir# r� �loggingrL rk rs rt � collectionsr r �
pkg_resourcesr r* r �honcho.processr �honcho.managerr �honcho.printerr r r
� basicConfig�INFO� getLoggerr r4 rM �basename�getcwd�BASENAME�dictrJ � Exceptionr r) �ArgumentParserr& �add_subparsers�
subparsers�requiredr: �
add_parser� parser_checkr[ �
parser_exportr% r� r$ r� �sorted�keysrf � parser_helpr| �
parser_run� REMAINDERr� � parser_startr� �parser_versionr� r� r1 r0 rF rG r� r3 rH rO rR r� r r r r �<module> s< �
�
� �$��������
�����$�������
�