Changeset 2409


Ignore:
Timestamp:
Feb 28, 2010 7:42:33 AM (9 years ago)
Author:
wehart
Message:

Rework of profiling imports.

Refinement of Pyomo command-line parsing, which is more
specific now.

Adding automatic import of pyomo to package.

Location:
coopr.pyomo/trunk/coopr/pyomo/scripting
Files:
3 edited

Legend:

Unmodified
Added
Removed
  • coopr.pyomo/trunk/coopr/pyomo/scripting/__init__.py

    r2264 r2409  
    99
    1010from convert import *
     11import pyomo
     12import util
     13
  • coopr.pyomo/trunk/coopr/pyomo/scripting/pyomo.py

    r2283 r2409  
    1313from coopr.pyomo import pyomo
    1414from coopr.opt.base import SolverFactory
     15from coopr.opt.parallel import SolverManagerFactory
    1516from pyutilib.misc import Options, Container
    1617
     
    4849        action="store",
    4950        dest="solver",
    50         type="string",
     51        type="choice",
     52        choices=filter(lambda x: x[0] != '_', SolverFactory()),
    5153        default="glpk")
    5254    parser.add_option("--path",
     
    6567            action="append",
    6668            dest="debug",
    67             choices=debug_choices)
     69            type='choice',
     70            choices=debug_choices,
     71            default=[]
     72            )
    6873    parser.add_option("-k","--keepfiles",
    6974            help="Keep temporary files",
     
    7580            action="store",
    7681            dest="tempdir",
     82            type="string",
    7783            default=None)
    7884    parser.add_option("-q","--quiet",
     
    9096            action="store",
    9197            dest="logfile",
     98            type="string",
    9299            default=None)
    93100    parser.add_option("-s","--summary",
     
    105112            action="store",
    106113            dest="profile",
     114            type="int",
    107115            default=0)
    108116    parser.add_option("--timelimit",
     
    166174            action="store",
    167175            dest="smanager_type",
     176            type="choice",
     177            choices=SolverManagerFactory(),
    168178            default="serial")
    169179    parser.add_option("--stream-output",
     
    176186            action="store",
    177187            dest="save_model",
     188            type="string",
    178189            default=None)
    179190    #
  • coopr.pyomo/trunk/coopr/pyomo/scripting/util.py

    r2399 r2409  
    1313import textwrap
    1414import traceback
    15 import cProfile
     15try:
     16    import cProfile as profile
     17except ImportError:
     18    import profile
    1619try:
    1720    import pstats
     
    326329            raise ValueError, "Cannot use the 'profile' option.  The Python 'pstats' package cannot be imported!"
    327330        tfile = pyutilib.services.TempfileManager.create_tempfile(suffix=".profile")
    328         tmp = cProfile.runctx(command.__name__+'(options=options,args=nargs,parser=parser)',globals(),locals(),tfile)
     331        tmp = profile.runctx(command.__name__+'(options=options,args=nargs,parser=parser)',globals(),locals(),tfile)
    329332        p = pstats.Stats(tfile).strip_dirs()
    330333        p.sort_stats('time', 'cum')
Note: See TracChangeset for help on using the changeset viewer.