([QhX)ddlZddlZddlmZddlmZmZddlmZddl m Z ddl m Z  ddlZdZGd d eZy#e$rdZYwxYw) N) UploadTarget) convert_bytesTIMEOUT_DEFAULT)DeviceAuthorizationClass) RHELPolicy)_sosTFceZdZdZdZdZdZdfd ZdZdZ e Z d Z dZ d Z d Zfd Zd ZdZdZdZddZdZdZdZdfd ZdZfdZxZS)RHELUploadTargetzVhttps://sso.redhat.com/auth/realms/redhat-external/protocol/openid-connect/auth/devicezPhttps://sso.redhat.com/auth/realms/redhat-external/protocol/openid-connect/tokenzRed Hat Upload TargetredhatNc*t||||y)N)parserargscmdline)super__init__)selfr rr __class__s ;/usr/lib/python3/dist-packages/sos/upload/targets/redhat.pyrzRHELUploadTarget.__init__!s T7Czhttps://api.access.redhat.comzsftp://sftp.access.redhat.composti@c<t|jdtS)z.Return true if we are running in a RHEL systempolicy) isinstancecommonsrrs rcheck_distributionz#RHELUploadTarget.check_distribution-s$,,x0*==rc`t|||jdj|_y)N cmdlineopts)rpre_workrupload_directory)r hook_commonsrs rrzRHELUploadTarget.pre_work1s' & $ ] ; L Lrc |jdjr$|jjt d|jdj s-|j |_|jjdyy)NrzVThe option --upload-user has been deprecated in favour of device authorization in RHELz&No case id provided, uploading to SFTP)r upload_userui_loginfo_case_id RH_SFTP_HOST upload_urlrs rprompt_for_upload_userz'RHELUploadTarget.prompt_for_upload_user7sl << & 2 2 KK  56 ||M*22"//DO KK  E F3rc|jdjr%|jjt dyy)NrzVThe option --upload-pass has been deprecated in favour of device authorization in RHEL)r upload_passr$r%r&rs rprompt_for_upload_passwordz+RHELUploadTarget.prompt_for_upload_passwordBs8 << & 2 2 KK  56  3rcJd|jdjd} |jr |jS|jdjr|jdjS|jdjdk(r |jS|jdjs7|jdj |jds |jS|j|S#t $r(}|jjd|Yd}~'..t??rc|jjdd}|jdjr|jdjd|}|jr*t j j|j|}|S)zThe RH SFTP server will only automatically connect file uploads to cases if the filename _starts_ with the case number r>r?rr&)upload_archive_namerFrr'r ospathjoin)rfnames r_get_sftp_upload_namez&RHELUploadTarget._get_sftp_upload_namesy((..s3B7 << & . .||M2::;1UGDE  GGLL!6!6>E rc|jjdd|jvrt |St s t dd}d}|js6 t|j|j}|j|_|jr,|jjd|j|j dz}d}|jr|j#} t%j&|| d }|j(d k(rFt+j,|j.d }t+j,|j.d }n|jj1d|j(d|j+|jj3dnddi} t%j&|t+j4| d } | j(d k(rRt+j,| j.} | d }| d }|jjt7d|dn9|jj1d| j(d| j+|r|rt |||St d#t $r3}dt|vr|jjdYd}~Gd}~wwxYw)zOverride the base upload_sftp to allow for setting an on-demand generated anonymous login for the RH SFTP server if a username and password are not given z//zPpython3-requests is not installed and is required for obtaining SFTP auth token.Nzend user deniedz:Device token authorization has been cancelled by the user.r@z/support/v2/sftp/token )rBrDusernametokenz$DEBUG: auth attempt failed (status: z): zUUnable to retrieve Red Hat auth token using provided credentials. Will try anonymous. isAnonymousT)datarDzUser z^ used for anonymous upload. Please inform your support engineer so they may retrieve the data.z)DEBUG: anonymous request failed (status: )userpasswordz1Could not retrieve valid or anonymous credentials)r(rFr6r upload_sftpREQUESTS_LOADEDr2r9rrHrIrJstrr$r%rKr3r;rLr status_codejsonloadstextdebugerrordumpsr&)rrerf_token_userrNr5urlretrBadataanonresprs rrgzRHELUploadTarget.upload_sftps    " "4 ( +43F3F3H H7&( (>? ? !! A3..''&.%>%>%@"    KK  !!%!;!;!= >@ A!99   113G--WbAC#% 388,Z8CHH-g6 !!:3??:K3xxzl$ !!7 #D)E==4::e+.>? KK  DCy!2!2 356 #//DOw-g6H 7sA A CAM G2,+E2 @ IMV 0rr )rXrksos.upload.targetsr sos.utilitiesrrsos.policies.authrsos.policies.distros.redhatrsosrr&rLrh ImportErrorr rrrsM +862O j|j Os<AA