U LdS@sUdZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZddlmZmZdd lmZeeZeZd Zd Zd d ddddddddddddddddddd d!d"d#d$d%d&d'gZd(d)d*gd*d+dd,d-ed.gd.d/dd,d0ed0gd0d/dd,d-d1gd-d2dd,d3d4gd5d6dd,d7Zd8d(d9d gd-d:d;did*d?d8iid d-d/d@dAd*gd(d+dBdCdDdEgdEdFdGdHddIdJdKdNdcCst|j}|r.|dkr.td|||iS|dd}i}|dkr|jD]2}||}t|drLtd||}qqL|s|jd}td|||}ntd|||i}|S) a Determine which ntp client is to be used, consulting the distro for its preference. @param ntp_client: String name of the ntp client to use. @param distro: Distro class instance. @returns: Dict of the selected ntp client or {} if none selected. autoz4Selected NTP client "%s" via user-data configuration ntp_clientr#z+Selected NTP client "%s", already installedrzr/LOGdebuggetZ get_optionZpreferred_ntp_clientsrwhich)r@r:Z distro_cfgZdistro_ntp_clientZ clientcfgZclientr;r<r<r=select_ntp_clientJs@         rEcCs(t|rdS|dkrdg}||dS)ahInstall ntp client package if not already installed. @param install_func: function. This parameter is invoked with the contents of the packages parameter. @param packages: list. This parameter defaults to ['ntp']. @param check_exe: string. The name of a binary that indicates the package the specified package is already installed. Nr*)rrD)Z install_funcr%r#r<r<r=install_ntp_client|s rFcCs tj|rt||ddS)zRename any existing ntp client config file @param confpath: string. Specify a path to an existing ntp client configuration file. z.distN)ospathexistsrrenamer$r<r<r=rename_ntp_confs rLcCsjg}|}|dkrd}n|dks&|dkr*d}tdtD]0}|ddd t|g|gd gDq4|S) zGenerate a list of server names to populate an ntp client configuration file. @param distro: string. Specify the distro name @returns: list: A list of strings representing ntp servers for this distro. r rrrr.cSsg|] }|r|qSr<r<).0nr<r<r= sz)generate_server_names..z pool.ntp.org)rangeNR_POOL_SERVERSappendjoinstr)r:namesZ pool_distroxr<r<r=generate_server_namessrYc Cs$|sg}|sg}|sg}|s g}t|dkrDt|dkrD|dkrDdSt|dkr||dkr||dkr|t|}tdd|n2t|dkrt|dkrt|}tdd||std |s|std ||||d } |rtjd d d} | d}tj ||dt ||| |r t |dS)aRender a ntp client configuration for the specified client. @param distro_name: string. The distro class name. @param service_name: string. The name of the NTP client service. @param servers: A list of strings specifying ntp servers. Defaults to empty list. @param pools: A list of strings specifying ntp pools. Defaults to empty list. @param allow: A list of strings specifying a network/CIDR. Defaults to empty list. @param peers: A list nodes that should peer with each other. Defaults to empty list. @param path: A string to specify where to write the rendered template. @param template_fn: A string to specify the template source file. @param template: A string specifying the contents of the template. This content will be written to a temporary file before being used to render the configuration file. @raises: ValueError when path is None. @raises: ValueError when template_fn is None and template is None. rrNrr)z%Adding distro default ntp servers: %s,z*Adding distro default ntp pool servers: %sz Invalid value for path parameterz$Not template_fn or template provided)serverspoolsallowpeersztemplate_name-z.tmpl)prefixsuffix)Zcontent) lenrYrArBrU ValueErrorrZmkstemprZ write_filerZrender_to_fileZdel_file) Z distro_namer&r[r\r]r^rH template_fnr(ZparamsZtfiler<r<r=write_ntp_config_templatesP   recCsHg}tt|}|rrErFrLrYrervrVrtrr<r<r<r=s      !    /      |  5H 2   Q=