3
'j( @ s d dl Z d dlZd dlmZ d dlmZmZ d dlmZ d dl m
Z
d dl m Z
yd dlZdZW n ek
rx dZY nX G d d
d
eZdS ) N) UploadTarget)
convert_bytesTIMEOUT_DEFAULT)DeviceAuthorizationClass)
RHELPolicy)_sosTFc s e Zd ZdZdZdZdZd' fdd ZdZd Z e Z
d
Z dZ d Z
d d
Z fddZdd Zdd Zdd Zdd Zd(ddZdd Zd d Zd d Zd) fd!d" Zd#d$ Z fd%d&Z ZS )*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 TargetZredhatNc s t j|||d d S )N)parserargscmdline)super__init__)selfr r
r ) __class__ /usr/lib/python3.6/redhat.pyr
! s zRHELUploadTarget.__init__z https://api.access.redhat.comz sftp://sftp.access.redhat.composti @c C s t | jd tS )z.Return true if we are running in a RHEL systemZpolicy)
isinstancecommonsr )r r r r check_distribution- s z#RHELUploadTarget.check_distributionc s t j| | jd j| _d S )N cmdlineopts)r pre_workr upload_directory)r Z hook_commons)r r r r 1 s zRHELUploadTarget.pre_workc C s@ | j d jr | jjtd | j d js<| j| _| jjd d S )Nr zVThe option --upload-user has been deprecated in favour of device authorization in RHELz&No case id provided, uploading to SFTP)r Z upload_userui_loginfo_case_id RH_SFTP_HOST
upload_url)r r r r prompt_for_upload_user7 s
z'RHELUploadTarget.prompt_for_upload_userc C s | j d jr | jjtd d S )Nr zVThe option --upload-pass has been deprecated in favour of device authorization in RHEL)r Z upload_passr r r )r r r r prompt_for_upload_passwordB s z+RHELUploadTarget.prompt_for_upload_passwordc C s d| j d j d}yP| jr"| jS | j d jr:| j d jS | j d jdkrP| jS | j d jsb| jS W n4 tk
r } z| jjd| W Y d d }~X nX | j | S )Nz/support/v1/cases/r z /attachmentsZsftpz=There was a problem while setting the remote upload target: ) r r r Zupload_protocolr Exceptionr r RH_API_HOST)r Z rh_case_apier r r get_upload_urlJ s
z RHELUploadTarget.get_upload_urlc C s d| j }d|iS )NzBearer Z
Authorization)
_device_token)r Zstr_authr r r _get_upload_https_auth^ s z'RHELUploadTarget._get_upload_https_authTc C sh d|j jdd || j fi}t| j| j}|j | _| jj d| j
t j | j
|| j |tdS )zIf upload_https() needs to use requests.post(), use this method.
Policies should override this method instead of the base upload_https()
:param archive: The open archive file object
file/ z/Device authorized correctly. Uploading file to )filesheadersverifytimeout)namesplit_get_upload_headersr client_identifier_urltoken_endpointget_access_tokenr% r r get_upload_url_stringrequestsr r$ r& r )r archiver, r* RHELAuthr r r _upload_https_postb s
z#RHELUploadTarget._upload_https_postc C s | j j| jrdddS i S )NZfalsezno-cache)Z isPrivatez
cache-control)r$
startswithr" )r r r r r1 { s
z$RHELUploadTarget._get_upload_headersc C s4 | j j| jrdS | j j| jr(dS | j| jS )NzRed Hat Customer PortalzRed Hat Secure FTP)r$ r: r" r Z_get_obfuscated_upload_urlr )r r r r r5 s
z&RHELUploadTarget.get_upload_url_stringc C sL | j jdd }| jd jr2| jd j d| }| jrHtjj| j|}|S )zThe RH SFTP server will only automatically connect file uploads to
cases if the filename _starts_ with the case number
r( r) r r r. )Zupload_archive_namer0 r r r ospathjoin)r Zfnamer r r _get_sftp_upload_name s z&RHELUploadTarget._get_sftp_upload_namec
s | j jdd | j kr"t j S ts.tdd}d}| jsyt| j | j
}W n: tk
r } z dt |krx| j j
d W Y dd}~X n X |j | _| jr| j j
d| j | jd }d}| jr8| j } tj|| d d
}|jd kr tj|jd }tj|jd
}n*| j jd|j d|j | j jd n~ddi}
tj|tj|
d d} | jd krtj| j} | d }| d
}| j j
td| d n | j jd| j d| j |r|rt j||dS tddS )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//r) 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.z/Device authorized correctly. Uploading file to z/support/v2/sftp/token
)r+ r- Zusernametokenz$DEBUG: auth attempt failed (status: z): zUUnable to retrieve Red Hat auth token using provided credentials. Will try anonymous.Z isAnonymousT)datar- zUser 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 r0 r$ r upload_sftpREQUESTS_LOADEDr! r% r r2 r3 strr r r4 r5 r" r& r6 r Z status_codejsonloadstextdebugerrordumpsr )
r rC rD Z_tokenZ_userr8 r# Zurlretr+ ZadataZanonZresp)r r r rE sV
z RHELUploadTarget.upload_sftpc C s@ t jj|}|| jkr<| jjtdt| j d | j| _ d S )NzLSize of archive is bigger than Red Hat Customer Portal limit for uploads of z via sos http upload.
)
r; r< getsize_max_size_requestr Zwarningr r r r )r r7 sizer r r check_file_too_big s
z#RHELUploadTarget.check_file_too_bigc s y*| j j| jr | j| t j|}W nh tk
r } zLd}| jj| jsP | jj t
d| d| j | j | _t j|}W Y dd}~X nX |S )zOverride the base upload_archive to provide for automatic failover
from RHCP failures to the public RH dropbox
Fz0Upload to Red Hat Customer Portal failed due to z . Trying N) r$ r: r" rR r upload_archiver! r r rL r r )r r7 Zuploadedr# )r r r rS s
z RHELUploadTarget.upload_archive)NNN)T)NN) __name__
__module__ __qualname__r2 r3 Zupload_target_nameZupload_target_idr
r" r Z _upload_urlZ_upload_methodr% rP r r r r r$ r& r9 r1 r5 r> rE rR rS
__classcell__r r )r r r s0
Kr )r; rH Zsos.upload.targetsr Z
sos.utilitiesr r Zsos.policies.authr Zsos.policies.distros.redhatr Zsosr r r6 rF ImportErrorr r r r r s