70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
"""
|
|
PasteScript shell command for Pecan.
|
|
"""
|
|
from webtest import TestApp
|
|
|
|
from base import Command
|
|
|
|
import sys
|
|
|
|
|
|
class ShellCommand(Command):
|
|
"""
|
|
Open an interactive shell with the Pecan app loaded.
|
|
"""
|
|
|
|
# command information
|
|
usage = 'CONFIG_NAME'
|
|
summary = __doc__.strip().splitlines()[0].rstrip('.')
|
|
|
|
# command options/arguments
|
|
min_args = 1
|
|
max_args = 1
|
|
|
|
def command(self):
|
|
|
|
# load the application
|
|
config = self.load_configuration(self.args[0])
|
|
setattr(config.app, 'reload', False)
|
|
app = self.load_app(config)
|
|
|
|
# prepare the locals
|
|
locs = dict(__name__='pecan-admin')
|
|
locs['wsgiapp'] = app
|
|
locs['app'] = TestApp(app)
|
|
|
|
# find the model for the app
|
|
model = self.load_model(config)
|
|
if model:
|
|
locs['model'] = model
|
|
|
|
# insert the pecan locals
|
|
exec('from pecan import abort, conf, redirect, request, response') in locs
|
|
|
|
# prepare the banner
|
|
banner = ' The following objects are available:\n'
|
|
banner += ' %-10s - This project\'s WSGI App instance\n' % 'wsgiapp'
|
|
banner += ' %-10s - The current configuration\n' % 'conf'
|
|
banner += ' %-10s - webtest.TestApp wrapped around wsgiapp\n' % 'app'
|
|
if model:
|
|
model_name = getattr(model, '__module__', getattr(model, '__name__', 'model'))
|
|
banner += ' %-10s - Models from %s\n' % ('model', model_name)
|
|
|
|
# launch the shell, using IPython if available
|
|
try:
|
|
from IPython.Shell import IPShellEmbed
|
|
shell = IPShellEmbed(argv=self.args)
|
|
shell.set_banner(shell.IP.BANNER + '\n\n' + banner)
|
|
shell(local_ns=locs, global_ns={})
|
|
except ImportError:
|
|
import code
|
|
py_prefix = sys.platform.startswith('java') and 'J' or 'P'
|
|
shell_banner = 'Pecan Interactive Shell\n%sython %s\n\n' % \
|
|
(py_prefix, sys.version)
|
|
shell = code.InteractiveConsole(locals=locs)
|
|
try:
|
|
import readline
|
|
except ImportError:
|
|
pass
|
|
shell.interact(shell_banner + banner)
|