geni/util/config.py

   1 ##
   2 # Geniwrapper Configuration Info
   3 #
   4 # This module holds configuration parameters for geniwrapper. There are two
   5 # main pieces of information that are used: the database connection and
   6 # the PLCAPI connection
   7 ##
   8
   9 ##
  10 # Geniwrapper uses a MYSQL database to store records. This database may be
  11 # co-located with the PLC database, or it may be a separate database. The
  12 # following parameters define the connection to the database.
  13 #
  14 # Note that Geniwrapper does not access any of the PLC databases directly via
  15 # a mysql connection; All PLC databases are accessed via PLCAPI.
  16
  17 import os
  18 import sys
  19
  20 # If we have been checked out into a directory at the same
  21 # level as myplc, where plc_config.py lives. If we are in a
  22 # MyPLC environment, plc_config.py has already been installed
  23 # in site-packages.
  24 myplc = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) + \
  25         os.sep + "myplc"
  26 geni =  os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))+ os.sep + "geni"
  27
  28 class Config:
  29     """
  30     Parse the bash/Python/PHP version of the configuration file. Very
  31     fast but no type conversions.
  32     """
  33
  34     def __init__(self, file = "/etc/geni/geni_config"):
  35         # Load plc_config
  36
  37         loaded = False
  38         path = os.path.dirname(os.path.abspath(__file__))
  39         filename = file.split(os.sep)[-1]
  40         alt_file = path + os.sep + 'util' + os.sep + filename
  41         geni_file = geni + os.sep + 'util' + os.sep + filename
  42         files = [file, alt_file, myplc_file]
  43
  44         for file in files:
  45             try:
  46                 execfile(file, self.__dict__)
  47                 loaded = True
  48             except:
  49                 pass
  50
  51         if not loaded:
  52             raise Exception, "Could not find config in " + ", ".join(files)
  53
  54         # set up some useful variables
  55
  56     def load(self, file):
  57         try:
  58             execfile(file, self.__dict__)
  59         except:
  60             raise Exception, "Could not find config in " + file
  61
  62 plcConfig = Config("/etc/planetlab/plc_config")
  63
  64 def get_default_dbinfo():
  65     dbinfo={ 'dbname' : plcConfig.PLC_DB_NAME,
  66     'address' : plcConfig.PLC_DB_HOST,
  67     'port' : plcConfig.PLC_DB_PORT,
  68     'user' : plcConfig.PLC_DB_USER,
  69     'password' : plcConfig.PLC_DB_PASSWORD
  70         }
  71
  72     return dbinfo
  73
  74 ##
  75 # Geniwrapper uses a PLCAPI connection to perform operations on the registry,
  76 # such as creating and deleting slices. This connection requires an account
  77 # on the PLC server with full administrator access.
  78 #
  79 # The Url parameter controls whether the connection uses PLCAPI directly (i.e.
  80 # Geniwrapper is located on the same machine as PLC), or uses a XMLRPC connection
  81 # to the PLC machine. If you wish to use the API directly, then remove the Url
  82 # field from the dictionary.
  83
  84 def get_pl_auth():
  85     pl_auth = {'Username': plcConfig.PLC_API_MAINTENANCE_USER,
  86     'AuthMethod': 'capability',
  87     'AuthString':  plcConfig.PLC_API_MAINTENANCE_PASSWORD,
  88     "Url": 'https://%s:%s%s' %(plcConfig.PLC_API_HOST, plcConfig.PLC_API_PORT, plcConfig.PLC_API_PATH)
  89     }
  90
  91     return pl_auth