File: //lib/python3.6/site-packages/sos/upload/__pycache__/__init__.cpython-36.pyc
3
\��h�3 � @ s` d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlm
Z
G dd� de�Z dS )� N)� SoSComponent)�_sos)� __version__c s� e Zd ZdZdZdddddddddddddddddd�Zd#� fd d
� Zed d � �Zed
d� �Z dd� Z
dd� Z dd� Z dd� Z
dd� Zedd� �Zedd � �Zed d � �Zd d � Zd!d"� Z� ZS )$� SoSUploadz�
This class is designed to upload files to a distribution
defined location. These files can be either sos reports,
sos collections, or other kind of files like: vmcores,
application cores, logs, etc.
z9Upload a file to a user or policy defined remote location� FN�autozhttps://s3.amazonaws.com)� upload_file�case_id� low_priorityZ
upload_url�upload_directoryZ upload_userZ upload_passZ
upload_methodZupload_no_ssl_verifyZupload_protocolZupload_s3_endpointZupload_s3_regionZupload_s3_bucketZupload_s3_access_keyZupload_s3_secret_keyZupload_s3_object_prefix�
upload_targetc s~ |s t � j|||� d| _nN|d | _|d | _|d | _|| _|| _|| _|| _ t
j d�| _ d| _|| _
| j� | _d | _d S )NTZoptions�policy�manifestZsos_uiF)�super�__init__� from_cmdline�optsr
r �parser�cmdline�argsZ _upload_file�loggingZ getLogger�ui_log�archive�load_upload_targets�upload_targetsr )�selfr r r Zin_placeZ hook_commonsr )� __class__� � /usr/lib/python3.6/__init__.pyr 5 s
zSoSUpload.__init__c C s: d|_ |jdd�}|jdddd� |jdd d
d d � |jd
d dd� |jdd dd� |jdd dd� |jdd ddd � |jdd dd� |jdd dd� |jdd d d� |jd d d d� |jd d d d� |jd!d d"d� |jd#d$d$d%d&gd'd(� |jd)d$d$d*d+d,d-gd.d(� |jd/d0d1d2d3� |jd4d5d6d7d8d5gd9d(� d S ):Nzsos upload FILE [options]zUpload Optionsz.These options control how upload manages filesr �FILEz The file or archive to upload)�metavar�helpz --case-idZstorer zspecify case identifier)�action�destr! z --upload-urlz&Upload the archive to specified server)�defaultr! z
--upload-userz Username to authenticate withz
--upload-passz Password to authenticate withz--upload-directoryr z$Specify upload directory for archivez--upload-s3-endpointz#Endpoint to upload to for S3 bucketz--upload-s3-regionz!Region to upload to for S3 bucketz--upload-s3-bucketz"Name of the S3 bucket to upload toz--upload-s3-access-keyz Access key for the S3 bucketz--upload-s3-secret-keyz Secret key for the S3 bucketz--upload-s3-object-prefixz Prefix for the S3 object/keyz--upload-methodr ZputZpostz HTTP method to use for uploading)r$ �choicesr! z--upload-protocolZhttpsZftpZsftpZs3z$Manually specify the upload protocolz--upload-no-ssl-verifyF�
store_truez'Disable SSL verification for upload url)r$ r" r! z--upload-target�localZredhatZ canonical�genericzmManually specify vendor-specific target for uploads. Supported options are:
redhat, canonical, generic, local)ZusageZadd_argument_group� add_argument)�clsr Z
upload_grpr r r �add_parser_optionsM sX
z SoSUpload.add_parser_optionsc C s |j d� |jd� d S )NzSoS Upload Detailed Helpz�The upload command is designed to upload already existing sos reports, as well as other files like logs and vmcores to a distribution specific location.)Z set_titleZadd_text)r* Zsectionr r r � display_help� s
zSoSUpload.display_helpc C s� d}| j jdt� d�� | j|�}| j j|� d}| jjs�yt|� | j jd� W nL tk
rr | jdd� Y n. t k
r� } z| j|d� W Y d d }~X nX d S )
zmPrint the intro message and prompts for a case ID if one is not
provided on the command line
a} This utility is used to upload files to a target location based either on a command line option or detecting the local distribution.
The archive to be uploaded may contain data considered sensitive and its content should be reviewed by the originating organization before being passed to any third party.
No configuration changes will be made to the system running this utility.
z
sos upload (version �)z,
Press ENTER to continue, or CTRL-C to quit
r zExiting on user cancel� � N)
r �infor Z_fmt_msgr Zbatch�input�KeyboardInterrupt�_exit� Exception)r Z
disclaimerZ intro_msg�prompt�er r r �intro� s
zSoSUpload.introc C s | j | j| j j| j jd�S )N)� cmdlineoptsr
r r )r r
r r )r r r r � get_commons� s zSoSUpload.get_commonsc C s
|| _ dS )zMSet common host data for the Upload targets
to reference
N)�commons)r r: r r r � set_commons� s zSoSUpload.set_commonsc C s� t | jj� �}xP| jj� D ]B}|j|� |j� r|jj}| jjd|� d�� || _ |j
� | _ qW | j s~| jd | _ | j j
� | _ | jj d| j � �� dS )a This sets the upload target and loads that target's options.
If no upload target is matched and no target is provided by
the user, then we abort.
If an upload target is provided in the command line,
this will not run.
zInstallation matches z , checking for upload targetsr( zUpload target set to N)
�listr �values�removeZcheck_distributionr �__name__r �debugr �nameZ upload_namer0 )r Zchecksr Zcnamer r r �determine_upload_target� s
z!SoSUpload.determine_upload_targetc C s\ ddl }i }xJ| j|jjd�D ]6}|d | j| j| jd�}|j| j� � |||j � < q W |S )zELoads all upload targets supported by the local installation
r N�targetsr/ )r r r )
Zsos.upload.targets�
_load_modulesZuploadrC r r r r; r9 Z
get_target_id)r �sosZsupported_upload_tgtsr Z target_classr r r r � s
z SoSUpload.load_upload_targetsc C s6 g }x,|j D ]"}tjj|�r |j| j||�� q W |S )z Helper to import upload targets)�__path__�os�path�isdir�extend�_find_modules_in_path)r* �packageZsubmod�modulesrH r r r rD � s
zSoSUpload._load_modulesc C sr g }t jj|�rnx\tt j|��D ]J}|jd�s0q d|kr:q t jj|�\}}d|� d|� �}|j| j|�� q W |S )aU Given a path and a module name, find everything that can be imported
and then import it
path - the filesystem path of the package
modulename - the name of the module in the package
E.G. a path of 'targets', and a modulename of 'redhat' equates to
importing sos.upload.targets.redhat
z.py�__z sos.upload.�.) rG rH �exists�sorted�listdir�endswith�splitextrJ �_import_modules)r* rH Z
modulenamerM ZpyfileZfname�_�modnamer r r rK � s
z SoSUpload._find_modules_in_pathc C s& t j|�}dd� tj|tj�D �}|S )z/Import and return all found classes in a modulec S s$ g | ] \}}|j jd �r||f�qS )zsos.upload.targets)�
__module__�
startswith)�.0rA r* r r r �
<listcomp> s z-SoSUpload._import_modules.<locals>.<listcomp>)� importlib�
import_module�inspectZ
getmembersZisclass)r* rW �modulerM r r r rU � s
zSoSUpload._import_modulesc C s( | j � | _| jd }|jr$| jj� d S )Nr8 )r9 r: r
r
Z_configure_low_priority)r � cmdline_optsr r r �pre_work
s
zSoSUpload.pre_workc C s� �y�| j � | jr6| j� | jj| _| jj| jd�| _| jj }|rX|dkrX| j
| | _ n| j � | j s|| j j
d� tjd� | j j | j� � y�tj| j�jdk�r@tjj| j��r&y,| j j| j� | j jtd| j� d��� W nD tk
�r" } z&| j j
td|� ��� tjd� W Y d d }~X nX n| j j
t| j� d ��� n| j j
td| j� d
��� W nD tk
�r� } z&| j j
td | j� d |� d
��� W Y d d }~X nX W n" tk
�r� | jdd� Y nX d S )N)r` r' z|No upload target set via command line options or autodetected.
Please specify one using the option --upload-target.
Exiting.r/ r zFile z uploaded successfullyzUpload attempt failed: z is not a file.z
is empty.zCannot upload z: � zExiting on user cancelr. ) ra r r7 r r r r
Zprompt_for_case_idZcaseidr r rB r �error�sys�exitr9 rG �stat�st_sizerH �isfileZupload_archiver0 rV r4 r2 r3 )r Zcmdline_target�errr6 r r r �execute sB
6zSoSUpload.execute)NNNFNN)r? rX � __qualname__�__doc__ZdescZ arg_defaultsr � classmethodr+ r, r7 r9 r; rB r rD rK rU ra rj �
__classcell__r r )r r r sB 5
r ) rG rd r r^ r\ Z
sos.componentr rE r rV r r r r r r �<module> s