lrms-ur-generator

#!/usr/bin/env python
#
# LRMS UR Generator executable.
#
# Module for the LRMS UR Generator module.
#
# Author: Henrik Thostrup Jensen <htj@ndgf.org>
# Copyright: Nordic Data Grid Facility (2009)

import os
import sys
import logging

from lrmsurgen import config


LOG_FORMAT         = "%(asctime)s [%(levelname)s] %(message)s"



def main():

    # start with command line parsing and various setups
    parser = config.getParser()
    options, args = parser.parse_args()
    if not os.path.exists(options.config):
        # we don't have logging yet
        print 'Configuration file %s does not exist' % options.config
        sys.exit(1)

    cfg = config.getConfig(options.config)

    logfile = options.logfile
    if logfile is None:
        logfile = config.getConfigValue(cfg, config.SECTION_COMMON, config.LOGFILE, config.DEFAULT_LOG_FILE)

    logging.basicConfig(filename=logfile, format=LOG_FORMAT, level=logging.DEBUG)

    hostname = config.getConfigValue(cfg, config.SECTION_COMMON, config.HOSTNAME)
    if hostname is None:
        import socket
        hostname = socket.getfqdn()

    user_map_file = config.getConfigValue(cfg, config.SECTION_COMMON, config.USERMAP, config.DEFAULT_USERMAP_FILE)
    project_map_file = config.getConfigValue(cfg, config.SECTION_COMMON, config.PROJECTMAP, config.DEFAULT_PROJECTMAP_FILE)

    # basic configuration done, read usermap+projectmap and start lrms ur module

    try:
        user_map = config.getUserMap(user_map_file)
    except IOError:
        logging.error('IOError while attempting to read user map at %s (missing file?)' % user_map_file)
        user_map = {}

    try:
        project_map = config.getProjectMap(project_map_file)
    except IOError:
        logging.error('IOError while attempting to read project map at %s (missing file?)' % user_map_file)
        project_map = {}

    sections = cfg.sections()
    if config.SECTION_MAUI in sections:
        from lrmsurgen import maui as lrms
    elif config.SECTION_TORQUE in sections:
        from lrmsurgen import torque as lrms
    # more LRMSes to come...

    try:
        lrms.generateUsageRecords(cfg, hostname, user_map, project_map)
    except Exception, e:
        logging.error('Got exception while generating usage records:')
        logging.exception(e)
        sys.exit(3)


if __name__ == '__main__':
    main()