o bJ@s ddlZddlZddlZddlmZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZd d Zd d Zd dZddZddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd/d'd(Zd)d*Z d/d+d,Z!d-d.Z"dS)0N)cli)failfail_conflicting print_stdout print_stderr) DeviceConsole)Guest)log)xmlutilcCsX |d}tj|tjtj}|dvrdS|dvr%dStt dq)Nrz (y/n): )yyesT)nnoFzPlease enter 'yes' or 'no'.) sysstdoutwriteflushstdinreadlinelowerstripr_)msgprintmsginpr+/usr/share/virt-manager/virtinst/virtxml.pyprompt_yes_or_nos   rcCs2t||dd}|rtd||Std|S)Nz Original XMLz Altered XMLz XML diff: %sz(No XML diff, didn't generate any change.)r diffr debug)origxmlnewxmlrrrrget_diff&s   r#cCs8|jdurdSt|j}|r||dSdSN) os_variantrparse_os_variantget_name set_os_name)optionsguestosdatarrrset_os_variant0s  r,cCs&tj|dd}t||d}||kS)NT)inactiveparsexml)r get_xmldescr get_xml)conndomain original_xmlrawxmlnew_xmlrrrdefined_xml_is_unchanged9sr7cCs t||}d}|durd}n|s"|dr&|ddr&t|}|durj|dkr6ttd||sAttd|jt |t |kr\tt ddt ||t ||jd|dkrd|d8}||}|S|d krv|dd}|S|||d }| }|sttd d ||f|S) Nr-rzInvalid --edit option '%s'z No --%s objects found in the XMLzS'--edit %(number)s' requested but there's only %(max)s --%(type)s object in the XMLzV'--edit %(number)s' requested but there are only %(max)s --%(type)s objects in the XML)numbermaxtypeall)r*z No matching objects found for %sz--%s %s) r listify lookup_propisdigit startswithintrr cli_arg_namelenabsngettextlookup_child_from_option_string)r* action_nameeditval parserclassobjlistidxinst parserobjrrr_find_objects_to_editCsP   rNcCsgd}g}|D]}|dd}t||dvr||qt|dkr3ttdddd |Dt|d krKttd dd d |DdSdS) N)editz add-device remove-devicez build-xmlr8r)FrzOne of %s must be specified.z, cSg|]}d|qSz--r.0crrr |z*check_action_collision..rzConflicting options %scSrRrSrrTrrrrWrX)replacegetattrappendrCrrjoin)r)actions collisionsclinameoptnamerrrcheck_action_collisionqs      racCstg}tjtjgD] }t||jr||q t|dkr#ttdt|dkr6ttddd|D|dS)NrzNo change specified.rzCOnly one change operation may be specified (conflicting options %s)cSsg|]}|qSr) cli_flag_namerTrrrrWrXz*check_xmlopt_collision..) r VIRT_PARSERS ParserXMLrZrBr[rCrr)r)r^rIrrrcheck_xmlopt_collisions      recCsx|jr t|d|j|}n|}|jr(|jdkr(|jdkr(ttd|j|jd|jdur3ttdtj|||ddS) NrO1r<zU'--edit %(option)s' doesn't make sense with --%(objecttype)s, just use empty '--edit')option objecttypez2--os-variant/--osinfo is not supported with --editT)editing) guest_propnamerNrOrrrBr%rparse_option_strings)r*r)rIrLrrr action_edits  rlcCsr||sttd|jt|||r!|D]}||q|St||d}t |}|D]}| |q/|S)Nz!Cannot use --add-device with --%s) prop_is_listrrrBr, add_devicerrkr r= set_defaults)r*r)rIdevsdevrrraction_add_devices     rrcCs||sttd|j|jdurttdt|dt||jd|}t|}|ddD]}t ||}|rHt d| ||q1|D]}||qK|S)Nz$Cannot use --remove-device with --%sz;--os-variant/--osinfo is not supported with --remove-devicerPrQz"Found duplicate console device: %s)rmrrrBr%rNrZr r=rget_console_duplicater r r1r[ remove_device)r*r)rIrprqconduprrraction_remove_devices&       rvcCs|js ttd|j|jdurttd||}||r&|}n||}t |||}t |}|D]}| |q9|S)Nz"--build-xml not supported for --%sz7--os-variant/--osinfo is not supported with --build-xml)rjrrrBr%r>rmnew __class__rrkr r=ro)r2r)rIr*rLrprqrrraction_build_xmls        rycCs2t|dddkr dStd||tdS)N DEVICE_TYPEdiskzDoing setup for disk=%s)rZr r build_storager get_meter)rqrrr setup_devices r~cCsX|r ttd|js dS|dkr|D]}t|q||}ttd|j|S)Nz!Define '%s' with the changed XML?Fhotplugz!Domain '%s' defined successfully.)rrnamer~ defineXMLr1r)r2inactive_xmlobjrpactionconfirmrqdomrrrdefine_changess  rc Cs|r ttd|js dS|dkr|D]}t|qz ||}Wn tjyC}zttd|j|dWYd}~dSd}~wwt td|j|S)Nz Start '%s' with the changed XML?Fr.Failed starting domain '%(domain)s': %(error)sr3error!Domain '%s' started successfully.) rrrr~ createXMLr1libvirt libvirtErrorrr)r2xmlobjrprrrqrerrrstart_domain_transients&  rc CsJ|dkrtd}td}td}n!|dkr"td}td}td}n|d kr2td }td }td }|D]n}|}|rJ|||d } t| sJq4|dkrRt|z%|dkr_||tjn|dkrk||tjn |d krv| |tjWn tj y} zdt j vrt |d| iWYd} ~ nd} ~ wwt||rtdq4dS)Nrz7%(xml)s Hotplug this device to the guest '%(domain)s'?zDevice hotplug successful.z*Error attempting device hotplug: %(error)s hotunplugz;%(xml)s Hotunplug this device from the guest '%(domain)s'?zDevice hotunplug successful.z,Error attempting device hotunplug: %(error)supdatez7%(xml)s Update this device for the guest '%(domain)s'?zDevice update successful.z)Error attempting device update: %(error)s)xmlr3$VIRTXML_TESTSUITE_UPDATE_IGNORE_FAILr)rr1rrr~attachDeviceFlagsrVIR_DOMAIN_AFFECT_LIVEdetachDeviceFlagsupdateDeviceFlagsrosenvironrr) r3rprr msg_confirm msg_successmsg_failrqrrrrrrupdate_changes sR   rc Cs|}|j|j|d}|jdk}|tju}|r |s ttd|r6|r-g}t||nt |||}d}n|j rCt ||||}d}n |j rNt |||}d}|} t|| } | s`ttd|jrj| rit| n|jrqt| |||fS)z Parse the command line device/XML arguments, and apply the changes to a copy of the passed in xmlobj. :returns: (list of device objects, action string, altered xmlobj) )r2r/rQz"--xml can only be used with --editrrrzENo XML diff was generated. The requested changes will have no effect.)r1rxr2rOrrdrr parse_xmlclirlrnrrrtrvr#r warning print_diffr print_xml) orig_xmlobjr)rIrpr!rhas_edit is_xmlclirr"rrrrprepare_changes:s:       rcCstjdtddd}t|d|jddtdd |td }|jd dd td d|jddtdd|jddtdd|jddtdd|td}|jddtdd|}|jddtdd|jdddtdd |jdd!|jd"dtd#d|jd$dtd%d|jd&dtd'd|jd(dtd)dtj|d*d+|td,}tj |dd-t |t |t |t |tj|dd-t|t|t|t||td.}tj|d*d*d*d/t||S)0Nz%(prog)s [options]z,Edit libvirt XML using command line options.T)introspection_epilogvirt-xmlr3?zDomain name, id, or uuid)nargshelpz XML actionsz--editrQzEdit VM XML. Examples: --edit --disk ... (edit first disk device) --edit 2 --disk ... (edit second disk device) --edit all --disk ... (edit all disk devices) --edit target=hda --disk ... (edit disk 'hda') )rdefaultrz--remove-device store_truezRemove specified device. Examples: --remove-device --disk 1 (remove first disk) --remove-device --disk all (remove all disks) --remove-device --disk /some/path)rrz --add-devicez6Add specified device. Example: --add-device --disk ...z --build-xmlzWOutput built device XML. Domain is optional but recommended to ensure optimal defaults.zOutput options--updatezApply changes to the running VM. With --add-device, this is a hotplug operation. With --remove-device, this is a hotunplug operation. With --edit, this is an update device operation.z--definezKForce defining the domain. Only required if a --print option was specified.z --no-definedefine store_falsezForce not defining the domain.)destrr)r--startzStart the domain.z --print-diffz/Only print the requested change, in diff formatz --print-xmlz3Only print the requested change, in full XML formatz --confirmz/Require confirmation before saving any results.F) virtinstallz XML options) editexamplezMiscellaneous Options)promptprintxmldryrun)r setupParserradd_connect_option add_argumentadd_argument_groupadd_mutually_exclusive_grouproadd_os_variant_optionadd_disk_optionadd_net_optionadd_gfx_optionadd_metadata_optionadd_memory_optionvcpu_cli_optionsadd_xml_optionadd_guest_xml_optionsadd_boot_optionsadd_device_optionsadd_misc_options autocomplete parse_args)parseractgoutgdefine_ggmiscrrrrms|             rc Cstt}|js|js|js|jrd|_td|j |jt |r&dSt |r-dSd|_ |j s^|js^tjjsXtjsX|jrHttd|jrQttdtj|_ nttd|jsv|jsv|j rn|jsmd|_n|jdurvd|_|jr|jsd|_t|t|}|jr|jsttd|jt|j|}d}d}d}|j rt||j \}}}nt||j d }t|}|jrt ||||}|D]}t!|"qdSd}d} |jr |jr|j#rt$d d |rt%|||\}} } t&||| |jd} nt'(td |js dS|"} t%||||d \}} } |js+|j#r)t)|| || |jdSt*|| || |j}|s9dS|j#roz|+Wn t,j-yc}zttd|j.|dWYd}~nd}~wwt!td|j.dS|r}| s}t!tddSt/||| rt'(tddS)NFrrz%Can't use --confirm with stdin input.z$Can't use --update with stdin input.zA domain must be specifiedTz#Don't know how to --update for --%sr.rrz0The VM is not running, --update is inapplicable.)rprrrz?Changes will take effect after the domain is fully powered off.zhXML did not change after domain define. You may have changed a value that libvirt is setting by default.)0r earlyLoggingrrrr build_xmlquiet setupLoggingr check_option_introspectioncheck_osinfo_liststdinxmlr3rrclosedisattyrrrreadrrarerjrB getConnectionconnectget_domain_and_guestr boolryrr1startrrrr rrrcreaterrrr7)r2r)rIr3 active_xmlobjr vm_is_runningrprqperformed_updaterdummyr4xmlobj_to_definerrrrrmains                  rc Csz ttWdSty"}z t|jWYd}~dSd}~wty8tjdddtt dYdSt yN}z t |WYd}~dSd}~ww)NrT)exc_infozAborted at user request) rexitr SystemExitcodeKeyboardInterruptr r rr Exceptionr)sys_emain_errrruncli9s rr$)#rrrrrrrrrdevicesrr*r loggerr r rr#r,r7rNrarerlrrrvryr~rrrrrrrrrrrs8        . /3 N ~