ó ½-'Nc@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   cCs tjdƒ|jt||ƒS(NsShutdown request succeeded.(tloggingtinfot_respondR(RRR%((sE/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!8s cCs8idd6|d6|d6|d6}|jjj|tƒS(Nsoperation-resultttypetstatuss result-texts operation-id(Rtbrokert send_messagetTrue(RR.RR%R$((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR*<s   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"Cs    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(RRRR2t_dataR1t_waiting(RRR2((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR es     cCsdj|jƒS(s'Get the data printed by the subprocess.t(tjoinR4(R((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pytget_datalsi 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.pyRpscCs |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(R5R4tappend(RtfdR((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pytchildDataReceivedws 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( R5tcheckRR:RterrbackRR8tFalse(Rtreason((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyt processEndeds   cCs/|jr+|jj|jƒƒt|_ndS(sAFire C{result}'s callback with data accumulated from the process.N(R5RtcallbackR8RA(R((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyR:s ( R R R RAR R8RR>RCR:(((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyRRs    (R(ttwisted.internet.deferRttwisted.internet.protocolRttwisted.internet.errorRtlandscape.manager.pluginRRRt ExceptionRR R(((sE/usr/lib/python2.7/dist-packages/landscape/manager/shutdownmanager.pyts  >