ads2invenio.py

# Copyright (C) 2011, The SAO/NASA Astrophysics Data System
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

'''
Main module

It parses the parameters and calls the global manager
'''

from optparse import OptionParser
from time import strftime
import sys
if sys.version_info < (2, 6):
    raise "Must use python 2.6 or greater"

import pipeline_manager
import pipeline_settings

import logging

def parse_parameters():
    """Function that parse the parameters passed to the script"""
    parser = OptionParser()

    parser.add_option("-m", "--mode", dest="mode", help="Specify the method of extraction (\"full\" or \"update\") ", metavar="MODE_VALUE")
    parser.add_option("-l", "--logtype", dest="logtype", help="Specify the type of logging you want (\"file\" or \"screen\"). If not specified \"file\" will be the default.", metavar="LOG_TIPE_VALUE")
    parser.add_option("-v", "--verbose", action="store_true", dest="verbose", help='Use this parameter if a verbose execution is needed ')
    parser.add_option("-u", "--uploadmode", dest="uploadmode", help="Specify the method of upload (\"concurrent\" or \"bibupload\") ", metavar="UPLOADMODE_VALUE")

    # catch the parameters from the command line
    options, _ = parser.parse_args()

    #Dictionary to return parameters
    parameters = {}

    if options.mode:
        if options.mode == 'full' or options.mode == 'update':
            parameters['mode'] = options.mode
        else:
            parser.print_help()
            return None
    else:
        parser.print_help()
        return None
    
    if options.logtype:
        if options.logtype == 'file' or options.logtype == 'screen':
            parameters['logtype'] = options.logtype
        else:
            parser.print_help()
            return None
    else:
        parameters['logtype'] = 'file'
        
    if options.uploadmode:
        if options.uploadmode in ('concurrent', 'bibupload',):
            parameters['uploadmode'] = options.uploadmode
        else:
            parser.print_help()
            return None
    else:
        parameters['uploadmode'] = 'concurrent'
    
    if options.verbose:
        parameters['verbose'] = True
    else:
        parameters['verbose'] = False

    return parameters

def main():
    """ Main Function"""
    #Manage parameters
    parameters = parse_parameters()
    if parameters == None:
        return
    #set the global logging
    if parameters['logtype'] == 'file':
        logging.basicConfig(
            filename=pipeline_settings.GLOBAL_PIPELINE_LOGGING_PATH +'/pipeline_'+strftime("%Y_%m_%d")+'.log',
            format=pipeline_settings.LOGGING_FORMAT)
    else:
        logging.basicConfig(format=pipeline_settings.LOGGING_FORMAT)
    logger = logging.getLogger(pipeline_settings.LOGGING_GLOBAL_NAME)
    logger.setLevel(logging.INFO)
    logger.warning('Pipeline started')
    #then I check the verbose level
    if not parameters['verbose']:
        logger.setLevel(logging.WARNING)
    #I call the global manager
    pipeline_manager.manage(parameters['mode'], parameters['uploadmode'])

if __name__ == "__main__":
    main()