Changeset 53

Timestamp:

09/19/05 15:55:56 (18 years ago)

Author:

bas

Message:

pxeconfig:

Added --debug and --remover option

File:

• trunk/pxeconfig/pxeconfig (modified) (10 diffs)

trunk/pxeconfig/pxeconfig

@@ -31 +31 @@
 
  [-d|--directory <pxe_config_dir>]
+ [-f|--file <filename>] [hostname(s)]
+ [-n|--net <C-net> -s|--start <number> -e|--end <number> -f|--file <filename>]
+ [-r|--remove]
  [-V|--version]
- [-n|--net <C-net> -s|--start <number> -e|--end <number> -f|--file <filename>]
- [-f|--file <filename>] [hostname(s)]
 
 With this program you can configure which PXE configuration file a node
@@ -75 +76 @@
 START='start'
 END='end'
-VERSION='0.5'
-
-SHORTOPT_LIST='hVd:n:s:e:f:'
-LONGOPT_LIST=['help', 'version', 'directory=', 'net=', 'start=', 'end=', 'file=']
+VERSION='0.6'
+REMOVE='remove'
+
+SHORTOPT_LIST='hVd:n:s:e:f:r'
+LONGOPT_LIST=['help', 'version', 'directory=', 'net=', 'start=', 'end=', 'file=', 'remove', 'debug']
 
 def select_pxe_configfile():
@@ -125 +127 @@
   return files[index-1]
 
-def create_links(dict):
+def manage_links(dict):
   """
   Create the links in the PXE_CONF_DIR,
@@ -136 +138 @@
   for i in range(dict[START], dict[END]+1):
     haddr = '%s%02X' %(naddr, i)
-    if DEBUG:
-      print 'linking %s to %s' %(haddr, pxe_filename)
-
-    if os.path.exists(haddr):
-       os.unlink(haddr)
-
-    os.symlink(pxe_filename, haddr)
-
+
+    if dict[REMOVE]:
+       if DEBUG:
+          print 'removing %s/%s' %(PXE_CONF_DIR, haddr)
+       if os.path.exists(haddr):
+          os.unlink(haddr)
+    else:
+       if DEBUG:
+          print 'linking %s to %s' %(haddr, pxe_filename)
+       if os.path.exists(haddr):
+          os.unlink(haddr)
+       os.symlink(pxe_filename, haddr)
 
 def convert_network(net, prefix=''):
@@ -243 +249 @@
   """
   global PXE_CONF_DIR
+  global DEBUG
 
   try:
@@ -277 +284 @@
       print VERSION
       sys.exit(0)
+
+    elif opt in ['-r', '--remove']:
+      binfo[REMOVE] = 1
+
+    elif opt in ['--debug']:
+      DEBUG = 1
 
     elif opt in ['-h', '--help']:
@@ -299 +312 @@
                 binfo[START] = int(net[3])
                 binfo[END] =  int(net[3])
-                create_links(binfo)
+                manage_links(binfo)
 
 
@@ -306 +319 @@
   #
   bootinfo = {}
+  bootinfo[REMOVE] = 0
 
   hostnames = check_args(sys.argv, bootinfo)
+
+  if bootinfo[REMOVE]:
+     bootinfo[FILENAME] = 'remove_boot_file'
 
 
@@ -315 +332 @@
   if hostnames:
         if not bootinfo.has_key(FILENAME):
-                bootinfo[FILENAME] = select_pxe_configfile()
+              bootinfo[FILENAME] = select_pxe_configfile()
         hosts_links(hostnames,bootinfo)
   else:
@@ -322 +339 @@
         else:
                 interactive(bootinfo)
-        create_links(bootinfo)
+        manage_links(bootinfo)
 
 if __name__ == '__main__':