File: //lib/python3.6/site-packages/sos/upload/targets/__pycache__/redhat.cpython-36.pyc
3
mk�h�( � @ 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
e�ZdS ) � N)� UploadTarget)�
convert_bytes�TIMEOUT_DEFAULT)�DeviceAuthorizationClass)�
RHELPolicy)�_sosTFc s� e Zd ZdZdZdZdZd'� fdd� ZdZd Z e Z
d
Z dZ d Z
d d
� Z� fdd�Zdd� Zdd� Zdd� Zdd� Zd(dd�Zdd � 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)�parser�args�cmdline)�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.com�posti @c C s t | jd t�S )z.Return true if we are running in a RHEL systemZpolicy)�
isinstance�commonsr )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_user�ui_log�info�_�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_api�er 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 jd�d || j� fi}t| j| j�}|j� | _| jj d| j
� � �� t j | j
� || j� |td�S )z�If 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 )�files�headers�verify�timeout���)�name�split�_get_upload_headersr �client_identifier_url�token_endpoint�get_access_tokenr% r r �get_upload_url_string�requestsr r$ r& r )r �archiver, r* �RHELAuthr r r �_upload_https_postb s
z#RHELUploadTarget._upload_https_postc C s | j � j| j�rddd�S 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| j�rdS | j � j| j�r(dS | j| j�S )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 jd�d }| jd jr2| jd j� d|� �}| jrHtjj| j|�}|S )z�The 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 �os�path�join)r Zfnamer r r �_get_sftp_upload_name� s z&RHELUploadTarget._get_sftp_upload_namec
s� | j jd�d | j� kr"t� j� S ts.td��d}d}| js�yt| 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}| j�r8| j� } tj|| d d
�}|jd k�r tj|j�d }tj|j�d
}n*| j jd|j� d|j� � �� | j jd� n~ddi}
tj|tj|
�d d�} | jd k�r�tj| j�} | d }| d
}| j j
td|� d��� n | j jd| j� d| j� � �� |�r�|�r�t� j||d�S td��dS )z�Override 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- �� Zusername�tokenz$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: )�user�passwordz1Could not retrieve valid or anonymous credentials) r r0 r$ r � upload_sftp�REQUESTS_LOADEDr! r% r r2 r3 �strr r r4 r5 r" r& r6 r Z status_code�json�loads�text�debug�error�dumpsr )
r rC rD Z_tokenZ_userr8 r# Zurl�retr+ 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| j�r | j|� t� j|�}W nh tk
r� } zLd}| jj| j�sP� | 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 �<module> s