File: //usr/lib/python3/dist-packages/sos/upload/__pycache__/__init__.cpython-310.pyc
o
([Qhn3 � @ sX d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z G dd� de�Z
dS )� N)� SoSComponent)�_sos)� __version__c s e Zd ZdZdZi dd�dd�dd�dd �d
d �d d �d d �d
d�dd�dd�dd�dd �dd �dd �dd �dd �dd �Z d3� fdd� Zedd � �Zed d � �Z d d � Z
d!d"� Z d#d$� Z d%d&� Z
d'd(� Zed)d*� �Zed+d,� �Zed-d.� �Zd/d0� Zd1d2� Z� ZS )4� 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_file� �case_id� low_priorityF�
upload_urlN�upload_directory� upload_user� upload_pass�
upload_method�auto�upload_no_ssl_verify�upload_protocol�upload_s3_endpointzhttps://s3.amazonaws.com�upload_s3_region�upload_s3_bucket�upload_s3_access_key�upload_s3_secret_key�upload_s3_object_prefix�
upload_targetc s~ |st � �|||� d| _n'|d | _|d | _|d | _|| _|| _|| _|| _ t
� d�| _ d| _|| _
| �� | _d | _d S )NT�options�policy�manifest�sos_uiF)�super�__init__� from_cmdline�optsr r �parser�cmdline�args� _upload_file�logging� getLogger�ui_log�archive�load_upload_targets�upload_targetsr )�selfr! r# r" �in_place� hook_commonsr( �� __class__� �5/usr/lib/python3/dist-packages/sos/upload/__init__.pyr 4 s
zSoSUpload.__init__c C s. d|_ |�dd�}|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$g d%�d&d'� |jd(d$g d)�d*d'� |jd+d,d-d.d/� |jd0d1g d2�d3d'� d S )4Nzsos upload FILE [options]zUpload Optionsz.These options control how upload manages filesr �FILEz The file or archive to upload)�metavar�helpz --case-id�storer zspecify case identifier)�action�destr4 z --upload-urlz&Upload the archive to specified server)�defaultr4 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 )r �put�postz HTTP method to use for uploading)r8 �choicesr4 z--upload-protocol)r �https�ftp�sftp�s3z$Manually specify the upload protocolz--upload-no-ssl-verifyF�
store_truez'Disable SSL verification for upload url)r8 r6 r4 z--upload-target�local)�redhat� canonical�genericrA zmManually specify vendor-specific target for uploads. Supported options are:
redhat, canonical, generic, local)�usage�add_argument_group� add_argument)�clsr! �
upload_grpr0 r0 r1 �add_parser_optionsL st ��
��������������
�z SoSUpload.add_parser_optionsc C s |� d� |�d� d S )NzSoS Upload Detailed Helpz�The 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_title�add_text)rH �sectionr0 r0 r1 � display_help� s
�zSoSUpload.display_helpc
C s� d}| j �dt� d�� | �|�}| j �|� d}| jjsSz
t|� | j �d� W d S ty: | �dd� Y d S t yR } z
| �|d� W Y d }~d S d }~ww 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 �_fmt_msgr �batch�input�KeyboardInterrupt�_exit� Exception)r+ �
disclaimer� intro_msg�prompt�er0 r0 r1 �intro� s
���zSoSUpload.introc C s | j | j| j j| j jd�S )N)� cmdlineoptsr r r )r r r r )r+ r0 r0 r1 � get_commons� s
�zSoSUpload.get_commonsc C s
|| _ dS )zMSet common host data for the Upload targets
to reference
N)�commons)r+ r` r0 r0 r1 � set_commons� s
zSoSUpload.set_commonsc C s� t | j�� �}| j�� D ]!}|�|� |�� r-|jj}| j�d|� d�� || _ |�
� | _ q | j s=| jd | _ | j �
� | _ | j� 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 targetsrD zUpload target set to N)
�listr* �values�remove�check_distributionr/ �__name__r' �debugr �name� upload_namerR )r+ �checksr �cnamer0 r0 r1 �determine_upload_target� s
�
�z!SoSUpload.determine_upload_targetc C sX ddl }i }| �|jjd�D ]}|d | j| j| jd�}|�| �� � |||� � < q|S )zELoads all upload targets supported by the local installation
r N�targetsrQ )r! r# r" )
�sos.upload.targets�
_load_modules�uploadrm r! r# r" ra r_ �
get_target_id)r+ �sos�supported_upload_tgtsr � target_classr0 r0 r1 r) � s �z SoSUpload.load_upload_targetsc C s2 g }|j D ]}tj�|�r|�| �||�� q|S )z Helper to import upload targets)�__path__�os�path�isdir�extend�_find_modules_in_path)rH �package�submod�modulesrw r0 r0 r1 ro � s
�zSoSUpload._load_modulesc C sn g }t j�|�r5tt �|��D ]%}|�d�sqd|v r qt j�|�\}}d|� d|� �}|�| �|�� q|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.�.) rv rw �exists�sorted�listdir�endswith�splitextry �_import_modules)rH rw �
modulenamer} �pyfile�fname�_�modnamer0 r0 r1 rz � s
z SoSUpload._find_modules_in_pathc
C sz |� d�d }z t|t� t� |g�}W n ty! } z|�d}~ww t�|tj�}|�� D ]
}|d dv r:|� |� q-|S )z/Import and return all found classes in a moduler � Nr )�DeviceAuthorizationClass�Upload�
RHELPolicy� UbuntuPolicy)
�split�
__import__�globals�locals� ImportError�inspect�
getmembers�isclass�copyrd )rH r� �mod_short_name�moduler\ r} �modr0 r0 r1 r� � s ��
�zSoSUpload._import_modulesc C s, | � � | _| jd }|jr| j�� d S d S )Nr^ )r_ r` r r �_configure_low_priority)r+ � cmdline_optsr0 r0 r1 �pre_work s
�zSoSUpload.pre_workc C s� | � � | jr| �� | jj| _| jj| jd�| _| jj }|r*|dkr*| j
| | _ n| � � | j s<| j �
d� t�d� | j � | �� � zlt�| j�jdkr�tj�| j�r�z| j �| j� | j �td| j� d��� W W d S ty� } z| j �
td|� ��� t�d� W Y d }~W d S d }~ww | j �
t| j� d ��� W d S | j �
td| j� d
��� W d S ty� } z| j �
td | j� d |� d
��� W Y d }~d S d }~ww )N)r� rA z|No upload target set via command line options or autodetected.
Please specify one using the option --upload-target.
Exiting.rQ r zFile z uploaded successfullyzUpload attempt failed: z is not a file.z
is empty.zCannot upload z: � )r� r r] r r r( r �prompt_for_case_id�caseidr r* rl r' �error�sys�exitr_ rv �stat�st_sizerw �isfile�upload_archiverR r� rX )r+ �cmdline_target�errr\ r0 r0 r1 �execute sH
� �
��� .��zSoSUpload.execute)NNNFNN)rf �
__module__� __qualname__�__doc__�desc� arg_defaultsr � classmethodrJ rN r] r_ ra rl r) ro rz r� r� r� �
__classcell__r0 r0 r. r1 r st �������� �
� � �
������
4
r ) rv r� r% r� �
sos.componentr rr r r� r r r0 r0 r0 r1 �<module> s