From a9094cc3ce004c9517003df5d90c3e59e625a823 Mon Sep 17 00:00:00 2001 From: Christophe Simonis Date: Fri, 23 Nov 2012 11:51:20 +0100 Subject: [PATCH] [IMP] move main script into openerp dir bzr revid: chs@openerp.com-20121123105120-m1h2uiwk8baa8kin --- openerp-server | 272 +----------------------------------------------- openerp/main.py | 271 +++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 273 insertions(+), 270 deletions(-) create mode 100644 openerp/main.py diff --git a/openerp-server b/openerp-server index 5501eba2159..a2d48ee8e61 100755 --- a/openerp-server +++ b/openerp-server @@ -1,271 +1,3 @@ #!/usr/bin/env python -# -*- coding: utf-8 -*- -############################################################################## -# -# OpenERP, Open Source Management Solution -# Copyright (C) 2004-2009 Tiny SPRL (). -# -# This program is free software: you can redistribute it and/or modify -# it under the terms of the GNU Affero General Public License as -# published by the Free Software Foundation, either version 3 of the -# License, or (at your option) any later version. -# -# This program is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU Affero General Public License for more details. -# -# You should have received a copy of the GNU Affero General Public License -# along with this program. If not, see . -# -############################################################################## - -""" -OpenERP - Server -OpenERP is an ERP+CRM program for small and medium businesses. - -The whole source code is distributed under the terms of the -GNU Public Licence. - -(c) 2003-TODAY, Fabien Pinckaers - OpenERP SA -""" - -import logging -import os -import signal -import sys -import threading -import traceback -import time - -import openerp -__author__ = openerp.release.author -__version__ = openerp.release.version - -# Also use the `openerp` logger for the main script. -_logger = logging.getLogger('openerp') - -def check_root_user(): - """ Exit if the process's user is 'root' (on POSIX system).""" - if os.name == 'posix': - import pwd - if pwd.getpwuid(os.getuid())[0] == 'root' : - sys.stderr.write("Running as user 'root' is a security risk, aborting.\n") - sys.exit(1) - -def check_postgres_user(): - """ Exit if the configured database user is 'postgres'. - - This function assumes the configuration has been initialized. - """ - config = openerp.tools.config - if config['db_user'] == 'postgres': - sys.stderr.write("Using the database user 'postgres' is a security risk, aborting.") - sys.exit(1) - -def report_configuration(): - """ Log the server version and some configuration values. - - This function assumes the configuration has been initialized. - """ - config = openerp.tools.config - _logger.info("OpenERP version %s", __version__) - for name, value in [('addons paths', config['addons_path']), - ('database hostname', config['db_host'] or 'localhost'), - ('database port', config['db_port'] or '5432'), - ('database user', config['db_user'])]: - _logger.info("%s: %s", name, value) - -def setup_pid_file(): - """ Create a file with the process id written in it. - - This function assumes the configuration has been initialized. - """ - config = openerp.tools.config - if config['pidfile']: - fd = open(config['pidfile'], 'w') - pidtext = "%d" % (os.getpid()) - fd.write(pidtext) - fd.close() - -def preload_registry(dbname): - """ Preload a registry, and start the cron.""" - try: - db, registry = openerp.pooler.get_db_and_pool(dbname, update_module=openerp.tools.config['init'] or openerp.tools.config['update'], pooljobs=False) - - # jobs will start to be processed later, when openerp.cron.start_master_thread() is called by openerp.service.start_services() - registry.schedule_cron_jobs() - except Exception: - _logger.exception('Failed to initialize database `%s`.', dbname) - -def run_test_file(dbname, test_file): - """ Preload a registry, possibly run a test file, and start the cron.""" - try: - config = openerp.tools.config - db, registry = openerp.pooler.get_db_and_pool(dbname, update_module=config['init'] or config['update'], pooljobs=False) - cr = db.cursor() - _logger.info('loading test file %s', test_file) - openerp.tools.convert_yaml_import(cr, 'base', file(test_file), 'test', {}, 'test', True) - cr.rollback() - cr.close() - except Exception: - _logger.exception('Failed to initialize database `%s` and run test file `%s`.', dbname, test_file) - -def export_translation(): - config = openerp.tools.config - dbname = config['db_name'] - - if config["language"]: - msg = "language %s" % (config["language"],) - else: - msg = "new language" - _logger.info('writing translation file for %s to %s', msg, - config["translate_out"]) - - fileformat = os.path.splitext(config["translate_out"])[-1][1:].lower() - buf = file(config["translate_out"], "w") - cr = openerp.pooler.get_db(dbname).cursor() - openerp.tools.trans_export(config["language"], - config["translate_modules"] or ["all"], buf, fileformat, cr) - cr.close() - buf.close() - - _logger.info('translation file written successfully') - -def import_translation(): - config = openerp.tools.config - context = {'overwrite': config["overwrite_existing_translations"]} - dbname = config['db_name'] - - cr = openerp.pooler.get_db(dbname).cursor() - openerp.tools.trans_load( cr, config["translate_in"], config["language"], - context=context) - cr.commit() - cr.close() - -# Variable keeping track of the number of calls to the signal handler defined -# below. This variable is monitored by ``quit_on_signals()``. -quit_signals_received = 0 - -def signal_handler(sig, frame): - """ Signal handler: exit ungracefully on the second handled signal. - - :param sig: the signal number - :param frame: the interrupted stack frame or None - """ - global quit_signals_received - quit_signals_received += 1 - if quit_signals_received > 1: - # logging.shutdown was already called at this point. - sys.stderr.write("Forced shutdown.\n") - os._exit(0) - -def dumpstacks(sig, frame): - """ Signal handler: dump a stack trace for each existing thread.""" - # code from http://stackoverflow.com/questions/132058/getting-stack-trace-from-a-running-python-application#answer-2569696 - # modified for python 2.5 compatibility - threads_info = dict([(th.ident, {'name': th.name, - 'uid': getattr(th,'uid','n/a')}) - for th in threading.enumerate()]) - code = [] - for threadId, stack in sys._current_frames().items(): - thread_info = threads_info.get(threadId) - code.append("\n# Thread: %s (id:%s) (uid:%s)" % \ - (thread_info and thread_info['name'] or 'n/a', - threadId, - thread_info and thread_info['uid'] or 'n/a')) - for filename, lineno, name, line in traceback.extract_stack(stack): - code.append('File: "%s", line %d, in %s' % (filename, lineno, name)) - if line: - code.append(" %s" % (line.strip())) - _logger.info("\n".join(code)) - -def setup_signal_handlers(): - """ Register the signal handler defined above. """ - SIGNALS = map(lambda x: getattr(signal, "SIG%s" % x), "INT TERM".split()) - if os.name == 'posix': - map(lambda sig: signal.signal(sig, signal_handler), SIGNALS) - signal.signal(signal.SIGQUIT, dumpstacks) - elif os.name == 'nt': - import win32api - win32api.SetConsoleCtrlHandler(lambda sig: signal_handler(sig, None), 1) - -def quit_on_signals(): - """ Wait for one or two signals then shutdown the server. - - The first SIGINT or SIGTERM signal will initiate a graceful shutdown while - a second one if any will force an immediate exit. - - """ - # Wait for a first signal to be handled. (time.sleep will be interrupted - # by the signal handler.) The try/except is for the win32 case. - try: - while quit_signals_received == 0: - time.sleep(60) - except KeyboardInterrupt: - pass - - config = openerp.tools.config - if config['pidfile']: - os.unlink(config['pidfile']) - - openerp.service.stop_services() - sys.exit(0) - -def configure_babel_localedata_path(): - # Workaround: py2exe and babel. - if hasattr(sys, 'frozen'): - import babel - babel.localedata._dirname = os.path.join(os.path.dirname(sys.executable), 'localedata') - -def main(): - os.environ["TZ"] = "UTC" - - check_root_user() - openerp.tools.config.parse_config(sys.argv[1:]) - - check_postgres_user() - openerp.netsvc.init_logger() - report_configuration() - - config = openerp.tools.config - - configure_babel_localedata_path() - - setup_signal_handlers() - - if config["test_file"]: - run_test_file(config['db_name'], config['test_file']) - sys.exit(0) - - if config["translate_out"]: - export_translation() - sys.exit(0) - - if config["translate_in"]: - import_translation() - sys.exit(0) - - if not config["stop_after_init"]: - setup_pid_file() - # Some module register themselves when they are loaded so we need the - # services to be running before loading any registry. - if config['workers']: - openerp.service.start_services_workers() - else: - openerp.service.start_services() - - if config['db_name']: - for dbname in config['db_name'].split(','): - preload_registry(dbname) - - if config["stop_after_init"]: - sys.exit(0) - - _logger.info('OpenERP server is running, waiting for connections...') - quit_on_signals() - -if __name__ == "__main__": - main() - -# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: +from openerp.main import main +main() diff --git a/openerp/main.py b/openerp/main.py new file mode 100644 index 00000000000..5501eba2159 --- /dev/null +++ b/openerp/main.py @@ -0,0 +1,271 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +############################################################################## +# +# OpenERP, Open Source Management Solution +# Copyright (C) 2004-2009 Tiny SPRL (). +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as +# published by the Free Software Foundation, either version 3 of the +# License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see . +# +############################################################################## + +""" +OpenERP - Server +OpenERP is an ERP+CRM program for small and medium businesses. + +The whole source code is distributed under the terms of the +GNU Public Licence. + +(c) 2003-TODAY, Fabien Pinckaers - OpenERP SA +""" + +import logging +import os +import signal +import sys +import threading +import traceback +import time + +import openerp +__author__ = openerp.release.author +__version__ = openerp.release.version + +# Also use the `openerp` logger for the main script. +_logger = logging.getLogger('openerp') + +def check_root_user(): + """ Exit if the process's user is 'root' (on POSIX system).""" + if os.name == 'posix': + import pwd + if pwd.getpwuid(os.getuid())[0] == 'root' : + sys.stderr.write("Running as user 'root' is a security risk, aborting.\n") + sys.exit(1) + +def check_postgres_user(): + """ Exit if the configured database user is 'postgres'. + + This function assumes the configuration has been initialized. + """ + config = openerp.tools.config + if config['db_user'] == 'postgres': + sys.stderr.write("Using the database user 'postgres' is a security risk, aborting.") + sys.exit(1) + +def report_configuration(): + """ Log the server version and some configuration values. + + This function assumes the configuration has been initialized. + """ + config = openerp.tools.config + _logger.info("OpenERP version %s", __version__) + for name, value in [('addons paths', config['addons_path']), + ('database hostname', config['db_host'] or 'localhost'), + ('database port', config['db_port'] or '5432'), + ('database user', config['db_user'])]: + _logger.info("%s: %s", name, value) + +def setup_pid_file(): + """ Create a file with the process id written in it. + + This function assumes the configuration has been initialized. + """ + config = openerp.tools.config + if config['pidfile']: + fd = open(config['pidfile'], 'w') + pidtext = "%d" % (os.getpid()) + fd.write(pidtext) + fd.close() + +def preload_registry(dbname): + """ Preload a registry, and start the cron.""" + try: + db, registry = openerp.pooler.get_db_and_pool(dbname, update_module=openerp.tools.config['init'] or openerp.tools.config['update'], pooljobs=False) + + # jobs will start to be processed later, when openerp.cron.start_master_thread() is called by openerp.service.start_services() + registry.schedule_cron_jobs() + except Exception: + _logger.exception('Failed to initialize database `%s`.', dbname) + +def run_test_file(dbname, test_file): + """ Preload a registry, possibly run a test file, and start the cron.""" + try: + config = openerp.tools.config + db, registry = openerp.pooler.get_db_and_pool(dbname, update_module=config['init'] or config['update'], pooljobs=False) + cr = db.cursor() + _logger.info('loading test file %s', test_file) + openerp.tools.convert_yaml_import(cr, 'base', file(test_file), 'test', {}, 'test', True) + cr.rollback() + cr.close() + except Exception: + _logger.exception('Failed to initialize database `%s` and run test file `%s`.', dbname, test_file) + +def export_translation(): + config = openerp.tools.config + dbname = config['db_name'] + + if config["language"]: + msg = "language %s" % (config["language"],) + else: + msg = "new language" + _logger.info('writing translation file for %s to %s', msg, + config["translate_out"]) + + fileformat = os.path.splitext(config["translate_out"])[-1][1:].lower() + buf = file(config["translate_out"], "w") + cr = openerp.pooler.get_db(dbname).cursor() + openerp.tools.trans_export(config["language"], + config["translate_modules"] or ["all"], buf, fileformat, cr) + cr.close() + buf.close() + + _logger.info('translation file written successfully') + +def import_translation(): + config = openerp.tools.config + context = {'overwrite': config["overwrite_existing_translations"]} + dbname = config['db_name'] + + cr = openerp.pooler.get_db(dbname).cursor() + openerp.tools.trans_load( cr, config["translate_in"], config["language"], + context=context) + cr.commit() + cr.close() + +# Variable keeping track of the number of calls to the signal handler defined +# below. This variable is monitored by ``quit_on_signals()``. +quit_signals_received = 0 + +def signal_handler(sig, frame): + """ Signal handler: exit ungracefully on the second handled signal. + + :param sig: the signal number + :param frame: the interrupted stack frame or None + """ + global quit_signals_received + quit_signals_received += 1 + if quit_signals_received > 1: + # logging.shutdown was already called at this point. + sys.stderr.write("Forced shutdown.\n") + os._exit(0) + +def dumpstacks(sig, frame): + """ Signal handler: dump a stack trace for each existing thread.""" + # code from http://stackoverflow.com/questions/132058/getting-stack-trace-from-a-running-python-application#answer-2569696 + # modified for python 2.5 compatibility + threads_info = dict([(th.ident, {'name': th.name, + 'uid': getattr(th,'uid','n/a')}) + for th in threading.enumerate()]) + code = [] + for threadId, stack in sys._current_frames().items(): + thread_info = threads_info.get(threadId) + code.append("\n# Thread: %s (id:%s) (uid:%s)" % \ + (thread_info and thread_info['name'] or 'n/a', + threadId, + thread_info and thread_info['uid'] or 'n/a')) + for filename, lineno, name, line in traceback.extract_stack(stack): + code.append('File: "%s", line %d, in %s' % (filename, lineno, name)) + if line: + code.append(" %s" % (line.strip())) + _logger.info("\n".join(code)) + +def setup_signal_handlers(): + """ Register the signal handler defined above. """ + SIGNALS = map(lambda x: getattr(signal, "SIG%s" % x), "INT TERM".split()) + if os.name == 'posix': + map(lambda sig: signal.signal(sig, signal_handler), SIGNALS) + signal.signal(signal.SIGQUIT, dumpstacks) + elif os.name == 'nt': + import win32api + win32api.SetConsoleCtrlHandler(lambda sig: signal_handler(sig, None), 1) + +def quit_on_signals(): + """ Wait for one or two signals then shutdown the server. + + The first SIGINT or SIGTERM signal will initiate a graceful shutdown while + a second one if any will force an immediate exit. + + """ + # Wait for a first signal to be handled. (time.sleep will be interrupted + # by the signal handler.) The try/except is for the win32 case. + try: + while quit_signals_received == 0: + time.sleep(60) + except KeyboardInterrupt: + pass + + config = openerp.tools.config + if config['pidfile']: + os.unlink(config['pidfile']) + + openerp.service.stop_services() + sys.exit(0) + +def configure_babel_localedata_path(): + # Workaround: py2exe and babel. + if hasattr(sys, 'frozen'): + import babel + babel.localedata._dirname = os.path.join(os.path.dirname(sys.executable), 'localedata') + +def main(): + os.environ["TZ"] = "UTC" + + check_root_user() + openerp.tools.config.parse_config(sys.argv[1:]) + + check_postgres_user() + openerp.netsvc.init_logger() + report_configuration() + + config = openerp.tools.config + + configure_babel_localedata_path() + + setup_signal_handlers() + + if config["test_file"]: + run_test_file(config['db_name'], config['test_file']) + sys.exit(0) + + if config["translate_out"]: + export_translation() + sys.exit(0) + + if config["translate_in"]: + import_translation() + sys.exit(0) + + if not config["stop_after_init"]: + setup_pid_file() + # Some module register themselves when they are loaded so we need the + # services to be running before loading any registry. + if config['workers']: + openerp.service.start_services_workers() + else: + openerp.service.start_services() + + if config['db_name']: + for dbname in config['db_name'].split(','): + preload_registry(dbname) + + if config["stop_after_init"]: + sys.exit(0) + + _logger.info('OpenERP server is running, waiting for connections...') + quit_on_signals() + +if __name__ == "__main__": + main() + +# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: