6Nc@s)ddlmZddlmZmZmZddlZejddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlmZddlmZdefdYZdZd Zd Zdd Zd Zd ddZdZdZdZ dZ!dZ"dej#fdYZ$ddZ%ddZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-d Z.d!Z/d"Z0d#Z1d$Z2d%Z3e4d&kr%e3GHndS('i(tgettext(tS_IMODEtST_MODEtS_IXUSRN(tcopy(turlsplitt ExecutionTimecBs,eZdZddZdZdZRS(s Helper that can be used in with statements to have a simple measure of the timming of a particular block of code, e.g. with ExecutinTime("db flush"): db.flush() tcCs ||_dS(N(tinfo(tselfR((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt__init__3scCstj|_dS(N(ttimetnow(R ((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt __enter__5scCs#d|jtj|jfGHdS(Ns%s: %s(RR R (R ttypetvaluetstack((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt__exit__7s(t__name__t __module__t__doc__R R R(((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyR,s  cCsst|}t|j}|jrB|jj|jd|_n|jri|jj|jd|_nt|S(Nshidden-ushidden-p(RRturitusernametreplacetpasswordtstr(tentryttmpt url_parts((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt)get_string_with_no_auth_from_source_entry:s   cCsNd}tjd}x1tjd|D]}|tjj|7}q*W|S(sE estimate the amount of space that the current kernel takes in /boot iis /boot/*%s*(tostunametglobtpathtgetsize(tsizetkvertf((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytestimate_kernel_size_in_bootCs cCsLt}y+ddl}|j}|jd}WntjdnX|S(s+ return True if Unity is currently running iNscom.canonical.Unitys&could not check for Unity dbus service(tFalsetdbust SessionBustname_has_ownertloggingt exception(t unity_runningR(tbus((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytis_unity_runningKs  cCs|stj}nx|dkrd|}t|j}|jddjdd}||krptSt|jddjd}qWtS(Nis /proc/%s/statt(it)i( Rtgetpidtopentreadt partitiontTruetinttsplitR'(t processnametpidt stat_filetstattcommand((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytis_child_of_process_nameVs   'cCsFtjjd s&tjjd r*tStjdtjdkS(s- returns True if we are inside a chroot s/procs/proc/1t/s /proc/1/root(RR!texistsR6R<(((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt inside_chrootds&iFRcCsid}x\|jD]N}t||jdt||krS|d|7}n||d7}qW|S(s helpers inspired after textwrap - unfortunately we can not use textwrap directly because it break packagenames with "-" in them into new lines Rs t (R8tlentrfind(tttwidthtsubsequent_indenttoutts((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytwrapms )cKsGd}|jd}x+|D]#}t||}||d7}qW|S(NRs (R8RJ(RItkwargstmsgtparastpar((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyttwrapys  cCsg}tjjdsgStjdgdtj}|jdjd}|d=x.|D]&}|r^|j|jdq^q^W|S(s= return list of loaded modules (or [] if lsmod is not found) s /sbin/lsmodtstdoutis ( RR!R@t subprocesstPopentPIPEt communicateR8tappend(tmodulestptlinestline((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytlsmods cCsZtjj|sdSttj|t}|t@tksVtj||tBndS(sB check if a given binary has the executable bit and if not, add itN(RR!R@RR<RRtchmod(R!tmode((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytcheck_and_fix_xbits cCsidd6}dtjkr dStjdj}|d |krO||d Sd|kr|jddjdd}d |kr|jd d}n|dS|d dSdS( s: helper to get the country mirror from the current locale RtctLANGit_t.iit@i(RtenvirontlowerR8(t lang_mirrortlangtcountry((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytcountry_mirrors      cCstjjd}|r-tjd||Sddlm}m}|dddgd|}|j}|d krt j j d |d S|j j j}|S( s2 return the codename of the current runing distro tMETA_RELEASE_FAKE_CODENAMEsQusing fake release name '%s' (because of META_RELEASE_FAKE_CODENAME environment) i(RRRSt lsb_releases-cs-sRPis"lsb_release returned exitcode: %i sunknown distribution(RRctgetR+twarnRQRRRStwaittsyststderrtwriteRPtreadlinetstrip(tdistRRRSRWtres((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytget_dists  t HeadRequestcBseZdZRS(cCsdS(NtHEAD((R ((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt get_methods(RRRx(((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyRvsc Cs|sdn|d|tj|\}}}}}|d|||||f|dkry:tjt|}|j|jdkrtStSWqt k r}|d|tSXn|dkrdd l } y| j |} | j | j tjj|| jtjj|} | j|rP|d | n| d kr`tSWqt k r}|r|d |ntSXntS( s 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) cSstS(N(R6(tx((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytssurl_downloadable: %ss"s='%s' n='%s' p='%s' q='%s' f='%s'thttpiserror from httplib: '%s'tftpiNsftplib.size() returned: %siserror from ftplib: '%s'(turlparseRturllib2turlopenRvtclosetcodeR6R't ExceptiontftplibtFTPtlogintcwdRR!tdirnameR#tbasenametquit( Rt debug_functschemetnetlocR!tquerrytfragmentt http_fileteRR%R#((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyturl_downloadables@        c Cs>d}d }tjjddkr9tjjd}ntjj|rtj}tj|||j dt }|r|jd}t |j d}|r|rd||f}qqn|r:t jd|stjd |IJd Stji|d 6}tj|}tj|tjd |n|S( s init proxy settings * first check for http_proxy environment (always wins), * then check the apt.conf http proxy, * then look into synaptics conffile * then into gconf (if gconfclient was supplied) s/root/.synaptic/synaptic.confsAcquire::http::ProxyRsSynaptic::useProxysSynaptic::httpProxysSynaptic::httpProxyPorts http://%s:%s/s http://\w+sproxy '%s' looks invalidNR{t http_proxy(tNonetapt_pkgtconfigtfindRR!R@t Configurationtread_config_filetfind_bR'Rtfind_itretmatchRnRoR~t ProxyHandlert build_openertinstall_openertputenv( t gsettingstSYNAPTIC_CONF_FILEtproxytcnft use_proxyt proxy_hostt proxy_portt proxy_supporttopener((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt init_proxys,   cCsyddl}|j|jj}y8|jdd}|j|d}|jddSWn_|jjk r}|jdkrn|jdd }|j|d}|jdd SXWnt k r}t SXdS( s 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. iNsorg.freedesktop.UPowers/org/freedesktop/UPowersorg.freedesktop.DBus.Propertiest OnBatterys)org.freedesktop.DBus.Error.ServiceUnknownsorg.freedesktop.DeviceKit.Powers /org/freedesktop/DeviceKit/Powert on_battery( R(tBust TYPE_SYSTEMt get_objectt InterfacetGett exceptionst DBusExceptiont_dbus_error_nameRR'(R(R.tdevobjtdevR((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyR s"     cCsaddl}|j|jj}|jdd}|j|d}|jdd}||fS(s Send a dbus signal to org.gnome.SettingsDaemon.Power to not suspend the system, this is to support upgrades from pre-gutsy g-p-m iNsorg.gnome.SettingsDaemons/org/gnome/SettingsDaemon/Powersorg.gnome.SettingsDaemon.Powert UpdateManagersUpdating system(R(Rt TYPE_SESSIONRRtInhibit(R(R.RRtcookie((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt_inhibit_sleep_old_interface:s   cCsaddl}|j|jj}|jdd}|j|d}|jdd}||fS(sK Send a dbus signal to gnome-power-manager to not suspend the system iNsorg.freedesktop.PowerManagements(/org/freedesktop/PowerManagement/Inhibits'org.freedesktop.PowerManagement.InhibitRsUpdating system(R(RRRRR(R(R.RRR((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt_inhibit_sleep_new_interfaceGs   cCsIy tSWn7tk rDy tSWqEtk r@ttfSXnXdS(s Send a dbus signal to power-manager to not suspend the system, try both the new freedesktop and the old gnome dbus interface N(RRRR'(((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt inhibit_sleepTs    cCs4y|j|Wntk r/}d|GHnXdS(sLSend a dbus signal to gnome-power-manager to allow a suspending the systems,could not send the dbus UnInhibit signal: %sN(t UnInhibitR(RRR((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt allow_sleepcscCs&|dks|jdkr"tStS(Nt0tFALSE(tupperR'R6(R((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt str_to_boollscCst|djdS(Ntlatin1sutf-8(tunicodetencode(R((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytutf8qscCsLddl}ytj\}}|SWntk rG|jddSXdS(Nisgedefaultlocale() failed(R+tlocaletgetdefaultlocaleRR,R(R+tlocale_stencoding((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytget_langts   cCsUtjjdd}d|kr%dSd|kr5dSd|ksMd|krQdSdS( s6 try to guess the flavor based on the running desktop tDESKTOP_SESSIONRtgnometubuntutkdetkubuntutxfcetxubuntu(RRcRk(tdenv((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytget_ubuntu_flavor}s  c Csddlm}m}|jd|d|jjd|jjd|jj }|j d||f|j |j j |j|jd|j|jtS( Ni(tGtktGdktparenttflagsRtbuttonss%s %sR(t gi.repositoryRRt MessageDialogt DialogFlagstMODALt MessageTypetERRORt ButtonsTypetCLOSEt set_markuptrealizetwindowt set_functionst FUNC_MOVEt set_titletruntdestroyR'(RtsummarytmessageRRtd((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyterrors      cCst|dkrtdS|dkr,tdS|d krRtjtd|dStjtd|ddSdS( sG Convert a given size in bytes to a nicer better readable unit is0 kBis1 kBs%.0f kBs%.1f MBg@@Ni@B(R`Rt format_string(tbytes((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyt humanize_sizes     cCstjjdS(NsAPT::Architecture(RtConfigR(((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytget_archscCsvd}tjdkr$tdntjjds:tStjddgdtjj d}||krrtSt S(s# Return True if iptables is active sChain 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 is%Need root to check the iptables states/sbin/iptablestiptabless-LRP( RtgetuidtOSErrorR!R@R'RQRRRSRTR6(tiptables_emptyRH((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pytiptables_actives   t__main__(5RR`R<RRRRt init_configRR+RRtos.pathR RQRnR R~R}RRtobjectRRR&R/RR>RARJRORZR]RhRutRequestRvRRRRRRRRRRRRRRRR(((s5/usr/lib/python2.7/dist-packages/DistUpgrade/utils.pyts\                      ) 3