ó ½-'Nc@sÆdZddlZddlZddlZddlZddlZddlZddlmZmZ ddl m Z ddl m Z ddl mZmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZdefd„ƒYZ de fd„ƒYZ!de"d„Z#d„Z$defd„ƒYZ%de&fd„ƒYZ'd„Z(d„Z)d„Z*d„Z+e,d„Z-d„Z.d„Z/dS( s¤Interactive configuration support for Landscape. This module, and specifically L{LandscapeSetupScript}, implements the support for the C{landscape-config} script. iÿÿÿÿN(t ConfigParsertError(tStringIO(t is_valid_tag(t SysVConfigt ProcessError(tMethodCallError(tgather_results(tfetcht FetchError(tTwistedReactor(tInvalidCredentialsError(tBrokerConfiguration(tRemoteBrokerConnectortConfigurationErrorcBseZdZRS(s6Raised when required configuration values are missing.(t__name__t __module__t__doc__(((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyRstImportOptionErrorcBseZdZRS(s?Raised when there are issues with handling the --import option.(RRR(((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyR ss cCs:|rtj}n tj}|j||ƒ|jƒdS(N(tsyststderrtstdouttwritetflush(ttexttendterrortstream((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt print_text$s   cCsÂ|dk r¾g|jdƒD]}|jƒ^q}d|krqt|ƒdkratdƒ‚n|jdƒng}x@|D]8}ytj|ƒWq~tk rµ|j |ƒq~Xq~W|SdS(sŽ Process a string with a list of comma separated usernames, this returns any usernames not known to the underlying user database. t,tALLis$Extra users specified with ALL usersN( tNonetsplittstriptlenRtremovetpwdtgetpwnamtKeyErrortappend(tuserstusert user_listt invalid_users((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytget_invalid_users-s (    tLandscapeSetupConfigurationcBs)eZdZd„Zd„Zd„ZRS( tno_starttdisabletsilenttok_no_registert import_fromcCs tt|ƒjƒ||_dS(N(tsuperR-t__init__t_fetch_import_url(tselftfetch_import_url((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyR4GscCsW|jrStƒ}y®d|jkr„|jr@|jtjdx7tr9t|ƒ}|r|S|s)Pn|jdƒqWdS(sœPrompt the user on the terminal for a value @param msg: Message to prompt user with @param required: True if value must be entered s/This option is required to configure Landscape.N(tTruet raw_inputR`(R6tmsgtrequiredtvalue((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytprompt_get_input¤s  cCs{t|j|dƒ}|r,|d|7}n |d7}|oFt|ƒ }|j||ƒ}|rwt|j||ƒndS(stPrompt the user on the terminal for a value. @param option: The attribute of C{self.config} that contains the default and which the value will be assigned to. @param msg: The message to prompt the user with (via C{raw_input}). @param required: If True, the user will be required to enter a value before continuing. s [%s]: s: N(tgetattrR[RtboolRftsetattr(R6toptionRcRdRRtresult((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytprompt²s  cCs®t|j|dƒ}|d7}xˆtr©tj|ƒ}|rOtjdƒ}n|rˆ||krq|jdƒq¦t|j||ƒPq"|s•| r™Pq"|jdƒq"WdS(sÛPrompt the user on the terminal for a password and mask the value. This also prompts the user twice and errors if both values don't match. @param option: The attribute of C{self.config} that contains the default and which the value will be assigned to. @param msg: The message to prompt the user with (via C{raw_input}). @param required: If True, the user will be required to enter a value before continuing. s: sPlease confirm: sPasswords must match.s/This option is required to configure Landscape.N(RgR[RRatgetpassR`Ri(R6RjRcRdRRRetvalue2((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytpassword_promptÅs     cCs||rd}nd}x`trwt||ƒjƒ}|rp|jdƒrMtS|jdƒr`tS|jdƒq|SqWdS(Ns [Y/n]s [y/N]tntysInvalid input.(RaRbtlowert startswithRXR`(R6tmessageRRt default_msgRe((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt prompt_yes_noâs  cCs=d|jjƒkrdS|jdƒ|jddtƒdS(Ntcomputer_titles% The computer title you provide will be used to represent this computer in the Landscape user interface. It's important to use a title that will allow the system to be easily recognized when it appears on the pending computers page. sThis computer's title(R[tget_command_line_optionsR`RlRa(R6((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytquery_computer_titleòs cCs=d|jjƒkrdS|jdƒ|jddtƒdS(Nt account_names You must now specify the name of the Landscape account you want to register this computer with. You can verify the names of the accounts you manage on your dashboard at https://landscape.canonical.com/dashboard s Account name(R[RxR`RlRa(R6((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytquery_account_names cCsDd|jjƒkrdS|jd|jjƒ|jddƒdS(Ntregistration_passwords¶ A registration password may be associated with your Landscape account to prevent unauthorized registration attempts. This is not your personal login password. It is optional, and unless explicitly set on the server, it may be skipped here. If you don't remember the registration password you can find it at https://landscape.canonical.com/account/%s sAccount registration password(R[RxR`RzRo(R6((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytquery_registration_passwords   cCsz|jjƒ}d|kr+d|kr+dS|jdƒd|krW|jddƒnd|krv|jddƒndS(NR8R9s> The Landscape client communicates with the server over HTTP and HTTPS. If your network requires you to use a proxy to access HTTP and/or HTTPS web sites, please provide the address of these proxies now. If you don't use a proxy, leave these fields empty. sHTTP proxy URLsHTTPS proxy URL(R[RxR`Rl(R6RJ((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt query_proxies s  cCs—|jjƒ}d|kr]d|kr]t|dƒ}|rYtddj|ƒƒ‚ndS|jdƒd}g|jjjdƒD]}|jƒ^q†}|dgkr¶g}nd |k}|j |d |ƒrbd |kró|j d ƒn|jd ƒx{t r^|j dd ƒt|jj ƒ}|s5Pq|jddj|ƒƒd|j_ qWnd |kr~|jd ƒndj|ƒ|j_dS( Ntinclude_manager_pluginst script_userssUnknown system users: %ss, sb Landscape has a feature which enables administrators to run arbitrary scripts on machines under their control. By default this feature is disabled in the client, disallowing any arbitrary script execution. If enabled, the set of users that scripts may run as is also configurable. sEnable script execution?RRStScriptExecutionRRs! By default, scripts are restricted to the 'landscape' and 'nobody' users. Please enter a comma-delimited list of users that scripts will be restricted to. To allow scripts to be run by any user, enter "ALL". s Script users(R[RxR,RR]R`RR R!RvR'RaRlR€RR#(R6RJR+Rctptincluded_pluginsRR((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytquery_script_plugin2s>.      cCs`g}|r\g|jdƒD]}|jƒ^q}g|D]}t|ƒs;|^q;}n|S(sm Splits a string on , and checks the validity of each tag, returns any invalid tags. R(R R!R(R6ttagnamest invalid_tagsttagttags((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt_get_invalid_tagsas ((cCs·|jjƒ}d|krT|j|dƒ}|rPtddj|ƒƒ‚ndS|jdƒxOtr²|jddtƒ|j|jj ƒr®|jdƒd|j_ qdPqdWdS(sQuery tags from the user.RˆsInvalid tags: %ss, Ns9You may provide tags for this computer e.g. server,hardy.tTagss3Tag names may only contain alphanumeric characters.( R[RxR‰RR]R`RaRlRXRˆR(R6RJR†((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt query_tagsls    cCs|jdƒdS(NsÖ This script will interactively set up the Landscape client. It will ask you a few questions about this computer and your Landscape account, and will submit that information to the Landscape server. After this computer is registered it will need to be approved by an account administrator on the pending computers page. Please see https://landscape.canonical.com for more information. (R`(R6((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt show_headers cCsJ|jƒ|jƒ|jƒ|jƒ|jƒ|jƒ|jƒdS(syKick off the interactive process which prompts the user for data. Data will be saved to C{self.config}. N(RŒRyR{R}R~R„R‹(R6((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytruns      (RRRR4R`RfRXRlRoRaRvRyR{R}R~R„R‰R‹RŒR(((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyRZ‘s             /  cCstƒ}|jtƒdS(N(Rtset_start_on_bootRa(t sysvconfig((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt"setup_init_script_and_start_client›s cCs$tƒ}|jƒ|jtƒdS(N(Rtstop_landscapeRŽRX(R((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt#stop_client_and_disable_init_script¡s  c Cshtƒ}|jso|jr%tƒqo|jƒsotdƒ}|jƒjdƒs\tƒqltj dƒqon|j dkr£t j jdƒr£t j d|_ n|jdkr×t j jdƒr×t j d|_n|jr€|jdƒ r|jdƒ s|jdƒ rtd ƒ‚n|jd ƒr–t|jd ƒƒ}|retd d j|ƒƒ‚n|js}d |_q}q–nt|ƒ}|jƒ|jrÜ|jjdƒrÜtj|jdƒ}t||ƒ|_n|jƒ|j rd|j rdy|jƒWqdtk r`tddtƒtddtƒd}|j rPd}ntj |ƒqdXndS(Nsl The Landscape client must be started on boot to operate correctly. Start Landscape client on boot? (Y/n): tNs Aborting Landscape configurationR8R9totpRzRws0An account name and computer title are required.R€sUnknown system users: %ss, Rsbase64:is&Couldn't restart the Landscape client.RsOThis machine will be registered with the provided details when the client runs.ii(!RR.R0Rtis_configured_to_runRbtupperRsRtexitR8RR:R;tgetR9RR,R]RRZRtssl_public_keytbase64t decodestringtstore_public_key_dataRR”trestart_landscapeRRRaR1(R[RtanswerR+tscriptt decoded_certt exit_code((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytsetup§sT       !!          cCsftjj|jtjj|jƒdƒƒ}td|ƒt|dƒ}|j|ƒ|j ƒ|S(s° Write out the data from the SSL certificate provided to us, either from a bootstrap.conf file, or from EC2-style user-data. @param config: The L{BrokerConfiguration} object in use. @param certificate_data: a string of data that represents the contents of the file to be written. @return the L{BrokerConfiguration} object that was passed in, updated to reflect the path of the ssl_public_key file. s.ssl_public_keys#Writing SSL CA certificate to %s...tw( R:R=R]t data_pathtbasenametget_config_filenameRtopenRtclose(R[tcertificate_datat key_filenametkey_file((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyRœàs   cstƒ‰g‰‡‡fd†‰‡fd†‰‡fd†‰‡fd†‰‡fd†‰‡fd†‰tddƒtjd ƒ‡‡‡‡‡fd †}‡‡‡ fd †}tˆˆ ƒ‰ˆjd d dtƒ}|j|ƒ|j|ƒˆj ƒˆrt j ˆd ƒn|S(sMInstruct the Landscape Broker to register the client. The broker will be instructed to reload its configuration and then to attempt a registration. @param reactor: The reactor to use. Please only pass reactor when you have totally mangled everything with mocker. Otherwise bad things will happen. csˆjƒˆjƒdS(N(t disconnecttstop((t connectortreactor(s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyR­s cstddtƒˆƒdS(Ns.Invalid account name or registration password.R(RRa((R­(s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytfailures cstdƒˆƒdS(NsSystem successfully registered.(R((R­(s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytsuccess s cstddtƒˆƒdS(NsšWe were unable to contact the server. Your internet connection may be down. The landscape client will continue to try and contact the server periodically.R(RRa((R­(s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytexchange_failures cs|jttƒˆjƒdS(N(ttrapR RR¬(R°(R®(s;/usr/lib/python2.7/dist-packages/landscape/configuration.pythandle_registration_errorsscs1t|jƒdtƒtddtƒˆƒdS(NRsUnknown error occurred.(Rt getTracebackRa(R°(R­(s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt catch_allssPlease wait... RSicsgiˆd6ˆd6ˆd6}|jƒ|j|ƒ|jƒjˆƒg}t|dtƒ}|jˆƒS(Nsregistration-donesregistration-failedsexchange-failedtconsume_errors(treload_configurationt call_on_eventtregistert addErrbackRRa(tremotethandlerst deferredstresults(R±R¶R°R²R´(s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytgot_connection+s    csDtddtƒtddtƒˆjs9ˆjdƒnˆƒdS(Ns;There was an error communicating with the Landscape client.RsOThis machine will be registered with the provided details when the client runs.i(RRaR1R'(R°(texit_with_errorR­R[(s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyt got_error8s   t max_retriesitquiet( R RttimetsleepR tconnectRat addCallbackR»RRR—(R[R¯RÀRÂRk(( R®R¯RÁR­R±R¶R°R²R´R[s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyRºôs(        cCsotd|ƒd}yt|ƒ}Wntk rE}t|ƒ}nX|dk rktd||fƒ‚n|S(s´Handle fetching of URLs passed to --url. This is done out of LandscapeSetupConfiguration since it has to deal with interaction with the user and downloading of files. s!Fetching configuration from %s...s+Couldn't download configuration from %s: %sN(RRRR RAR(turlt error_messageRIR((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyR7Ns cCs=ttƒ}|dgdgfkr4|j|ƒntjƒdkrVtjdƒny|j|ƒWn6tk rŸ}tt |ƒdt ƒtjdƒnX|j r´t ƒdSyt |ƒWn0tk rô}tt |ƒƒtjdƒnX|jr t|ƒn.tdƒ}|jƒjd ƒs9t|ƒndS( Ns-hs--helpis%landscape-config must be run as root.Ris Aborting Landscape configurations: Request a new registration for this computer now? (Y/n): R“(R-R7tloadR:tgetuidRR—RRRARaR/R’R¢t ExceptionR0RºRbR–Rs(targsR[RteRž((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pytmainbs.     (0RRšRÅRR:RmR$RRR@Rtlandscape.lib.tagRtlandscape.sysvconfigRRtlandscape.lib.ampRtlandscape.lib.twisted_utilRtlandscape.lib.fetchRR tlandscape.reactorR tlandscape.broker.registrationR tlandscape.broker.configR tlandscape.broker.ampR RÍRRRXRR,R-tobjectRZRR’R¢RœRRºR7RÐ(((s;/usr/lib/python2.7/dist-packages/landscape/configuration.pyts>       Oÿ   9  Z