([Qhn3^ddlZddlZddlZddlZddlmZddlmZddlm Z GddeZ y)N) SoSComponent)_sos) __version__ceZdZdZdZidddddddd d d d d d d d ddddddddd dd dd dd dd dd Z d&fd ZedZedZ dZ dZ dZ dZ d Zed!Zed"Zed#Zd$Zd%ZxZS)' 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 upload_filecase_id low_priorityF upload_urlNupload_directory upload_user upload_pass upload_methodautoupload_no_ssl_verifyupload_protocolupload_s3_endpointzhttps://s3.amazonaws.comupload_s3_regionupload_s3_bucketupload_s3_access_keyupload_s3_secret_keyupload_s3_object_prefix upload_targetc8|st||||d|_nb|d|_|d|_|d|_||_||_||_||_ tjd|_ d|_||_ |j|_d|_y)NToptionspolicymanifestsos_uiF)super__init__ from_cmdlineoptsrrparsercmdlineargs _upload_filelogging getLoggerui_logarchiveload_upload_targetsupload_targetsr)selfr$r&r%in_place hook_commonsr+ __class__s 5/usr/lib/python3/dist-packages/sos/upload/__init__.pyr!zSoSUpload.__init__4s G VT7 3 $D  %Y/DI&x0DK(4DM DK"DLDI 'D !++H5DK %D "DL"668!cd|_|jdd}|jddd|jdd d d |jd dd|jddd|jddd|jdd dd |jddd|jddd|jddd|jddd|jddd |jd!dd"|jd#d$gd%d&'|jd(d$gd)d*'|jd+d,d-d./|jd0d1gd2d3'y)4Nzsos upload FILE [options]zUpload Optionsz.These options control how upload manages filesrFILEzThe file or archive to upload)metavarhelpz --case-idstorer zspecify case identifier)actiondestr7z --upload-urlz&Upload the archive to specified server)defaultr7z --upload-userzUsername to authenticate withz --upload-passzPassword 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-keyzAccess key for the S3 bucketz--upload-s3-secret-keyzSecret key for the S3 bucketz--upload-s3-object-prefixzPrefix for the S3 object/keyz--upload-methodr)rputpostz HTTP method to use for uploading)r;choicesr7z--upload-protocol)rhttpsftpsftps3z$Manually specify the upload protocolz--upload-no-ssl-verifyF store_truez'Disable SSL verification for upload url)r;r9r7z--upload-targetlocal)redhat canonicalgenericrDzmManually specify vendor-specific target for uploads. Supported options are: redhat, canonical, generic, local)usageadd_argument_group add_argument)clsr$ upload_grps r2add_parser_optionszSoSUpload.add_parser_optionsLs2 ..  <   v%D  F G)%>  @%M  O%D  F%D  F 4W%7%K  M  6%J  L 4d%H  J 4d%I  K 8$%C  E 8$%C  E ;T%C  E 16(?%G  I  3V(N%K  M  8%'3%N  P  17)- '7 9r3cH|jd|jdy)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.) set_titleadd_text)rKsections r2 display_helpzSoSUpload.display_helps$45 3 r3cd}|jjdtd|j|}|jj|d}|jj s( t ||jjdy y #t$r|jddYy t$r}|j|dYd }~y d }~wwxYw) 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 cancelN) r*infor_fmt_msgr#batchinputKeyboardInterrupt_exit Exception)r. disclaimer intro_msgpromptes r2introzSoSUpload.intros   1+a@AMM*-  #Ayy !f    $% : 3S9 ! 1a   !s+&BC0C8CCc|j|j|jj|jjdS)N) cmdlineoptsrr r )r#rr r )r.s r2 get_commonszSoSUpload.get_commonss499kkyy(( $ : :   r3c||_y)zMSet common host data for the Upload targets to reference N)commons)r.rgs r2 set_commonszSoSUpload.set_commonss  r3c<t|jj}|jjD]u}|j||j s%|j j }|jjd|d||_ |j|_ w|js3|jd|_ |jj|_ |jjd|jy)aThis 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 targetsrGzUpload target set to N) listr-valuesremovecheck_distributionr1__name__r*debugrname upload_namerW)r.checksrcnames r2determine_upload_targetz!SoSUpload.determine_upload_targetsd))0023!00779 8M MM- (//1%//88 !!$9%A4#45%2"#0#5#5#7  8!!!%!4!4Y!?D #11668D  +D,<,<+=> @r3c$ddl}i}|j|jjdD]`}|d|j|j |j }|j|j|||j<b|S)zELoads all upload targets supported by the local installation rNtargetsrV)r$r&r%) sos.upload.targets _load_modulesuploadrvr$r&r%rhre get_target_id)r.sossupported_upload_tgtsr target_classs r2r,zSoSUpload.load_upload_targetss " "!// 0B0BIN OM+=+#';;!%$(LLL  $ $T%5%5%7 8BN !,"<"<"> ? O%$r3cg}|jD]C}tjj|s#|j |j ||E|S)zHelper to import upload targets)__path__ospathisdirextend_find_modules_in_path)rKpackagesubmodmodulesrs r2rxzSoSUpload._load_modulessP$$ HDww}}T"s88vFG Hr3cPg}tjj|rttj|D]c}|j dsd|vrtjj |\}}d|d|}|j|j|e|S)aUGiven 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..) rrexistssortedlistdirendswithsplitextr_import_modules)rKr modulenamerpyfilefname_modnames r2rzSoSUpload._find_modules_in_paths 77>>$  D!12 =u-6>77++F3q' |1UG<s227;< =r3c8|jdd} t|tt|g}t j |t j}|jD]}|ddvs |j||S#t$r}|d}~wwxYw)z/Import and return all found classes in a modulerNr)DeviceAuthorizationClassUpload RHELPolicy UbuntuPolicy) split __import__globalslocals ImportErrorinspect getmembersisclasscopyrl)rKrmod_short_namemodulerarmods r2rzSoSUpload._import_moduless!s+A. FH~>NOF$$VW__=<<> $C1v$$ s#  $ G sB BBBc|j|_|jd}|jr|jj yy)Nrd)rergr r_configure_low_priority)r. cmdline_optss r2pre_workzSoSUpload.pre_works>'') ||M2  $ $ KK / / 1 %r3c |j|jrV|j|jj|_|j j|j|_|jj}|r|dk7r|j||_ n|j|js0|jjdtjd|jj|j! t#j$|j j&dkDrt"j(j+|j rY |jj-|j |jj/t1d|j dy|jjt1|j d y|jjt1d|j d y#t2$rF}|jjt1d|tjdYd}~yd}~wwxYw#t2$r?}|jjt1d |j d |d Yd}~yd}~wwxYw)N)rrDz|No upload target set via command line options or autodetected. Please specify one using the option --upload-target. Exiting.rVrzFile z uploaded successfullyzUpload attempt failed: z is not a file.z is empty.zCannot upload z:  )rr"rbr#rr+rprompt_for_case_idcaseidrr-rtr*errorsysexitrerstatst_sizerisfileupload_archiverWrr])r.cmdline_targeterrras r2executezSoSUpload.executes     JJL9900DL++88!YY9DK00 n7!%!4!4^!DD   ( ( *!! KK    HHQK ##D$4$4$67 Hwwt||$,,q077>>$,,/$**99$,,G ((dll^3IJKKK%%a4<<.(H&IJ !!!eDLL>$D"EF %$ ))!.EcU,K*LM $ H KK  a.b1 EF G G HsJAJ1AH. 1J;2J. I=7rs, &kH kHr3