U
�V� � @ sp d dl Z d dlZd dlZd dlZd dlZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z e dkrle�
e � � dS )� Nc C sV zt �t�| ��d��W S tjk
r< td�| ���Y n tk
rP Y d S X d S )Nzpbr.jsonzPackage {0} not installed)�json�loads�
pkg_resources�get_distributionZ get_metadataZDistributionNotFound� Exception�format)Z package_name� r �/main.py�
_get_metadata s ���r
c C s t | j�d }|rt|� d S )N�sha)� _get_info�name�print)�argsr r r r �get_sha$ s r c C s t djf t| j��� d S )Nz!{name} {version} {released} {sha})r r r r
)r r r r �get_info* s �r c C s� t | �}t�| �j}|r4|d r&d}nd}|d }nL|�d�}|d �d�rb|d dd � }d}n d }d}|D ]}|�� snd}qnt| |||d
�S ) NZ
is_release�releasedz pre-releaseZ git_version�.����g� � )r
�versionr r )r
r r r �split�
startswith�isdigit�dict)r
Zmetadatar r r Z
version_parts�partr r r r / s"
r c C sF t jD ]:}t|j�}djf |�}|d r8|djf |�7 }t|� qd S )Nz{name}=={version}r z # git sha {sha})r Z working_setr Z project_namer r )r Zdist�info�outputr r r �freezeF s
r c
C s� t jdd�} | jdddttj�d��d� | jdd d
d �}|jd d
d�}|j t
d� |jddd� |jddd�}|j t d� |jddd� |jddd�}|j t d� | �
� }z|�|� W n* tk
r� } z t|� W 5 d }~X Y nX d S )Nz pbr: Python Build Reasonableness)� descriptionz-vz --versionr �pbr)�actionr Zcommandszvalid commandszadditional help)�titler! �helpr zprint sha of package)r% )�funcr
zpackage to print sha ofr z print version info for packagezpackage to print info ofr z-print version info for all installed packages)�argparse�ArgumentParser� add_argument�strr" r Z VersionInfoZadd_subparsersZ
add_parserZ set_defaultsr r r �
parse_argsr& r r )�parserZ
subparsersZcmd_shar �er r r �mainO sB � � � � � r. �__main__)r' r �sysr Z pbr.versionr" r
r r r r r. �__name__�exitr r r r �<module> s