138 lines
5.2 KiB
Python
138 lines
5.2 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
import argparse
|
|
import os
|
|
import logging
|
|
import logging.config
|
|
import logging.handlers
|
|
from contextlib import contextmanager
|
|
|
|
from trytond import __version__
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_base_parser():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--version', action='version',
|
|
version='%(prog)s ' + __version__)
|
|
parser.add_argument("-c", "--config", dest="configfile", metavar='FILE',
|
|
nargs='+', default=[os.environ.get('TRYTOND_CONFIG')],
|
|
help='Specify configuration files')
|
|
return parser
|
|
|
|
|
|
def get_parser():
|
|
parser = get_base_parser()
|
|
|
|
parser.add_argument("-v", "--verbose", action='count',
|
|
dest="verbose", default=0, help="enable verbose mode")
|
|
parser.add_argument('--dev', dest='dev', action='store_true',
|
|
help='enable development mode')
|
|
|
|
parser.add_argument("-d", "--database", dest="database_names", nargs='+',
|
|
default=[], metavar='DATABASE', help="specify the database name")
|
|
parser.add_argument("--logconf", dest="logconf", metavar='FILE',
|
|
help="logging configuration file (ConfigParser format)")
|
|
|
|
return parser
|
|
|
|
|
|
def get_parser_daemon():
|
|
parser = get_parser()
|
|
parser.add_argument("--pidfile", dest="pidfile", metavar='FILE',
|
|
help="file where the server pid will be stored")
|
|
parser.add_argument("--coroutine", action="store_true", dest="coroutine",
|
|
help="use coroutine for concurrency")
|
|
return parser
|
|
|
|
|
|
def get_parser_worker():
|
|
parser = get_parser_daemon()
|
|
parser.add_argument("--name", dest='name',
|
|
help="work only on the named queue")
|
|
parser.add_argument("-n", dest='processes', type=int,
|
|
help="number of processes to use")
|
|
parser.add_argument("--max", dest='maxtasksperchild', type=int,
|
|
help="number of tasks a worker process before being replaced")
|
|
parser.add_argument("-t", "--timeout", dest='timeout', default=60,
|
|
type=int, help="maximum timeout when waiting notification")
|
|
return parser
|
|
|
|
|
|
def get_parser_cron():
|
|
parser = get_parser_daemon()
|
|
parser.add_argument("-1", "--once", dest='once', action='store_true',
|
|
help="run pending tasks and halt")
|
|
return parser
|
|
|
|
|
|
def get_parser_admin():
|
|
parser = get_parser()
|
|
|
|
parser.add_argument("-u", "--update", dest="update", nargs='+', default=[],
|
|
metavar='MODULE', help="activate or update a module")
|
|
parser.add_argument("--all", dest="update", action="append_const",
|
|
const="ir", help="update all activated modules")
|
|
parser.add_argument("--activate-dependencies", dest="activatedeps",
|
|
action="store_true",
|
|
help="Activate missing dependencies of updated modules")
|
|
parser.add_argument("--email", dest="email", help="set the admin email")
|
|
parser.add_argument("-p", "--password", dest="password",
|
|
action='store_true', help="set the admin password")
|
|
parser.add_argument("--reset-password", dest='reset_password',
|
|
action='store_true', help="reset the admin password")
|
|
parser.add_argument("--test-email", dest='test_email',
|
|
help="Send a test email to the specified address.")
|
|
parser.add_argument("-m", "--update-modules-list", action="store_true",
|
|
dest="update_modules_list", help="Update list of tryton modules")
|
|
parser.add_argument("-l", "--language", dest="languages", nargs='+',
|
|
default=[], metavar='CODE', help="Load language translations")
|
|
parser.add_argument("--hostname", dest="hostname", default=None,
|
|
help="Limit database listing to the hostname")
|
|
|
|
parser.epilog = ('The first time a database is initialized '
|
|
'or when the password is set, the admin password is read '
|
|
'from file defined by TRYTONPASSFILE environment variable '
|
|
'or interactively asked from the user.\n'
|
|
'The config file can be specified in the TRYTOND_CONFIG '
|
|
'environment variable.\n'
|
|
'The database URI can be specified in the TRYTOND_DATABASE_URI '
|
|
'environment variable.')
|
|
return parser
|
|
|
|
|
|
def get_parser_console():
|
|
parser = get_base_parser()
|
|
parser.add_argument("-d", "--database", dest="database_name",
|
|
required=True, metavar='DATABASE', help="specify the database name")
|
|
parser.add_argument("--histsize", dest="histsize", type=int, default=500,
|
|
help="The number of commands to remember in the command history")
|
|
parser.epilog = "To store changes, `transaction.commit()` must be called."
|
|
return parser
|
|
|
|
|
|
def config_log(options):
|
|
if options.logconf:
|
|
logging.config.fileConfig(options.logconf)
|
|
logging.getLogger('server').info('using %s as logging '
|
|
'configuration file', options.logconf)
|
|
else:
|
|
logformat = ('%(process)s %(thread)s [%(asctime)s] '
|
|
'%(levelname)s %(name)s %(message)s')
|
|
level = max(logging.ERROR - options.verbose * 10, logging.NOTSET)
|
|
logging.basicConfig(level=level, format=logformat)
|
|
logging.captureWarnings(True)
|
|
|
|
|
|
@contextmanager
|
|
def pidfile(options):
|
|
path = options.pidfile
|
|
if not path:
|
|
yield
|
|
else:
|
|
with open(path, 'w') as fd:
|
|
fd.write('%d' % os.getpid())
|
|
yield
|
|
os.unlink(path)
|