o td1@sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZddlmZddlmZmZmZdd lmZGd d d eZed kroeddZed edddSdS))TemplateN)gettext) SourcesList)urlopen) HTTPError)get_disturl_downloadablecountry_mirror)readlinec@seZdZdZdZdZdejvZddZ ddZ d d Z d d Z d dZ d"ddZddZddZddZddZddZddZddZd d!ZdS)#DistUpgradeFetcherCorez2 base class (without GUI) for the upgrade fetcher http://archive.ubuntu.com/ubuntumainDEBUG_UPDATE_MANAGERcCs||_t|_||_g|_dSN)new_distrcurrent_dist_name _progress run_options)selfrprogressrD/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeFetcherCore.py__init__1s zDistUpgradeFetcherCore.__init__cCs|jr tj|ddSdS)z" helper to show debug information  N)DEBUGsysstderrwrite)rmsgrrr_debug8szDistUpgradeFetcherCore._debugcCsd|jvrdS|jjdurX||jj}t}ztdt|}| dd}Wn't y?| t dt dYdSt yR| t d t d YdSwt|tt|tt d d d t}|t drwdSdS)Nz(--frontend=DistUpgradeViewNonInteractiveTzUTF-8replacez'Could not find the release announcementzThe server may be overloaded.Fz+Could not download the release announcementz&Please check your internet connection.zContinue [yN] )endy)rrreleaseNotesURI _expandUrisocketgetdefaulttimeoutsetdefaulttimeoutrreaddecodererror_IOErrorprintr striplower startswith)ruritimeout release_notesnotesresrrrshowReleaseNotes=s8        z'DistUpgradeFetcherCore.showReleaseNotescCst|t|dS)z minimal implementation for error display, should be overwriten by subclasses that want to more fancy method F)r0)rsummarymessagerrrr-XszDistUpgradeFetcherCore.errorcCsv|jjr9|jdtj|jj}|jdtj|jj}ttdtj|tj|d| ||r9dSdS)N/z0authenticate '%(file)s' against '%(signature)s' )file signatureTF) rupgradeToolSigtmpdirospathbasename upgradeToolr0r.gpgauthenticate)rfsigrrr authenticate`s   z#DistUpgradeFetcherCore.authenticateNcCs@dg}|sd}|d|g7}|d||g7}tj|tjd}|dkS)zx authenticated a file against a given signature, if no keyring is given use the apt default keyring zapt-keyz./usr/share/keyrings/ubuntu-archive-keyring.gpgz --keyringverify)rr) subprocesscallPIPE)rr=r>keyringgpgretrrrrEls z&DistUpgradeFetcherCore.gpgauthenticatec Cstj|jtj|j}ttdtj|tj|s!dSz!t |jdtj|jd}|D]}| |q4| WdSt j y]}ztd|WYd}~dSd}~ww)Nzextracting '%s'Fr<rzfailed to open tarfile (%s)T)rArBjoinr@rCr4r0r.existstarfileopenextractclose ReadErrorloggingr-)rfnametartarinfoerrrextractDistUpgrader|s&     z*DistUpgradeFetcherCore.extractDistUpgradercCsHd|j|jjf|_}tj|s"|tdtddtdSdS)Nz%s/%szCould not run the upgrade toolz. zThis is most likely a bug in the upgrade tool. Please report it as a bug using the command 'ubuntu-bug ubuntu-release-upgrader-core'.T) r@rnamescriptrArBrRr-r.)rr_rrrverifyDistUpraders   z(DistUpgradeFetcherCore.verifyDistUpradercCs|d|jtdd}t}|jD]P}|js |js |jdks!q|j|vr'q|j |r=|j |jkr=|j |j vr=|S|j |jkrcd|j vrc|j|t |d}t||jr]|S||jq|ddS) a try to figure what the mirror is from current sources.list do this by looing for matching DEFAULT_COMPONENT, current dist in sources.list and then doing a http HEAD/ftp size request to see if the uri is available on this server zmirror_from_sources_list: %sF) withMatcherdebrNzno mirror foundr#)r rrsetlistdisabledinvalidtyper4r3distDEFAULT_COMPONENTcompslenr add)rr4 default_urisourcesseenr\ mirror_urirrrmirror_from_sources_lists(         z/DistUpgradeFetcherCore.mirror_from_sources_listc Cs||jr|d|||j}|r|St|}t}|j|d}zt||js0td|W|StyR}z|d||jdd}WYd}~|Sd}~ww)z expand the uri so that it uses a mirror if the url starts with a well known string (like archive.ubuntu.com) ztrying to find suitable mirror) countrymirrorzfailed to download %sz url '%s' could not be downloadedr#N) r3DEFAULT_MIRRORr rqrr safe_substituter Exception)rr4new_uri uri_templatemr\rrrr's&     z!DistUpgradeFetcherCore._expandUricCsFtjdd}||_t|td||jdkr(tj ddtj ddt |j }|j jdurG||j j}tj||td d }||j jdur||j j|_tj||jtd d }||}||jkrttd |d Stj|j jtj|j jfD]}tj|rtj|dkstd|d SqdSd S)z. download the tarball with the upgrade script zubuntu-release-upgrader-)prefixzusing tmpdir: '%s'rzDebug::Acquire::http1zDebug::Acquire::ftpNzUpgrade tool signature)descrz Upgrade toolzfetch result != continue (%s)Fzfile '%s' missingT)tempfilemkdtempr@rAchdirrXdebugrapt_pkgconfigrcAcquirerrr?r' AcquireFiler.rDr4runRESULT_CONTINUEwarningrBrCrRgetsize)rr@fetcherr4af1af2resultrFrrrfetchDistUpgradersF       z(DistUpgradeFetcherCore.fetchDistUpgradercCsD|jg|j}tdkrtdddg|dSt|j|dS)Nrz /usr/bin/sudosudoz-E)r_rrAgetuidexecv)rargsrrrrunDistUpgraders z&DistUpgradeFetcherCore.runDistUpgradercCstdt|jdS)Nz..)rAr~shutilrmtreer@)rrrrcleanups zDistUpgradeFetcherCore.cleanupc Cs@|sdS|s|tdtddS|s*|tdtd|dS|s:|tdtddS|sN|tdtd|dSzt |j tj sct d |j }d |_ ||WdSt y}z*|j d kr|td td WYd}~d S|td td|jWYd}~dSd}~ww)NzFailed to fetchz=Fetching the upgrade failed. There may be a network problem. zAuthentication failedz_Authenticating the upgrade failed. There may be a problem with the network or with the server. zFailed to extractz[Extracting the upgrade failed. There may be a problem with the network or with the server. zVerification failedz[Verifying the upgrade failed. There may be a problem with the network or with the server. zCan not execute '%s' zCan not run the upgradezyThis usually is caused by a system where /tmp is mounted noexec. Please remount without noexec and run the upgrade again.FzThe error message is '%s'.T)r9rr-r.rHrr]r`rAaccessr_X_OKOSErrorerrnorstrerror)rexr\rrrrsV          zDistUpgradeFetcherCore.runr)__name__ __module__ __qualname____doc__rsrirAenvironrrr r9r-rHrEr]r`rqr'rrrrrrrrr *s(  ' r __main__zgot mirror: '%s'zghttp://archive.ubuntu.com/ubuntu/dists/intrepid-proposed/main/dist-upgrader-all/0.93.34/intrepid.tar.gzr )stringrrArrXrSr|rr(rrJrr.aptsources.sourceslistrurllib.requestr urllib.errorrutilsrr r DistUpgradeViewTextr objectr rdr0rqrrrrs8