3
ܓi3 @ s` d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlm
Z
G dd deZ dS ) N) SoSComponent)_sos) __version__c s e Zd ZdZdZddddddddddddddddddZd# 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 FNautozhttps://s3.amazonaws.com) upload_filecase_id low_priorityZ
upload_urlupload_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 )NTZoptionspolicymanifestZsos_uiF)super__init__ from_cmdlineoptsr
r parsercmdlineargsZ _upload_fileloggingZ getLoggerui_logarchiveload_upload_targetsupload_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)metavarhelpz --case-idZstorer zspecify case identifier)actiondestr! 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-targetlocalZredhatZ canonicalgenericzmManually 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 HelpzThe 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}| jjsyt| | 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 ZbatchinputKeyboardInterrupt_exit Exception)r Z
disclaimerZ intro_msgprompter r r intro s
zSoSUpload.introc C s | j | j| j j| j jdS )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 valuesremoveZcheck_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|jjdD ]6}|d | j| j| jd}|j| j |||j < q W |S )zELoads all upload targets supported by the local installation
r Ntargetsr/ )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__ospathisdirextend_find_modules_in_path)r* packageZsubmodmodulesrH r r r rD s
zSoSUpload._load_modulesc C sr g }t jj|rnx\tt j|D ]J}|jds0q 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 existssortedlistdirendswithsplitextrJ _import_modules)r* rH Z
modulenamerM ZpyfileZfname_modnamer r r rK s
z SoSUpload._find_modules_in_pathc C s& t j|}dd tj|tjD }|S )z/Import and return all found classes in a modulec S s$ g | ] \}}|j jd r||fqS )zsos.upload.targets)
__module__
startswith).0rA r* r r r
s z-SoSUpload._import_modules..) importlib
import_moduleinspectZ
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 ytj| jjdkr@tjj| jr&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 errorsysexitr9 rG statst_sizerH isfileZupload_archiver0 rV r4 r2 r3 )r Zcmdline_targeterrr6 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 s