#!/usr/bin/python

import commands
import os
import time
import sys

if len(sys.argv) <= 1:
	print "Usage: updatettls.py <FILENAME>\n"
	sys.exit(1)
else:
	fname = sys.argv[1]

date = time.strftime("%Y%m%d%H")

ttl = raw_input("What TTL do you want to use? ")
ttl = str(ttl)

#print "Backing up zone file directory..."

#try:
#	os.mkdir('/home/named.backup')
#except:
#	print "Backup directory already exists, skipping."

#cmd = "rsync -aq /var/named /home/named.backup"

#os.system(cmd)

print "Adjusting TTLs..."

#os.chdir("/var/named")
#files = commands.getoutput("ls *.db").split("\n")

f = open(fname, "r")
data = f.readline()

domains = data.split(" ")

for domain in domains:
	domain = domain.rstrip()
	file = "/var/named/%s.db" %domain
	os.system("sed -i -e 's/\$TTL.*/\$TTL %s/' %s" %(ttl, file))
	os.system("sed -i -e 's/[0-9]*\s; minimum TTL/%s\t; minimum TTL/' %s" %(ttl, file))
	os.system("sed -i -e 's/[0-9]\{10\}/%s/' %s" %(date, file))
	os.system("rndc reload %s" %domain)
	#print file
	#print domain

print
print "TTL adjustment done.\n"