['dZddlmZmZGddeZdZdaddZ dd Z dd Z d Z d Z d Z dZGddeZdS)z Task functions This module provies synchronization of client/server operations since many VIM operations return 'tasks' which can have varying completion times. )vmodlvimceZdZdZdS) TaskBlockedzk Exception class to represent when task is blocked (e.g.: waiting for an answer to a question. N)__name__ __module__ __qualname____doc__,/usr/lib/python3/dist-packages/pyVim/task.pyrr&s Dr rc  t|jjtrc|j}t|tsd|j|jfz}t d|jjjd|jd|j d|dSdS)Nz %d%% (%s)Task z (key:z, desc:z) - ) isinstanceinfoprogressintstrstateprintnamekey description)taskrrs r TaskUpdatesVerboser2s$)$c**Hy(C(( A"dmTZ%@@H  IN   4+;+;+;XXG H H H H H HHr NTc |r tadSdadSN)rglobalTaskUpdate)verboses r SetTasksVerboser >s" -r cv|tjd|j}| |jj}t ||}|dt||}d\}}|tjj j tjj j fvr t|||\}}| n8#tjj$r!} t#d| jzYd} ~ n6d} ~ wwxYw|tjj j tjj j fv||dkrg|dt)|jj z|r |jj t#dt)|jj zn|d |S) a Wait for task to complete. @type raiseOnError : bool @param raiseOnError : Any exception thrown is thrown up to the caller if raiseOnError is set to true. @type si : ManagedObjectReference to a ServiceInstance. @param si : ServiceInstance to use. If None, use the information from the task. @type pc : ManagedObjectReference to a PropertyCollector. @param pc : Property collector to use. If None, get it from the ServiceInstance. @type onProgressUpdate : callable @param onProgressUpdate : Callable to call with task progress updates. For example:: def OnTaskProgressUpdate(task, percentDone): print 'Task %s is %d%% complete.' % (task, percentDone) NServiceInstancecreatedNNz Task object has been deleted: %serror error: %szTask reported error: completed)rr"_stubcontentpropertyCollectorProgressUpdaterUpdate CreateFilterTaskInfoStatesuccessr% GetTaskStatusUpdateIfNeededrfaultManagedObjectNotFoundrobjDestroyrr) r raiseOnErrorsipconProgressUpdateprogressUpdaterfilterversionres r WaitForTaskr?Xs4 z  !2DJ ? ? z Z )%d,<==O9%%% "d # #FNGU  *2CL4F4LM M M *4"==NGU  * * , , , ,{0    4qu< = = = EEEE   *2CL4F4LM M M NN {S-A-AABBB  B)/ ! )C ,@,@@ A A A A{+++ Ls(B==C2C--C2c |sdS| tjd|dj}| |jj}|g}i}|D]9}t ||}|d||t|<:t||} d\} } t|r| | } | j D]w} | j D]k}|j }t|}|jD]I}|jdkr |jj} n|jdkr|j} n/||}|sG| tjjjkrJ|d||||jj| tjjjkrn|jj}|d t|z|r|t7d |d t|||3|r|Kmy| j} t|| r| n#| r| wwxYwdS) z` Wait for mulitiple tasks to complete. Much faster than calling WaitForTask N times Nr"rr#r$rz info.stater'r&rz reported error: )rr"r(r)r*r+r,rCreateTasksFilterlenWaitForUpdates filterSet objectSetr5 changeSetrvalrgetr.r/r0popappendrresultr%rr2r=r6)tasksr7r8r9r:resultsprogressUpdatersrr;r<r=rupdaterDobjSettaskIdchangeerrs r WaitForTasksrTs  z  !2E!HN C C z Z )66)$0@AAy)))&5T## r5 ) )F+#"##$ %&&w//F#- A A '1AAF!:D YYF"("2AA!;&00$*J$4EE#[L88$*JEE$*:*>*>v*F*F.%$ CL$6$>>>+22;???,00888$NN49+;<<<<"cl&8&>>>"&)/C+22;S3IJJJ+=&) %fffcRUhhh&W X X X 0 4 4V < < < </A / > > @ @ @9AABnGI"##$ %L   NN      NN      Fs GI I9c||}|jj}|dkrO|jjC|jjjjdkr)|jjjjdkrt ||j|fS)Nrunningr6Relocate)rCrrrCheckForQuestionPendingr=)rr=r9rOrs r r1r1sw   w ' 'F IOE ty~9din>Q>VZc>c>c IN  $ 2 2%%% >5  r c$t||gS)z+ Create property collector filter for task )rA)r9rs r r-r-s R$ ( ((r c|sdSd|D}tjjtjgd}tjj}||_|g|_| |dS)z, Create property collector filter for tasks NcXg|]'}tjj|(S))r5)rqueryPropertyCollector ObjectSpec).0rs r z%CreateTasksFilter..s>### -88T8BB###r T)typepathSetall) rr\r] PropertySpecrTask FilterSpecrEpropSetr-)r9rLobjspecspropspec filterspecs r rArAs t##!###H{,99 Xrt:--H.99;;J#J"J ??:t , ,,r c|jj}|7t|tjr|jj}|tddSdSdS)zE Check to see if VM needs to ask a question, throw exception Nz(Task blocked, User Intervention required)rentityrrVirtualMachineruntimequestionr)rvmqsts r rXrX sR  B ~*R);<<~j! ?HII I~~~ ?r c$eZdZdZdZdZdZdS)r+zo Class that keeps track of task percentage complete and calls a provided callback when it changes. c>||_||_d|_d|_dS)Nr)rr: prevProgressr)selfrr:s r __init__zProgressUpdater.__init__#s$ 0 r cZt}|jr|j}|r||j|dSdSr)rr:r)rur taskUpdates r r,zProgressUpdater.Update)sF%   /.J  ) Jty% ( ( ( ( ( ) )r c|jjj|_|j|jkr||j|j|_dSr)rrrrtr,)rus r r2zProgressUpdater.UpdateIfNeeded1sD / =D- - - KK & & & Mr N)rrr r rvr,r2r r r r+r+sK  )))*****r r+)T)TNNN)TNNNN)r pyVmomirr Exceptionrrrr r?rTr1r-rArXobjectr+r r r r}sA.     )   HHH    6"!% 9999H#"& I I I I X!!!))) ---, J J J *****f*****r