ó ½-'Nc@s ddlZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z ddl mZmZmZmZmZdefd„ƒYZd efd „ƒYZd efd „ƒYZd efd„ƒYZde efd„ƒYZdS(iÿÿÿÿN(tfailt DeferredListtsucceed(tgenerate_script_hash(t Accumulator(t ManagerPlugin(tProcessFailedErrortScriptRunnerMixintProcessTimeLimitReachedErrort get_user_infotUnknownUserErrort StoreProxycBs)eZdZd„Zd„Zd„ZRS(sI Persist-like interface to store graph-points into SQLite store. cCs ||_dS(N(tstore(tselfR ((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyt__init__scCs(|jj|ƒ}|r |dS|SdS(Ni(R tget_graph_accumulate(R tkeytdefaulttgraph_accumulate((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pytgetscCs"|jj||d|dƒdS(Nii(R tset_graph_accumulate(R Rtvalue((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pytsets(t__name__t __module__t__doc__RRR(((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR s  tInvalidFormatErrorcBseZd„Zd„ZRS(cCs#||_tj||jƒƒdS(N(Rt ExceptionRt _get_message(R R((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR$s cCs d|jS(Nu!Failed to convert to number: '%s'(R(R ((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR(s(RRRR(((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR"s t NoOutputErrorcBseZd„ZRS(cCstj|dƒdS(NuScript did not output any value(RR(R ((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR.s(RRR(((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR,stProhibitedUserErrorcBs eZdZd„Zd„ZRS(s Raised when an attempt to run a script as a user that is not allowed. @ivar username: The username that was used cCs#||_tj||jƒƒdS(N(tusernameRRR(R R((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR9s cCs d|jS(Nu%Custom graph cannot be run as user %s(R(R ((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR=s(RRRRR(((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR2s tCustomGraphPlugincBsžeZdZdZdZdZdZdejd„Z d„Z d„Z d„Z d „Z ed „Zd „Zd „Zd „Zd„Zd„Zd„ZRS(sb Manage adding and deleting custom graph scripts, and then run the scripts in a loop. i,ièi s custom-graphcCs5tt|ƒj|ƒ||_i|_t|_dS(sn @param process_factory: The L{IReactorProcess} provider to run the process with. N(tsuperR Rt _create_timet_datatTruetdo_send(R tprocess_factoryt create_time((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyRKs  cCsmtt|ƒj|ƒ|jd|jƒ|jd|jƒt|jjƒ|_ t |j |j ƒ|_ dS(Nscustom-graph-addscustom-graph-remove( R!R tregistertregister_messaget_handle_custom_graph_addt_handle_custom_graph_removeR tregistryR t_persistRt run_intervalt _accumulate(R R,((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR(Us  cCsxt|dƒ}|jjj|ƒ}|rE|d}tj|ƒn|jjj|ƒ||jkrt|j|=ndS(sk Handle remove custom-graph operation, deleting the custom graph scripts if found. sgraph-idiN(tintR,R t get_graphtostunlinkt remove_graphR#(R tmessagetgraph_idtgraphtfilename((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR+^s c Cs3|d}|d}|d}t|dƒ}|jjj}tjj|dƒ}tjj|d|fƒ}tjj|ƒrtj|ƒnyt |ƒd \} } Wn"t k rÎt j d|ƒnHXt |d ƒ} |j| |||| | ƒ||jkr|j|=n|jjj|||ƒd S( ss Handle add custom-graph operation, which can also update an existing custom graph script. Rt interpretertcodesgraph-idscustom-graph-scriptssgraph-%diu)Attempt to add graph with unknown user %stwN(R0R,tconfigt data_pathR2tpathtjointexistsR3R R tloggingterrortfiletwrite_script_fileR#R t add_graph( R R5tusertshellR:R6R=tscripts_directoryR8tuidtgidt script_file((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR*ms*        cCsd|jj|jdfS(Nu%s: %si(t __class__Rtargs(R te((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyt_format_exceptionŒscCs#|jjj|j|j|ƒdS(N(R,tbrokertcall_if_acceptedt message_typet send_message(R turgent((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pytexchanges c Cs/|js dSt|_t|jjjƒƒ}xo|D]g\}}}||jkr5tjj |ƒrœ|j |ƒ}igd6dd6|d6|j|tisfilet_get_script_hashRRt iteritemsRPRS( R RTtgraphsR6R8RFt script_hashR5tnew_datatitem((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyRS“s$  )   cCs||jkrdSyt|ƒ}Wn/tk rT|rHt|ƒ‚qUtƒ‚nX|j|||ƒ}|r‹|j|dj|ƒndS(NRV(R#tfloatt ValueErrorRRR/tappend(R toutputR6tnowRXt step_data((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyt _handle_dataªs  cCs¾||jkrdS|jtƒrp|jjjdƒ}|jjr\d||jjf}n||j|dR\R]R#R R Rt addErrbackRpRetis_user_allowedRt _run_scriptRmt addCallbackRiR(R R_t deferred_listRgR6R8RFR`RIRJR>RNtdtresult((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyRuÞs<"   N(RRRR.t size_limitRmRRtNonettimeRR(R+R*RORYRURSRiRpR]RvRu(((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyR As"         (R2RRAttwisted.internet.deferRRRtlandscape.lib.scriptcontentRtlandscape.accumulateRtlandscape.manager.pluginRt!landscape.manager.scriptexecutionRRRR R tobjectR RRRRR (((sA/usr/lib/python2.7/dist-packages/landscape/manager/customgraph.pyts   (