2014-03-28 17:58:45 -03:00
|
|
|
from base import Task
|
|
|
|
|
from common import phases
|
|
|
|
|
from common.tasks import packages
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AddNtpPackage(Task):
|
|
|
|
|
description = 'Adding NTP Package'
|
|
|
|
|
phase = phases.package_installation
|
|
|
|
|
successors = [packages.InstallPackages]
|
|
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
|
def run(cls, info):
|
|
|
|
|
info.packages.add('ntp')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class SetNtpServers(Task):
|
|
|
|
|
description = 'Setting NTP servers'
|
|
|
|
|
phase = phases.system_modification
|
|
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
|
def run(cls, info):
|
|
|
|
|
import fileinput
|
|
|
|
|
import os
|
|
|
|
|
import re
|
|
|
|
|
ntp_path = os.path.join(info.root, 'etc/ntp.conf')
|
2014-03-29 11:45:08 -03:00
|
|
|
servers = list(info.manifest.plugins['ntp']['servers'])
|
2014-03-28 17:58:45 -03:00
|
|
|
debian_ntp_server = re.compile('.*[0-9]\.debian\.pool\.ntp\.org.*')
|
|
|
|
|
for line in fileinput.input(files=ntp_path, inplace=True):
|
|
|
|
|
# Will write all the specified servers on the first match, then supress all other default servers
|
|
|
|
|
if re.match(debian_ntp_server, line):
|
|
|
|
|
while servers:
|
|
|
|
|
print 'server {server_address} iburst'.format(server_address=servers.pop(0))
|
|
|
|
|
else:
|
|
|
|
|
print line,
|
