mirror of
https://github.com/galaxyproject/galaxy.git
synced 2026-09-24 16:30:27 +08:00
Nate Coraor
806438edc9
galaxy.util.script so that get_uwsgi_args.py still works, but that will be PR'd separately and merged later.
120 lines
4.2 KiB
Python
120 lines
4.2 KiB
Python
from __future__ import print_function
|
|
|
|
import os
|
|
import sys
|
|
|
|
from six import string_types
|
|
from six.moves import shlex_quote
|
|
|
|
sys.path.insert(1, os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, 'lib')))
|
|
|
|
from galaxy.util.path import get_ext
|
|
from galaxy.util.properties import load_app_properties, nice_config_parser
|
|
from galaxy.util.script import main_factory
|
|
|
|
|
|
DESCRIPTION = "Script to determine uWSGI command line arguments"
|
|
# socket is not an alias for http, but it is assumed that if you configure a socket in your uwsgi config you do not
|
|
# want to run the default http server (or you can configure it yourself)
|
|
ALIASES = {
|
|
'virtualenv': ('home', 'venv', 'pyhome'),
|
|
'pythonpath': ('python-path', 'pp'),
|
|
'http': ('httprouter', 'socket', 'uwsgi-socket', 'suwsgi-socket', 'ssl-socket'),
|
|
}
|
|
DEFAULT_ARGS = {
|
|
'_all_': ('virtualenv', 'pythonpath', 'master', 'threads', 'http', 'static-map', 'die-on-term', 'enable-threads'),
|
|
'galaxy': ('py-call-osafterfork', 'mule', 'farm'),
|
|
'reports': (),
|
|
'tool_shed': (),
|
|
}
|
|
DEFAULT_PORTS = {
|
|
'galaxy': 8080,
|
|
'reports': 9001,
|
|
'tool_shed': 9009,
|
|
}
|
|
|
|
|
|
def __arg_set(arg, kwargs):
|
|
if arg in kwargs:
|
|
return True
|
|
for alias in ALIASES.get(arg, ()):
|
|
if alias in kwargs:
|
|
return True
|
|
return False
|
|
|
|
|
|
def __add_arg(args, arg, value):
|
|
optarg = '--%s' % arg
|
|
if isinstance(value, bool):
|
|
if value is True:
|
|
args.append(optarg)
|
|
elif isinstance(value, string_types):
|
|
# the = in --optarg=value is usually, but not always, optional
|
|
if value.startswith('='):
|
|
args.append(shlex_quote(optarg + value))
|
|
else:
|
|
args.append(optarg)
|
|
args.append(shlex_quote(value))
|
|
else:
|
|
[__add_arg(args, arg, v) for v in value]
|
|
|
|
|
|
def __add_config_file_arg(args, config_file, app):
|
|
ext = None
|
|
if config_file:
|
|
ext = get_ext(config_file)
|
|
if ext in ('yaml', 'json'):
|
|
__add_arg(args, ext, config_file)
|
|
elif ext == 'ini':
|
|
config = nice_config_parser(config_file)
|
|
has_logging = config.has_section('loggers')
|
|
if config.has_section('app:main'):
|
|
# uWSGI does not have any way to set the app name when loading with paste.deploy:loadapp(), so hardcoding
|
|
# the name to `main` is fine
|
|
__add_arg(args, 'ini-paste' if not has_logging else 'ini-paste-logged', config_file)
|
|
return # do not add --module
|
|
else:
|
|
__add_arg(args, ext, config_file)
|
|
if has_logging:
|
|
__add_arg(args, 'paste-logger', True)
|
|
__add_arg(args, 'module', 'galaxy.webapps.{app}.buildapp:uwsgi_app()'.format(app=app))
|
|
|
|
|
|
def _get_uwsgi_args(cliargs, kwargs):
|
|
# it'd be nice if we didn't have to reparse here but we need things out of more than one section
|
|
config_file = cliargs.config_file or kwargs.get('__file__')
|
|
uwsgi_kwargs = load_app_properties(config_file=config_file, config_section='uwsgi')
|
|
handlerct = int(kwargs.get('job_handler_count', 1))
|
|
args = []
|
|
__add_config_file_arg(args, config_file, cliargs.app)
|
|
defaults = {
|
|
'virtualenv': os.environ.get('VIRTUAL_ENV', './.venv'),
|
|
'pythonpath': 'lib',
|
|
'master': True,
|
|
'threads': '4',
|
|
'http': 'localhost:{port}'.format(port=DEFAULT_PORTS[cliargs.app]),
|
|
'static-map': ('/static/style={here}/static/style/blue'.format(here=os.getcwd()),
|
|
'/static={here}/static'.format(here=os.getcwd())),
|
|
'die-on-term': True,
|
|
'enable-threads': True,
|
|
'py-call-osafterfork': True,
|
|
'mule': ('=lib/galaxy/main.py',) * handlerct,
|
|
'farm': '={name}:{mules}'.format(
|
|
name=kwargs.get('job_handler_pool_name', 'job-handlers'),
|
|
mules=','.join([str(x) for x in range(1, handlerct + 1)])) if handlerct > 0 else '',
|
|
}
|
|
for arg in DEFAULT_ARGS['_all_'] + DEFAULT_ARGS[cliargs.app]:
|
|
if not __arg_set(arg, uwsgi_kwargs):
|
|
__add_arg(args, arg, defaults[arg])
|
|
print(' '.join(args))
|
|
|
|
|
|
ACTIONS = {
|
|
"get_uwsgi_args": _get_uwsgi_args,
|
|
}
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main = main_factory(description=DESCRIPTION, actions=ACTIONS, default_action="get_uwsgi_args")
|
|
main()
|