Mini Shell
3
C��e�$ � @ s� d dl Z d dlmZ d dlmZ d dlmZmZmZm Z m
Z
m Z m Z m
Z
d dlmZmZ d dlmZmZ d dlmZ d dlmZ ee�Zd d
gZG d d
� d
e�ZG d d
� d
e �Z e � Z dS )� N)�IntEnum)� DBusError)�THREAD_STORAGE�THREAD_PAYLOAD�THREAD_PAYLOAD_RESTART� THREAD_WAIT_FOR_CONNECTING_NM�THREAD_SUBSCRIPTION�PAYLOAD_TYPE_DNF�THREAD_STORAGE_WATCHER�THREAD_EXECUTE_STORAGE)�_�N_)� threadMgr�AnacondaThread)� PayloadError)�get_module_logger�
payloadMgr� PayloadStatec @ s0 e Zd ZdZdZdZdZdZdZdZ dZ
d
Z d S ) r z3Enum for payload state after payload was restarted.r � � � � � � N���) �__name__�
__module__� __qualname__�__doc__�STARTED�WAITING_STORAGE�WAITING_NETWORK�VERIFYING_AVAILABILITY�DOWNLOADING_PKG_METADATA�DOWNLOADING_GROUP_METADATA�FINISHED�ERROR� r' r' � /usr/lib64/python3.6/manager.pyr ' s c @ sj e Zd ZdZed�Zed�Zdd� Zedd� �Z dd � Z
dd d
�Z edd� �Z dd� Z
dd� Zdd� ZdS )�PayloadManageraZ Framework for starting and watching the payload thread.
This class defines several states (PayloadState enum), and events can
be triggered upon reaching a state. Depending on whether a state has
already been reached when a listener is added, the event code may be
run in either the calling thread or the payload thread. The event code
will block the payload thread regardless, so try not to run anything
that takes a long time.
All states except ERROR are expected to happen linearly, and adding
a listener for a state that has already been reached or passed will
immediately trigger that listener. For example, if the payload thread is
currently in DOWNLOADING_GROUP_METADATA, adding a listener for
WAITING_NETWORK will immediately run the code being added
for WAITING_NETWORK.
The payload thread data should be accessed using the payloadMgr object,
and the running thread can be accessed using threadMgr with the
THREAD_PAYLOAD constant, if you need to wait for it or something. The
thread should be started using payloadMgr.restart_thread.
z$Failed to set up installation sourcez"Error downloading package metadatac C sH t j� | _i | _tj| _d | _x$tjj � D ]\}}g | jt|�< q*W d S )N)
� threadingZLock� _event_lock�_event_listenersr r �
_thread_state�_error� __members__�items)�self�_name�valuer' r' r( �__init__O s
zPayloadManager.__init__c C s
t | j�S )N)r r. )r1 r' r' r( �errorY s zPayloadManager.errorc C sR | j �B | j| j|� |tjkr4|| jkrD|� n|| jkrD|� W dQ R X dS )z�Add a listener for an event.
:param int event_id: The event to listen for, one of the EVENT_* constants
:param function func: An object to call when the event is reached
N)r+ r, �appendr r& r- )r1 �event_id�funcr' r' r( � add_listener] s
zPayloadManager.add_listenerFTc C s: t jd� tjt�rdS tjtt| j||||fd�� dS )a� Start or restart the payload thread.
This method starts a new thread to restart the payload thread, so
this method's return is not blocked by waiting on the previous payload
thread. If there is already a payload thread restart pending, this method
has no effect.
:param payload.Payload payload: The payload instance
:param bool fallback: Whether to fall back to the default repo in case of error
:param bool checkmount: Whether to check for valid mounted media
:param bool onlyOnChange: Restart thread only if existing repositories changed.
This won't restart thread even when a new repository was added!!
zRestarting payload threadN)�name�target�args)�log�debugr �getr �addr �_restart_thread)r1 �payload�fallback�
checkmount� onlyOnChanger' r' r( �restart_threadt s
z PayloadManager.restart_threadc C s t jt�pt jt�S )z(Is the payload thread running right now?)r �existsr r )r1 r' r' r( �running� s zPayloadManager.runningc C s, t jt� t jtt| j||||fd�� d S )N)r: r; r< )r �waitr r@ r � _run_thread)r1 rB rC rD rE r' r' r( rA � s
z PayloadManager._restart_threadc C sD t jd|j� | j�& || _x| j| D ]
}|� q(W W d Q R X d S )Nz!Updating payload thread state: %s)r= r> r: r+ r- r, )r1 r7 r8 r' r' r( �
_set_state� s
zPayloadManager._set_statec C s� d | _ | jtj� | jtj� tjt� tjt� tjt � | jtj
� tjt � tjt � |j
� |jtkr~| jtj� d S |r�tjd� | jtj� |j� r�tjd� | jtj� d S | jtj� y|j||d� |j� W nP tttfk
�r0 } z,tjd|� | j | _ | jtj � |j � d S d }~X nX | jtj � |j � |j!� |j"�s�tjd� | j#| _ | jtj � |j � d S |j$� | jtj� d S )Nz!Testing repositories availabilityz:Payload isn't restarted, repositories are still available.)rC rD zPayloadError: %szNo base repo configured)%r. rK r r r r rI r r
r r! r r Zsetup�typer r% r= r> r" Z verify_available_repositoriesr# Zupdate_base_repoZadd_driver_repos�OSErrorr r r5 � ERROR_SETUPr& Zunsetupr$ Zgather_repo_metadata�releaseZ base_repo�ERROR_MDZ
post_setup)r1 rB rC rD rE �er' r' r( rJ � sR
zPayloadManager._run_threadN)FTF)r r r r r
rN rP r4 �propertyr5 r9 rF rH rA rK rJ r' r' r' r( r) 5 s
r) ) r* �enumr Z dasbus.errorr Zpyanaconda.core.constantsr r r r r r r
r Zpyanaconda.core.i18nr r
Zpyanaconda.threadingr r Zpyanaconda.payload.errorsr Zpyanaconda.anaconda_loggersr r r= �__all__r �objectr) r r' r' r' r( �<module> s ( =