File: //lib/python3/dist-packages/DistUpgrade/__pycache__/utils.cpython-38.pyc
U
�f�c�G � @ s2 d dl mZ d dlmZ d dlmZ d dlmZmZmZ d dl m
Z
d dl Z d dl Z e �
� d dlZd dlZd dlZd dlZd dlZd dlZd dlZz,d dlmZmZmZmZmZ d dlm Z W n< e k
r� d dl mZmZmZmZmZ d dl m Z Y nX d d l m Z G d
d � d e!�Z"d d
� Z#dd� Z$dKdd�Z%dd� Z&dLdd�Z'dd� Z(dd� Z)d d � Z*d d � Z+d d!� Z,d"d#� Z-G d$d%� d%e�Z.dMd&d'�Z/dNd(d)�Z0d*d+� Z1d,d-� Z2d.d/� Z3d0d1� Z4dOd2d3�Z5d4d5� Z6dPd6d7�Z7dQd8d9�Z8d:d;� Z9d<d=� Z:d>d?� Z;d@dA� Z<dBdC� Z=dDdE� Z>dFdG� Z?e@dHk�r.e9ddIdJ� dS )R� )�print_function)�gettext)�ngettext)�S_IMODE�ST_MODE�S_IXUSR)�ceilN)� ProxyHandler�Request� build_opener�install_opener�urlopen)�urlsplit)�copyc @ s* e Zd ZdZd
dd�Zdd� Zdd� Zd S ) �
ExecutionTimez�
Helper that can be used in with statements to have a simple
measure of the timing of a particular block of code, e.g.
with ExecutionTime("db flush"):
db.flush()
� c C s
|| _ d S �N)�info)�selfr � r �3/usr/lib/python3/dist-packages/DistUpgrade/utils.py�__init__G s zExecutionTime.__init__c C s t � � | _d S r )�time�now�r r r r � __enter__J s zExecutionTime.__enter__c C s t d| jt�� | j f � d S )Nz%s: %s)�printr r r )r �type�value�stackr r r �__exit__M s zExecutionTime.__exit__N)r )�__name__�
__module__� __qualname__�__doc__r r r r r r r r @ s
r c C sJ t | �}t|j�}|jr*|j�|jd�|_|jrB|j�|jd�|_t|�S )Nzhidden-uzhidden-p)r r �uriZusername�replaceZpassword�str)�entryZtmpZ url_partsr r r �)get_string_with_no_auth_from_source_entryQ s
r) c C sF d} z ddl }|�� }|�d�} W n tk
r@ t�d� Y nX | S )z+ return True if Unity is currently running Fr Nzcom.canonical.Unityz&could not check for Unity dbus service)�dbusZ
SessionBusZname_has_owner� Exception�logging� exception)Z
unity_runningr* �busr r r �is_unity_running[ s r/ c C s| |s t �� }|dkrxd| }t|��}|�� }W 5 Q R X |�d�d �d�d }|| kr\dS t|�d�d �� d �}q dS ) Nr z
/proc/%s/stat�(� �)T� F)�os�getpid�open�read� partition�
rpartition�int�split)Z processname�pidZ stat_fileZstat_f�statZcommandr r r �is_child_of_process_nameg s
r> c C s0 t j�d�rt j�d�s dS t �d�t �d�kS )z, returns True if we are inside a chroot
z/procz/proc/1T�/z /proc/1/root)r4 �path�existsr= r r r r �
inside_chrootw s rB �F r c C sL d}| � � D ]:}t|�|�d� t|� |kr:|d| 7 }||d 7 }q |S )z� helpers inspired after textwrap - unfortunately
we can not use textwrap directly because it break
packagenames with "-" in them into new lines
r �
� )r; �len�rfind)�t�widthZsubsequent_indent�out�sr r r �wrap� s rL c K s4 d}| � d�}|D ] }t|f|�} || d 7 }q|S )Nr rD )r; rL )rK �kwargs�msgZparasZparr r r �twrap� s
rO c C sd g } t j�d�sg S tjdgtjdd�}|�� d �d�}|d= |D ]}|rD| �|�� d � qD| S )z= return list of loaded modules (or [] if lsmod is not found) z /sbin/lsmodT��stdoutZuniversal_newlinesr rD ) r4 r@ rA �
subprocess�Popen�PIPE� communicater; �append)�modules�p�lines�liner r r �lsmod� s �r[ c C sB t j�| �sdS tt �| �t �}|t@ tks>t �| |tB � dS )zB check if a given binary has the executable bit and if not, add itN)r4 r@ rA r r= r r �chmod)r@ �moder r r �check_and_fix_xbit� s
r^ c C s� ddi} dt jkrdS t jd �� }|dd� | krD| |dd� S d|kr�|�d�d �d�d }d
|krz|�d
�d }|d S |dd � d S dS ) z: helper to get the country mirror from the current locale �cr ZLANGN� �_�.r r3 �@r1 )r4 �environ�lowerr; )Z lang_mirrorZlangZcountryr r r �country_mirror� s
rf c C s� t j�d�} | r"t�d| � | S ddlm}m} |dddg|dd �}|�� }|dkrjt j
� d
| � d S |j �
� �� } |j �� | S ) z2 return the codename of the current runing distro ZMETA_RELEASE_FAKE_CODENAMEzQusing fake release name '%s' (because of META_RELEASE_FAKE_CODENAME environment) r �rS rT � lsb_releasez-c�-sTrP �"lsb_release returned exitcode: %i
�unknown distribution�r4 rd �getr, ZwarningrR rS rT �wait�sys�stderr�writerQ �readline�strip�close)ZdistrS rT rX �resr r r �get_dist� s" � �
rv c C s� t j�d�} | r"t�d| � | S ddlm}m} |dddg|dd �}|�� }|dkrjt j
� d
| � d S |j �
� �� } |j �� | S ) z2 return the version of the current running distro ZMETA_RELEASE_FAKE_VERSIONzSusing fake release version '%s' (because of META_RELEASE_FAKE_VERSION environment) r rg rh z-rri TrP rj rk rl )ZdescrS rT rX ru r r r �get_dist_version� s" � �
rw c @ s e Zd Zdd� ZdS )� HeadRequestc C s dS )NZHEADr r r r r �
get_method� s zHeadRequest.get_methodN)r! r" r# ry r r r r rx � s rx c
C sb |s dd� |d| � t | �\}}}}}|d|||||f � |dkr�z*tt| ��}|�� |jdkrnW dS W dS tk
r� } z|d | � W Y �dS d
}~X Y nX n�|d k�r^d d
l} zd| �|�}
|
�� |
� t
j � |�� |
�
t
j �|��} |
�� |�r|d
| � | d k�r W dS W n< tk
�r\ } z |�rF|d| � W Y �dS d
}~X Y nX dS )z�
helper that checks if the given uri exists and is downloadable
(supports optional debug_func function handler to support
e.g. logging)
Supports http (via HEAD) and ftp (via size request)
c S s dS )NTr )�xr r r �<lambda> � z"url_downloadable.<locals>.<lambda>zurl_downloadable: %sz"s='%s' n='%s' p='%s' q='%s' f='%s')�http�https�� TFzerror from httplib: '%s'NZftpr zftplib.size() returned: %szerror from ftplib: '%s')r r
rx rt �coder+ �ftplibZFTPZlogin�cwdr4 r@ �dirname�size�basename�quit) r% Z
debug_funcZschemeZnetlocr@ ZquerryZfragmentZ http_file�er� �fr� r r r �url_downloadable� sF
�
r� c
C sH d}i }t j�d�dkr*t j�d�|d< n`tj�|�r�t �� }t �||� |�dd�}|r�|�d�}t |�
d��}|r�|r�d ||f |d< t j�d
�dkr�t j�d
�|d < nd|kr�|d |d < |�rD|� � D ](}t �
d |�s�td
| tjd� dS q�t|�}t|�} t| � d|k�r*t�d|d � d |k�rDt�d|d � |S )z� init proxy settings
* use apt.conf http proxy if present,
* otherwise look into synaptics config file,
* otherwise the default behavior will use http_proxy environment
if present
z /root/.synaptic/synaptic.confzAcquire::http::Proxyr r} zSynaptic::useProxyFzSynaptic::httpProxyzSynaptic::httpProxyPortz
http://%s:%s/zAcquire::https::Proxyr~ z https?://\w+zproxy '%s' looks invalid)�fileNZ
http_proxyZ https_proxy)�apt_pkg�config�findr4 r@ rA Z
ConfigurationZread_config_fileZfind_br' Zfind_i�values�re�matchr ro rp r r r �putenv)
Z gsettingsZSYNAPTIC_CONF_FILEZproxiesZcnfZ use_proxyZ
proxy_hostZ
proxy_port�proxyZ
proxy_supportZopenerr r r �
init_proxy" s<
r� c
C s� z�ddl } | �| jj�}z(|�dd�}| �|d�}|�dd�W W S | jjk
r� } z>d}|j|krf� |�dd �}| �|d�}|�dd
� W Y �W S d}~X Y nX W n t k
r� Y d S X dS ) z�
Check via dbus if the system is running on battery.
This function is using UPower per default, if UPower is not
available it falls-back to DeviceKit.Power.
r Nzorg.freedesktop.UPowerz/org/freedesktop/UPowerz org.freedesktop.DBus.PropertiesZ OnBatteryz)org.freedesktop.DBus.Error.ServiceUnknownz org.freedesktop.DeviceKit.Powerz /org/freedesktop/DeviceKit/Power�
on_batteryF)
r* ZBusZ TYPE_SYSTEMZ
get_objectZ InterfaceZGet�
exceptionsZ
DBusExceptionZ_dbus_error_namer+ )r* r. ZdevobjZdevr� Z
error_unknownr r r r� N s( �
� (r� c C s� zdddl m} m} | �| jj�}|�dddd|�dd�d dd
d d �
\}}| j|� � |d d �}|W S t
k
rz Y d S X d S )
z�
Send a dbus signal to logind to not suspend the system, it will be
released when the return value drops out of scope
r )�Gio�GLibzorg.freedesktop.login1z/org/freedesktop/login1z org.freedesktop.login1.ManagerZInhibitz(ssss))zshutdown:sleepZ
UpdateManagerzUpdating System�blockN���)�fdF) �
gi.repositoryr� r� Z bus_get_syncZBusTypeZSYSTEMZcall_with_unix_fd_list_syncZVariantZUnixInputStreamZ steal_fdsr+ )r� r� Z
connection�varZfdlistZ inhibitorr r r �
inhibit_sleepj s* � �r� c C s | dks| � � dkrdS dS )N�0ZFALSEFT)�upper)r' r r r � str_to_bool� s r� c C s@ dd l } zt�� \}}|W S tk
r: | �d� Y d S X d S )Nr zgedefaultlocale() failed)r, �localeZgetdefaultlocaler+ r- )r, Zlocale_s�encodingr r r �get_lang� s
r� c C s t | d�}|�dd�d S )z6 try to guess the flavor based on the running desktop ��cache�-r3 r )�get_ubuntu_flavor_packager; )r� �pkgr r r �get_ubuntu_flavor� s
r� c C s ddl m} | d�}|�dd�S )Nr )�DistUpgradeConfigz"/usr/share/ubuntu-release-upgraderZDistroZMetaPkgs)Z#DistUpgrade.DistUpgradeConfigParserr� Zgetlist)r� �parserr r r �_load_meta_pkg_list� s r� c
C s� dg}z|� tt� �� W n, tk
rF } ztd|� W 5 d}~X Y nX | dkrXt�� } |D ]*}|| krp| | nd}|r\|jr\| S q\dS )zB try to guess the flavor metapackage based on the running desktop zubuntu-desktopz%Could not load list of meta packages:N)�extend�sortedr� r+ r �aptZCacheZ is_installed)r� Z meta_pkgsr� Zmeta_pkgZ cache_pkgr r r r� � s
r� c C sf t | d�}ddi}||kr"|| S |�d�r@t|�dd�d �S |�d�r^t|�dd�d �S dS d S )
z; try to guess the flavor name based on the running desktop r� zubuntustudio-desktopz
Ubuntu Studioz-desktopr3 r z-netbookZUbuntuN)r� �endswith�capitalize_first_word�rsplit)r� r� �lookupr r r �get_ubuntu_flavor_name� s
r� c C s� dd l }|�dd� ddlm}m} |j| |jj|jj |j
j d�}|� d||f � |�
� |�� �|jj� |�d� |�� |�� dS ) Nr �Gtkz3.0)r� �Gdk)�parent�flagsr Zbuttonsz<big><b>%s</b></big>
%sr F)�giZrequire_versionr� r� r� Z
MessageDialogZ DialogFlagsZMODALZ MessageTypeZERRORZ ButtonsTypeZCLOSEZ
set_markupZrealizeZ
get_windowZ
set_functionsZ
WMFunctionZMOVEZ set_title�runZdestroy)r� �summary�messager� r� r� �dr r r �error� s �
r� c C sL | dk r0t t| td� ��}tdd|�d|i S t�td�| d d �S dS )zG
Convert a given size in bytes to a nicer better readable unit
i@B i� z
%(size).0f kBr� z%.1f MBg @�@N)r: r �floatr r� �
format_stringra )�bytesZ
size_in_kbr r r �
humanize_size� s
�r� c C s t j�d�S )NzAPT::Architecture)r� r� r� r r r r �get_arch� s r� c
C s� d}d}d}t d��p}|�� D ]`}|�� }|s0q |�� }|| }||krJq || �d�d }t|d�} | | kr W 5 Q R � dS q W 5 Q R X dS ) z@ check if the current system is listening on the given tcp port r3 � Z0Az
/proc/net/tcp�:� TF)r6 � readlinesrs r; r: )
ZportZINDEX_LOCAL_ADDRZ INDEX_STATEZSTATE_LISTENINGZnet_tcprZ r� �stateZlocal_port_strZ
local_portr r r �is_port_already_listening� s"
r� c C sV d} t �� dkrtd��t j�d�s(dS tjddgtjdd ��� d }|| krRdS dS )
z# Return True if iptables is active z�Chain INPUT (policy ACCEPT)
target prot opt source destination
Chain FORWARD (policy ACCEPT)
target prot opt source destination
Chain OUTPUT (policy ACCEPT)
target prot opt source destination
r z%Need root to check the iptables statez/sbin/iptablesFZiptablesz-nLTrP ) r4 �getuid�OSErrorr@ rA rR rS rT rU )Ziptables_emptyrJ r r r �iptables_active
s
��r� c C s@ t | �dkr<| d �� r<| d �� s<| d �� | dd� S | S )z3 this uppercases the first word's first letter
r3 r N)rF �isalpha�isupper�
capitalize)�stringr r r r� " s $r� c C s t | jdd�}t|�S )zX this takes a package synopsis and uppercases the first word's
first letter
r� r )�getattr� candidater� )r� �namer r r �get_package_label* s r� �__main__ZbarZbaz)N)rC r )N)N)N)N)N)AZ
__future__r r ra r r= r r r Zmathr r� r� Z init_configr� r, r� r4 rR ro r Zurllib.requestr r
r r r
Z urllib.parser � ImportErrorZurllib2Zurlparser �objectr r) r/ r> rB rL rO r[ r^ rf rv rw rx r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r! r r r r �<module> sj
+
,