1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
#!/usr/bin/python
import sys, os
from datetime import date
os.environ['DJANGO_SETTINGS_MODULE'] = 'Hostbase.settings'
from Hostbase.hostbase.models import *
from Hostbase.settings import DEFAULT_MX, PRIORITY
host_attribs = ['hostname', 'whatami', 'netgroup', 'security_class', 'support',
'csi', 'printq', 'dhcp', 'outbound_smtp', 'primary_user',
'administrator', 'location', 'expiration_date', 'comments']
def checkformat(values):
filelist = []
for pair in values:
filelist.append(pair[0])
lines = len(filelist)
while True:
if filelist and not filelist[0:14] == host_attribs:
# figure out what to do here
return 0
sys.exit()
elif not filelist:
return 1
else:
filelist = filelist[14:]
while True:
if filelist and filelist[0] == 'mac_addr':
filelist.pop(0)
if filelist and filelist[0] == 'hdwr_type':
filelist.pop(0)
while filelist and filelist[0] == 'ip_addr':
filelist.pop(0)
while filelist and filelist[0] == 'cname':
filelist.pop(0)
if (filelist and filelist[0] == 'hostname') or not filelist:
break
# argument handling for batchadd
try:
fd = open(sys.argv[1], 'r')
except (IndexError, IOError):
print "\nUsage: batchadd.py filename\n"
sys.exit()
lines = fd.readlines()
info = []
for line in lines:
if not line.lstrip(' ')[0] == '#' and not line == '\n':
info.append(line.split("->"))
for x in range(0,len(info)):
if len(info[x]) > 1:
info[x][0] = info[x][0].strip()
info[x][1] = info[x][1].strip()
else:
print "Error: file format"
if info[0][0] == 'mx' and info[1][0] == 'priority':
mx, created = MX.objects.get_or_create(mx=info[0][1], priority=info[1][1])
try:
info.pop(0)
info.pop(0)
except:
print "Error: file format"
sys.exit()
else:
mx, created = MX.objects.get_or_create(mx=DEFAULT_MX, priority=PRIORITY)
if created:
mx.save()
if not checkformat(info):
print "Error: file format"
sys.exit()
while True:
blank = Host()
for attrib in host_attribs:
try:
pair = info.pop(0)
except:
sys.exit()
if pair[0] == 'dhcp' or pair[0] == 'outbound_smtp':
if pair[1] == 'y':
blank.__dict__[pair[0]] = 1
else:
blank.__dict__[pair[0]] = 0
elif pair[0] == 'expiration_date':
(year, month, day) = pair[1].split("-")
blank.expiration_date = date(int(year), int(month), int(day))
elif pair[0] in host_attribs:
blank.__dict__[pair[0]] = pair[1]
try:
host = Host.objects.get(hostname=blank.hostname)
print "Error: %s already exists in hostbase" % blank.hostname
sys.exit()
except:
pass
# do something here
blank.status = 'active'
blank.save()
newhostname = blank.hostname.split(".")[0]
newdomain = blank.hostname.split(".", 1)[1]
while True:
try:
info[0]
except:
sys.exit()
if info[0][0] == 'mac_addr':
pair = info.pop(0)
try:
Interface.get(mac_addr=pair[1])
print "Error: %s already exists" % inter.mac_addr
sys.exit()
except:
inter = Interface.objects.create(host=blank, mac_addr=pair[1], hdwr_type='eth')
inter.save()
elif info[0][0] == 'hdwr_type':
pair = info.pop(0)
inter.hdwr_type = pair[1]
inter.save()
elif info[0][0] == 'ip_addr':
pair = info.pop(0)
ip = IP.objects.create(interface=inter, ip_addr=pair[1], num=1)
hostnamenode = Name(ip=ip, name=blank.hostname, dns_view='global', only=False)
hostnamenode.save()
namenode = Name(ip=ip, name=".".join([newhostname + "-" + inter.hdwr_type,
newdomain]),
dns_view="global", only=False)
namenode.save()
subnetnode = Name(ip=ip, name=newhostname + "-" +
ip.ip_addr.split(".")[2] + "." +
newdomain, dns_view="global", only=False)
subnetnode.save()
hostnamenode.mxs.add(mx)
namenode.mxs.add(mx)
subnetnode.mxs.add(mx)
elif info[0][0] == 'cname':
pair = info.pop(0)
cname = CName.objects.create(name=hostnamenode, cname=pair[1])
cname.save()
else:
break
|
