source: coopr.pyomo/trunk/coopr/pyomo/scripting/convert.py @ 3681

#  _________________________________________________________________________
#
#  Coopr: A COmmon Optimization Python Repository
#  Copyright (c) 2008 Sandia Corporation.
#  This software is distributed under the BSD License.
#  Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
#  the U.S. Government retains certain rights in this software.
#  For more information, see the Coopr README.txt file.
#  _________________________________________________________________________

__all__ = ['pyomo2lp', 'pyomo2nl']


import argparse
from coopr.opt import ProblemFormat
from coopr.opt.base import SolverFactory
from pyutilib.misc import Options, Container

import util
import pyomo

def create_parser(cmd):
    #
    #
    # Setup command-line options
    #
    #
    parser = argparse.ArgumentParser(
                usage = '%s [options] <model.py> [<model.dat>]' % cmd
                )
    pyomo.add_model_group(parser)
    pyomo.add_logging_group(parser)
    pyomo.add_misc_group(parser)
    parser.add_argument('model_file', action='store', nargs='?', default='', help='A Python      module that defines a Pyomo model')
    parser.add_argument('data_files', action='store', nargs='*', default=[], help='Pyomo data    files that defined data used to create a model instance')
    return parser

format = None

def convert(options=Options(), parser=None):
    global format
    if options.save_model is None:
        if format == ProblemFormat.cpxlp:
            options.save_model = 'unknown.lp'
        else:
            options.save_model = 'unknown.'+str(format)
    options.format = format
    #
    if options.help_components:
        util.print_components(options)
        return Container()
    #
    if not util.setup_environment(options):
        return Container()
    #
    if not util.apply_preprocessing(options, parser):
        return Container()
    #
    model_data = util.create_model(options)
    #
    util.finalize(options, model=model_data.model)
    #
    return Container(model=model_data.model)

def pyomo2lp(args=None):
    global format
    parser = create_parser('pyomo2lp')
    format = ProblemFormat.cpxlp
    return util.run_command(convert, parser, args=args, name='pyomo2lp')

def pyomo2nl(args=None):
    global format
    parser = create_parser('pyomo2nl')
    format = ProblemFormat.nl
    return util.run_command(convert, parser, args=args, name='pyomo2nl')