0x1949 Team - FAZEMRX - MANAGER
Edit File: shortcuthandler.cpython-310.pyc
o U�7eea � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZm Z m Z d dl m Z d dl m Z d dlmZ d dlmZ e �� dZG d d � d e�ZG d d � d e�ZG d d� de�ZdS )� N)� get_distro)� SourceEntry� SourcesList�CollapsedSourcesList)�suppress)�copy)�gettext)�urlparsez=gpg -q --no-options --no-default-keyring --batch --keyring %sc @ s� e Zd ZdZdTdd�Zedd� �Zedd � �Zed d � �Zed d � �Z e dd� �Z e dd� �Z e dd� �Z e dd� �Ze dd� �Ze dd� �ZdUdd�Ze d d � �Ze d d � �Zd d!� Zd"d#� Zd$d%� Zd&d'� Ze d(d)� �Ze d*d+� �Ze d,d-� �Ze d.d/� �Zd0d1� Z d2d3� Z e d4d5� �Z e d6d7� �Z e d8d9� �Z e d:d;� �Z!e d<d=� �Z"d>d?� Z#d@dA� Z$e dBdC� �Z%e dDdE� �Z&e dFdG� �Z'e dHdI� �Z(e dJdK� �Z)dLdM� Z*dUdNdO�Z+dUdPdQ�Z,dRdS� Z-dS )V�ShortcutHandleraC Superclass for shortcut handler implementations. This provides a way to take a apt repository reference, in various forms, and write the specific apt configuration to local files. This also can remove previously written configuration from local files. This class and any subclasses should never modify any main apt configuration files, only specifically named files in '.d' subdirs (e.g. sources.list.d, etc) should be modified. The only exception to that rule is adding or removing sourceslist lines or components of existing source entries. NFc K sT || _ |pg | _|| _t� | _|p| jj| _|| _|| _d | _d | _ d | _ d | _ d S �N) �shortcut� components� enable_sourcer �distro�codename�pocket�dry_run� _source_entry� _filebase� _username� _password)�selfr r r r r r �kwargs� r �D/usr/lib/python3/dist-packages/softwareproperties/shortcuthandler.py�__init__8 s zShortcutHandler.__init__c C s t |�}|jo |jS )z(Return if the uri is in valid uri format)r �scheme�netloc��cls�uri�parsedr r r � is_valid_uriJ s z ShortcutHandler.is_valid_uric C s$ t |�}|j|j�d�d d��� S )z6Return the uri with the username and password stripped�@� �r )r �_replacer � rpartition�geturlr r r r �uri_strip_authP s z ShortcutHandler.uri_strip_authc C s. t | �|��}d|||jf }|j|d��� S )z6Return the uri with the username and password includedz%s:%s@%sr% )r r) r r& r( )r r �username�passwordr! r r r r �uri_insert_authX s z ShortcutHandler.uri_insert_authc C s� d}|d7 }z5t �� �'}|d|� �7 }t|t�s |�� }tj|�� d|tjd�j � � }W d � n1 s5w Y W n tj yX } zt t d�| � g W Y d}~S d}~ww z dd � |�� D �W S tyv t t d �| � g Y S w ) z�Return an array of fingerprint(s) for provided key(s). The 'keys' parameter should be in text (str) or binary (bytes) format; it is converted to bytes if needed, and then passed to the 'gpg' program. z6gpg -q --no-options --no-keyring --batch --with-colonsz& --with-fingerprint --with-fingerprint� --homedir T)�check�input�stdoutNz,Warning: gpg error while processing keys: %sc S s$ g | ]}|� d �r|�d�d �qS )�fpr�:� )� startswith�split��.0�lr r r � <listcomp>v s $ z0ShortcutHandler.fingerprints.<locals>.<listcomp>z Warning: invalid gpg output: %s)�tempfile�TemporaryDirectory� isinstance�bytes�encode� subprocess�runr5 �PIPEr0 �decode�CalledProcessError�print�_� splitlines�KeyError)r �keys�cmd�homedirr0 �er r r � fingerprints_ s2 �� ���� �z ShortcutHandler.fingerprintsc C s" t d| �� jd�| �� j�f �S )Nz'Archive for codename: %s components: %s�,)rE r �dist�join�comps�r r r r � description{ s ��zShortcutHandler.descriptionc C � | j S r )� archive_linkrQ r r r �web_link� s zShortcutHandler.web_linkc C s | � � jS r )r r rQ r r r rT � s z ShortcutHandler.archive_linkc C s | j r d| j| j f S | jS )Nz%s-%s)r r rQ r r r rN � s zShortcutHandler.distc C � | j jS )z*Text indicating a binary-type SourceEntry.)r � binary_typerQ r r r rW � � zShortcutHandler.binary_typec C rV )z*Text indicating a source-type SourceEntry.)r � source_typerQ r r r rY � rX zShortcutHandler.source_typec C s| | j std��t| j �}|s|S || jkr |�d� | j|_n|| jkr/|�| j� | j|_ntd| ��t t |�|j d�S )a" Get the SourceEntry representing this archive/shortcut. This should never include any authentication data; if required, the username and password should only be available from the username and password properties, as well as from the netrcparts_content property. If pkgtype is provided, it must be either binary_type or source_type, in which case this returns a SourceEntry with the requested type. If pkgtype is not specified, this returns a SourceEntry with an implementation-dependent type (in most cases, implementations should default to binary_type). Note that the default SourceEntry will be returned without modification, and the implementation will determine if it is enabled or disabled; while the source-type SourceEntry will be enabled or disabled based on self.enable_source. The binary-type SourceEntry will always be enabled. The SourceEntry 'file' field should always be set to the value of sourceparts_file. z3Implementation class did not set self._source_entryTzInvalid pkgtype: %s��file) r �NotImplementedErrorr rW � set_enabled�typerY r � ValueErrorr �strr[ )r �pkgtyperK r r r r � s zShortcutHandler.SourceEntryc C rS )aW Return the username used for authentication If authentication is used, return the username; otherwise return None. By default, this returns the private variable self._username, which defaults to None. Subclasses should override this method and/or set self._username if they have authentication data. )r rQ r r r r* � � zShortcutHandler.usernamec C rS )aW Return the password used for authentication If authentication is used, return the password; otherwise return None. By default, this returns the private variable self._password, which defaults to None. Subclasses should override this method and/or set self._password if they have authentication data. )r rQ r r r r+ � rb zShortcutHandler.passwordc C s | � � | �� | �� dS )z�Save all data for this shortcut to file(s). This writes everything to the relevant files. By default, it calls add_source(), add_key(), and add_login(). Subclasses should override it if other actions are required. N)� add_source�add_key� add_loginrQ r r r �add� s zShortcutHandler.addc C s | � � | �� dS )a< Remove all data for this shortcut from file(s). This removes everything from the relevant files. By default, it only calls remove_source() and remove_login(). Subclasses should override it if other actions are required. Note that by default is does not call remove_key(). N)� remove_source� remove_loginrQ r r r �remove� s zShortcutHandler.removec C s | � | j�}| � | j�}| j}t� }t|�}|�|�}|r*ttd�|j |j f � n|� |�}|j |j kr?ttd�|j � n/|j rRt d�j |_ ttd�|j � n |j rbttd�|j |j f � n ttd�|j |j f � |}|j |_ |�|�}|r�ttd�|j |j f � n|j ||d�}|j |j kr�ttd�|j � n |j r�ttd�|j |j f � n ttd�|j |j f � |}| j�st|j |j g�D ]8}tj�|��s|�| j�r�tj�| j�s�t�| jd� t|d �� t�||� W d � n1 s�w Y q�|�� d S d S ) aO Add the apt SourceEntries. This uses SourcesList to add the binary-type and source-type SourceEntries. If the SourceEntry matches a known apt template, this will ignore the sourceparts_file and instead place the SourceEntries into the main/default sources.list file. Otherwise, this will add the SourceEntries into the sourceparts_file. If either the binary-type or source-type entry exist in the current SourcesList, the existing entries are updated instead of placing the entries in the sourceparts_file. z Found existing %s entry in %sz+Updating existing entry instead of using %s� z!Archive has template, updating %sz Adding disabled %s entry to %szAdding %s entry to %s)�after�� �wN)r rW rY �sourceparts_moder r � get_entryrD rE r^ r[ � add_entry�template�disabledr �set�os�path�existsr4 �sourceparts_path�mkdir�open�chmod�save)r �binentry�srcentry�mode� sourceslist� collapsedlist�newentry� entryfiler r r rc � sT � �� �zShortcutHandler.add_sourcec s t � }t|�}� �� j�}� �� j�}|�d� |�|�r2ttd�|j |j f � |� |j dd�� |�d� |�|�rQttd�|j |j f � |� |j dd�� � fdd�|D �}dd� |D �sz|D ]}|j stttd�|j |j f � |�|� qc� js�|jdd� d S d S ) a~ Remove the apt SourceEntries. This uses SourcesList to remove the binary-type and source-type SourceEntries. This must disable the corresponding SourceEntries, from whatever file(s) they are located in. This must not disable more than matches, e.g. if the existing SourceEntry line contains more components this must edit the existing line to remove this SourceEntry's component(s). After disabling all matching SourceEntries, if the sourceparts_file is empty or contains only invalid and/or disabled SourceEntries, this may remove the sourceparts_file. TzDisabling %s entry in %s)rr c s g | ] }|j � jkr|�qS r )r[ �sourceparts_file)r7 �srQ r r r9 T � z1ShortcutHandler.remove_source.<locals>.<listcomp>c S s g | ] }|j s|js|�qS r )�invalidrr )r7 rK r r r r9 U r� z"Removing disabled %s entry from %s)ri N)r r r rW rY r] � has_entryrD rE r^ r[ rp r& r� ri r r{ )r r r� r| r} � file_entriesrK r rQ r rg 5 s* �z ShortcutHandler.remove_sourcec C � t j�d�S )zAReturn result of apt_pkg.config.find_dir("Dir::Etc::sourceparts")zDir::Etc::sourceparts��apt_pkg�config�find_dirrQ r r r rw _ � z ShortcutHandler.sourceparts_pathc C s | j d| jd�S )a$ Get the sources.list.d filename, without the leading path. By default, this combines the filebase with the codename, and uses a extension of 'list'. This is different than the trustedparts or netrcparts filenames, which use only the filebase plus extension. �list��suffix)�_filebase_to_filenamer rQ r r r �sourceparts_filenamed s z$ShortcutHandler.sourceparts_filenamec C � | � | j| j�S )a� Get the sources.list.d absolute-path filename. Note that the add_source() function will not use this file if this shortcut's SourceEntry matches a known apt template; instead the entries will be placed in the main sources.list file. Also, if the SourceEntry already exists in the SourcesList, it will be edited in place, instead of using this file. See add_source() for more details. )�_filename_to_filerw r� rQ r r r r� n s z ShortcutHandler.sourceparts_filec C � dS )z�Mode of sourceparts file. Note that add_source() will only use this mode if it creates a new file for sourceparts_file; if the file already exists or if the SourceEntry is saved in a different file, this mode is not used. � r rQ r r r rn z s z ShortcutHandler.sourceparts_modec C s^ t | j| jf�s dS | j}| j}t|t�s|�� }| �|�}ttd�|d� |�f � t | }d}| j s�t j �|�sn|�| j�rOt j �| j�sOt j| jdd� | jrnt|d�� t �|| j� W d � n1 siw Y z-t�� � }|d|� d |� �7 }tj|�� d |d � W d � W dS 1 s�w Y W dS tjy� } zt|��d}~ww dS ) a[ Add the GPG key(s) corresponding to this repo. By default, if self.trustedparts_content contains content, and self.trustedparts_file points to a file, the key(s) will be added to the file. If the file does not yet exist, and self.trustedparts_mode is set, the file will be created with that mode. Nz$Adding key to %s with fingerprint %srM z--importrl �r~ �wbr- � T)r. r/ ) �all�trustedparts_file�trustedparts_contentr< r= r> rL rD rE rO �GPG_KEYRING_CMDr rt ru rv r4 �trustedparts_pathrx �trustedparts_modery rz r: r; r? r@ r5 rC �ShortcutException)r �destrH �fprI �actionrJ rK r r r rd � s< � � &����zShortcutHandler.add_keyc C s6 t | j| jf�s dS | j}| �| j�}tj�|�sdS ttd�|d� |�f � t | }dd� |� }| j s�z)t � � �}|d|� d|� �7 }tj|�� dd� W d � n1 sYw Y W n tjyp } zt|��d}~ww t|d ��}| �|�� � }W d � n1 s�w Y |r�t�|� dS dS dS ) a� Remove the GPG key(s) corresponding to this repo. By default, if self.trustedparts_content contains content, and self.trustedparts_file points to a file, the key(s) will be removed from the file. If the file contains no more keys after removal, the file will be removed. This does not consider other files; multiple repositories may use the same signing key. This only modifies/removes self.trustedparts_file. Nz(Removing key from %s with fingerprint %srM z--delete-keys %sr� r- T)r. �rb)r� r� r� rL rt ru rv rD rE rO r� r r: r; r? r@ r5 rC r� ry �readri ) r r� r� rI r� rJ rK �f�emptyr r r � remove_key� s6 ���� �� zShortcutHandler.remove_keyc C r� )zBReturn result of apt_pkg.config.find_dir("Dir::Etc::trustedparts")zDir::Etc::trustedpartsr� rQ r r r r� � r� z!ShortcutHandler.trustedparts_pathc C � | � d�S )z9Get the trusted.gpg.d filename, without the leading path.�gpg�r� rQ r r r �trustedparts_filename� � z%ShortcutHandler.trustedparts_filenamec C r� )z-Get the trusted.gpg.d absolute-path filename.)r� r� r� rQ r r r r� � � z!ShortcutHandler.trustedparts_filec C r� )z&Content to put into trusted.gpg.d fileNr rQ r r r r� � � z$ShortcutHandler.trustedparts_contentc C r� )zMode of trustedparts filer� r rQ r r r r� � r� z!ShortcutHandler.trustedparts_modec s� t | j| jf�s dS | j}| j}tj�|� }d}|sjt|d��}dd� |�� D �� W d � n1 s4w Y tt �� � d d }W d � n1 sNw Y t � fdd�|� � D ��rjt t d�| � dS t t d �| � | j s�|r�|�| j�r�tj�| j�s�tj| jd d � | jr�t|d �� t�|| j� W d � n1 s�w Y t|d ��}|dkr�|�d� |�| j� W d � dS 1 s�w Y dS dS )z�Add the login credentials corresponding to this repo. By default, if self.netrcparts_content contains content, and self.netrcparts_file points to a file, the file will be created and content placed into it. N� �rc S � g | ]}|� � �qS r ��stripr6 r r r r9 � � z-ShortcutHandler.add_login.<locals>.<listcomp>���c s g | ]}|� � � v �qS r r� r6 ��linesr r r9 s z!Authentication data already in %sz Adding authentication data to %srl r� rm �a)r� �netrcparts_file�netrcparts_contentrt ru rv ry � readlinesr rG rF rD rE r r4 �netrcparts_pathrx �netrcparts_moderz �write)r r� �content�newfile� finalcharr� r r� r re � sB � � � � "��zShortcutHandler.add_loginc C sH t | j| jf�s dS | j}tdd� | j�� D ��}tj�|�s!dS t|d��}tdd� |� � D ��}W d � n1 s<w Y ||@ sNt t d�| � n*t t d�| � | j sxt|d��}|� d �|| �� W d � n1 ssw Y | j s�t|d��}|�� �� }W d � n1 s�w Y |r�t�|� dS dS dS ) a� Remove the login credentials corresponding to this repo. By default, if self.netrcparts_content contains content, and self.netrcparts_file points to a file, the content will be removed from the file. If the file is empty (other than whitespace) after removal, the file will be removed. This does not consider other files; this only modifies/removes self.netrcparts_file. Nc S r� r r� r6 r r r r9 % r� z0ShortcutHandler.remove_login.<locals>.<listcomp>r� c S r� r r� r6 r r r r9 + r� z'Authentication data not contained in %sz$Removing authentication data from %srm r� )r� r� r� rs rF rt ru rv ry r� rD rE r r� rO r� r� ri )r r� r� r� � filecontentr� r r r rh s0 � � ��z ShortcutHandler.remove_loginc C r� )z@Return result of apt_pkg.config.find_dir("Dir::Etc::netrcparts")zDir::Etc::netrcpartsr� rQ r r r r� : r� z ShortcutHandler.netrcparts_pathc C r� )z7Get the auth.conf.d filename, without the leading path.�confr� rQ r r r �netrcparts_filename? r� z#ShortcutHandler.netrcparts_filenamec C r� )z+Get the auth.conf.d absolute-path filename.)r� r� r� rQ r r r r� D r� z ShortcutHandler.netrcparts_filec C sR t | j| jf�s dS t| �� j�j}t| �� j�j}d|� |� d| j� d| j� �S )z�Content to put into auth.conf.d file By default, if both username and password are set, this will return a proper netrc-formatted line with the authentication information, including the hostname and path. Nzmachine z login z password )r� r* r+ r r r �hostnameru )r r� ru r r r r� I s z"ShortcutHandler.netrcparts_contentc C r� )zMode of netrcparts filei� r rQ r r r r� X r� z ShortcutHandler.netrcparts_modec C s: t |�}tt|j�t| j�B �|_t t|�| jd�| _dS )z�Set the SourceEntry. This should be called from subclasses to set the SourceEntry. The SourceEntry file will be set to the sourceparts_file value. The self.components, if any, will be added to the line's component(s). rZ N)r r� rs rP r r` r� r )r �linerK r r r �_set_source_entry] s z!ShortcutHandler._set_source_entryc C s0 | j }|sd S |r|d| 7 }t�dd|�� �S )Nz-%sz [^a-z0-9_-]+rE )r �re�sub�lower)r r� �baser r r �_encode_filebasei s z ShortcutHandler._encode_filebasec C s | j |d�}|s d S d||f S )Nr� z%s.%s)r� )r �extr� r� r r r r� q s z%ShortcutHandler._filebase_to_filenamec C s |sd S t j�||�S r )rt ru rO )r ru �namer r r r� w s z!ShortcutHandler._filename_to_file)NFNNFr ).�__name__� __module__� __qualname__�__doc__r � classmethodr" r) r, rL �propertyrR rU rT rN rW rY r r* r+ rf ri rc rg rw r� r� rn rd r� r� r� r� r� r� re rh r� r� r� r� r� r� r� r� r� r r r r r , s� % I* '( (& r c @ � e Zd ZdZdS )r� z-General Exception during shortcut processing.N�r� r� r� r� r r r r r� } s r� c @ r� )�InvalidShortcutExceptionz�Invalid shortcut. This should only be thrown from the constructor of a ShortcutHandler subclass, and only to indicate that the provided shortcut is invalid for that ShortcutHandler class. Nr� r r r r r� � s r� )r� rt r� r? r: �aptsources.distror �&softwareproperties.extendedsourceslistr r r � contextlibr r r rE � urllib.parser �initr� �objectr � Exceptionr� r� r r r r �<module> s( U