plc-config

   1 #!/usr/bin/python
   2 #
   3 # Script for basic access to the PlanetLab Central (PLC) configuration
   4 # file store.
   5 #
   6 # Mark Huang <mlhuang@cs.princeton.edu>
   7 # Copyright (C) 2006 The Trustees of Princeton University
   8 #
   9 # $Id$
  10 #
  11
  12 import sys
  13 import os
  14 import fcntl
  15 import getopt
  16 import signal
  17 from plc_config import PLCConfiguration
  18
  19
  20 def usage():
  21     print """
  22 Script to access the PLC configuration file store.
  23
  24 Usage: %s [OPTION]... [FILES]
  25         Conversion:
  26
  27         --shell         Output shell configuration file
  28         --python        Output Python configuration file
  29         --php           Output PHP configuration file
  30
  31         Information:
  32
  33         --variables     List names of all variables
  34         --packages      List names of all packages
  35         --comps         List comps.xml from configuration
  36
  37         Basic variable value manipulation:
  38
  39         --category=     Category identifier
  40         --variable=     Variable identifier
  41         --value=        Variable value
  42
  43         Basic package list manipulation:
  44
  45         --group=        Package group identifier
  46         --package=      Package name
  47         --type=         Package type
  48
  49         Miscellaneous:
  50
  51         -h, --help      This message
  52         -s, --save      Save changes to first configuration file
  53 """.lstrip() % sys.argv[0]
  54     sys.exit(1)
  55
  56
  57 def deprecated (message):
  58     print "%s: deprecated usage"%sys.argv[0]
  59     print message
  60     sys.exit(1)
  61
  62 def main():
  63     plc = PLCConfiguration()
  64     fileobjs = []
  65     output = None
  66     category = {}
  67     variable = {}
  68     group = {}
  69     package = {}
  70     save = False
  71
  72     # Standard options
  73     shortopts = "hs:"
  74     longopts = ["shell", "bash", "python",
  75                 "php",
  76                 "xml",
  77                 "variables",
  78                 "packages",
  79                 "groups",
  80                 "comps",
  81                 "category=", "variable=", "value=",
  82                 "group=", "package=", "type=",
  83                 "help",
  84                 "save="]
  85
  86     try:
  87         (opts, argv) = getopt.gnu_getopt(sys.argv[1:], shortopts, longopts)
  88     except Exception, err:
  89         sys.stderr.write("Error: " + str(err) + os.linesep)
  90         sys.exit(1)
  91
  92     for (opt, optval) in opts:
  93         if opt == "--shell" or \
  94              opt == "--bash" or \
  95              opt == "--python":
  96             output = plc.output_shell
  97         elif opt == "--php":
  98             output = plc.output_php
  99         elif opt == "--xml":
 100             output = plc.output_xml
 101         elif opt == "--variables":
 102             output = plc.output_variables
 103         elif opt == "--packages":
 104 #            output = plc.output_packages
 105             deprecated("option --packages deprecated -- use .lst files instead")
 106         elif opt == "--groups":
 107 #            output = plc.output_groups
 108             deprecated("option --groups deprecated -- use .lst files instead")
 109         elif opt == "--comps":
 110 #            output = plc.output_comps
 111             deprecated("option --comps deprecated -- use .lst files instead")
 112         elif opt == "--category":
 113             category['id'] = optval
 114         elif opt == "--variable":
 115             variable['id'] = optval
 116         elif opt == "--value":
 117             variable['value'] = optval
 118         elif opt == "--group":
 119 #            group['id'] = optval
 120             deprecated("option --group deprecated -- use .lst files instead")
 121         elif opt == "--package":
 122 #            package['name'] = optval
 123             deprecated("option --package deprecated -- use .lst files instead")
 124         elif opt == "--type":
 125             package['type'] = optval
 126         elif opt == '-s' or opt == "--save":
 127             if not optval:
 128                 usage()
 129             print 'parsed save option',optval
 130             save = optval
 131         elif opt == '-h' or opt == "--help":
 132             usage()
 133
 134     # Try the default
 135     if not argv:
 136         argv = ["/etc/planetlab/plc_config.xml"]
 137
 138     # Merge all files
 139     for file in argv:
 140         try:
 141             plc.load(file)
 142         except IOError:
 143             pass
 144         except Exception, err:
 145             sys.stderr.write("Error: %s: %s" % (file, str(err)) + os.linesep)
 146             sys.exit(1)
 147
 148     # --category, --variable, --value
 149     if category.has_key('id') and variable.has_key('id'):
 150         if variable.has_key('value'):
 151             plc.set(category, variable)
 152         else:
 153             (category, variable) = plc.get(category['id'], variable['id'])
 154             if variable.has_key('value'):
 155                 print variable['value']
 156
 157     # --shell, --php, --xml
 158     if output is not None:
 159         sys.stdout.write(output())
 160
 161     # --save
 162     if save:
 163         # create directory if needed
 164         # so that plc.d/{api,postgres} can create configs/site.xml
 165         dirname = os.path.dirname (save)
 166         if (not os.path.exists (dirname)):
 167             os.makedirs(dirname,0755)
 168             if (not os.path.exists (dirname)):
 169                 print "Cannot create dir %s - exiting" % dirname
 170                 sys.exit(1)
 171
 172         plc.save(save)
 173
 174
 175 if __name__ == '__main__':
 176     main()