adding functionality to hostbase command line tools

git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@2432 ce84e21b-d406-0410-9b95-82705330c041

1 files changed, 65 insertions, 0 deletions

diff --git a/tools/hostbase.py b/tools/hostbase.py
new file mode 100755
index 000000000..c25be3370
--- /dev/null
+++ b/tools/hostbase.py
@@ -0,0 +1,65 @@
+#!/usr/bin/python
+import sys, os
+os.environ['DJANGO_SETTINGS_MODULE'] = 'Hostbase.settings'
+from Hostbase.hostbase.models import Host, Interface
+from getopt import getopt, GetoptError
+from re import split
+
+attribs = ['hostname', 'whatami', 'netgroup', 'security_class', 'support',
+           'csi', 'printq', 'dhcp', 'outbound_smtp', 'primary_user',
+           'administrator', 'location', 'expiration_date', 'comments',
+           'status', 'last']
+
+already_exists = None
+#here's my attempt at making the command line idiot proof
+#you must supply and arugument and hostname for hostbase.py to run
+try:
+    (opts, args) = getopt(sys.argv[1:],'l:c:')
+    sys.argv[1]
+    if len(split("\.", opts[0][1])) == 1:
+        hosttouse = opts[0][1] + ".mcs.anl.gov"
+    else:
+        hosttouse = opts[0][1]
+except (GetoptError, IndexError):
+    print "\nUsage: hostbase.py -flag (hostname)\n"
+    print "Flags:"
+    print "\t-l   look (hostname)\n"
+#    print "\t-c   copy (hostname)\n"
+    sys.exit()
+
+try:
+    host = Host.objects.get(hostname=hosttouse)
+except:
+    print "Error: host %s not in hostbase" % hosttouse
+    sys.exit(1)
+interfaces = []
+for interface in host.interface_set.all():
+    interfaces.append([interface, interface.ip_set.all()])
+hostinfo = "\n"
+for attrib in attribs:
+    if not (opts[0][0] == '-c' and attrib in ['status', 'last']):
+        if attrib == 'dhcp' or attrib == 'outbound_smtp':
+            if host.__dict__[attrib]:
+                hostinfo += "%-32s-> %s\n" % (attrib, 'y')
+            else:
+                hostinfo += "%-32s-> %s\n" % (attrib, 'n')
+        else:
+            hostinfo += "%-32s-> %s\n" % (attrib, host.__dict__[attrib])
+for interface in interfaces:
+    hostinfo += "\n%-32s-> %s\n" % ('mac_addr', interface[0].mac_addr)
+    hostinfo += "%-32s-> %s\n" % ('hdwr_type', interface[0].hdwr_type)
+    for ip in interface[1]:
+        hostinfo += "%-32s-> %s\n" % ('ip_addr', ip.ip_addr)
+
+if opts[0][0] == '-l':
+    """Displays general host information"""
+    print hostinfo
+
+if opts[0][0] == '-c':
+    """Provides pre-filled template to copy a host record"""
+    fd = open('/tmp/hostbase.%s.tmp' % host.id, 'w')
+    fd.write(hostinfo)
+    fd.close()
+    os.system('vi + /tmp/hostbase.%s.tmp' % host.id)
+    os.system('batchadd.py /tmp/hostbase.%s.tmp' % host.id)
+