ó ÍÿŠTc@sžddlZddlmZddlmZddlmZddlmZm Z m Z de fd„ƒYZ defd „ƒYZ d efd „ƒYZdS( iÿÿÿÿN(tDeferred(tProcessProtocol(t ProcessDone(t ManagerPlugint SUCCEEDEDtFAILEDtShutdownFailedErrorcBseZdZd„ZRS(svRaised when a call to C{/sbin/shutdown} fails. @ivar data: The data that the process printed before failing. cCs ||_dS(N(tdata(tselfR((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyt__init__s(t__name__t __module__t__doc__R (((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR stShutdownManagercBsJeZdd„Zd„Zd„Zd„Zd„Zd„Zd„Z RS(cCs,|dkrddlm}n||_dS(Niÿÿÿÿ(treactor(tNonettwisted.internetRt_process_factory(Rtprocess_factory((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR s cCs-tt|ƒj|ƒ|jd|jƒdS(sAdd this plugin to C{registry}. The shutdown manager handles C{shutdown} activity messages broadcast from the server. tshutdownN(tsuperR tregistertregister_messagetperform_shutdown(Rtregistry((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyRscCs‘|d}|d}tƒ}|j|jjƒ|jj|j|ƒ|jj|j|ƒ|j ||ƒ\}}|j j ||d|ƒdS(sîRequest a system restart or shutdown. If the call to C{/sbin/shutdown} runs without errors the activity specified in the message will be responded as succeeded. Otherwise, it will be responded as failed. s operation-idtreboottargsN( tShutdownProcessProtocolt set_timeoutRRtresultt addCallbackt_respond_successt addErrbackt_respond_failuret_get_command_and_argsRt spawnProcess(Rtmessaget operation_idRtprotocoltcommandR((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR$s   cs<tjdƒˆjt||ƒ}|j‡fd†ƒ|S(NsShutdown request succeeded.csˆjjjƒS(N(Rtbrokertstop_exchanger(t_(R(sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyt:s(tloggingtinfot_respondRR(RRR%tdeferred((RsE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR4s  cCs&tjdƒ|jt|jj|ƒS(NsShutdown request failed.(R,R-R.RtvalueR(RtfailureR%((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR!=s cCs>idd6|d6|d6|d6}|jjj||jtƒS(Nsoperation-resultttypetstatuss result-texts operation-id(RR(t send_messaget _session_idtTrue(RR3RR%R$((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR.As    cCsKd|jdf}|r/dd|dg}ndd|dg}d|fS(ss Returns a C{command, args} 2-tuple suitable for use with L{IReactorProcess.spawnProcess}. s+%di<s/sbin/shutdowns-rs!Landscape is rebooting the systems-hs%Landscape is shutting down the system(tdelay(RR&RtminutesR((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR"Is    N( R R RR RRRR!R.R"(((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR s    RcBsMeZdZedd„Zd„Zdd„Zd„Zd„Zd„Z RS( s~A ProcessProtocol for calling C{/sbin/shutdown}. C{shutdown} doesn't return immediately when a time specification is provided. Failures are reported immediately after it starts and return a non-zero exit code. The process protocol calls C{shutdown} and waits for failures for C{timeout} seconds. If no failures are reported it fires C{result}'s callback with whatever output was received from the process. If failures are reported C{result}'s errback is fired. @ivar result: A L{Deferred} fired when C{shutdown} fails or succeeds. @ivar reboot: A flag indicating whether a shutdown or reboot should be performed. Default is C{False}. @ivar delay: The time in seconds from now to schedule the shutdown. Default is 240 seconds. The time will be converted to minutes using integer division when passed to C{shutdown}. iðcCs4tƒ|_||_||_g|_t|_dS(N(RRRR7t_dataR6t_waiting(RRR7((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR ks     cCsdj|jƒS(s'Get the data printed by the subprocess.t(tjoinR9(R((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pytget_datarsi cCs|j||jƒdS(si Set the error checking timeout, after which C{result}'s callback will be fired. N(t call_latert_succeed(RRttimeout((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyRvscCs |jr|jj|ƒndS(ssSome data was received from the child. Add it to our buffer to pass to C{result} when it's fired. N(R:R9tappend(RtfdR((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pytchildDataReceived}s cCsQ|jrM|jtƒr%|jƒqM|jjt|jƒƒƒt|_ndS(sFire back the C{result} L{Deferred}. C{result}'s callback will be fired with the string of data received from the subprocess, or if the subprocess failed C{result}'s errback will be fired with the string of data received from the subprocess. N( R:tcheckRR?RterrbackRR=tFalse(Rtreason((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyt processEnded…s   cCs/|jr+|jj|jƒƒt|_ndS(sAFire C{result}'s callback with data accumulated from the process.N(R:RtcallbackR=RF(R((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR?“s ( R R R RFR R=RRCRHR?(((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyRXs    (R,ttwisted.internet.deferRttwisted.internet.protocolRttwisted.internet.errorRtlandscape.manager.pluginRRRt ExceptionRR R(((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyts  D