U
    Ùªzc  ã                   @   s0   d dl Z d dlZddlmZ G dd„ deƒZdS )é    Né   )ÚDistUpgradeFetcherCorec                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚDistUpgradeFetcherSelfc                 C   sh   t  | ||¡ || _|  jdg7  _|jd k	rD|  jd|j g7  _|jd k	rd|  jd|j g7  _d S )Nz--with-networkz
--cdrom=%sz--frontend=%s)r   Ú__init__ÚviewZrun_optionsZ	cdromPathZfrontend)ÚselfZnew_distZprogressZoptionsr   © r   úD/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeFetcherSelf.pyr      s    

zDistUpgradeFetcherSelf.__init__c                 C   s   | j  ||¡S )N)r   Úerror)r   ZsummaryÚmessager   r   r	   r
   )   s    zDistUpgradeFetcherSelf.errorc                 C   s,   t  d¡ t  ¡  t dd¡ t | ¡ dS )z, overwrite to ensure that the log is copied z&runDistUpgrader() called, re-exec selfz/var/log/dist-upgrade/main.logz*/var/log/dist-upgrade/main_update_self.logN)ÚloggingÚinfoZshutdownÚshutilÚcopyr   ÚrunDistUpgrader)r   r   r   r	   r   ,   s    
ÿz&DistUpgradeFetcherSelf.runDistUpgraderN)Ú__name__Ú
__module__Ú__qualname__r   r
   r   r   r   r   r	   r      s   r   )r   r   r   r   r   r   r   r	   Ú<module>   s   