~therp-nl/ocb-server/6.1-lp1175516-expression_search_translations_with_fallback_on_untranslated

#!/usr/bin/env python

# -*- coding: utf-8 -*-

##############################################################################

#

#    OpenERP, Open Source Management Solution

#    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).

#

#    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 <http://www.gnu.org/licenses/>.

#

##############################################################################

"""

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 imp

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=config['init'] or 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:

        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', 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

    thread_map = dict(threading._active, **threading._limbo)

    id2name = dict([(threadId, thread.getName()) for threadId, thread in thread_map.items()])

    code = []

    for threadId, stack in sys._current_frames().items():

        code.append("\n# Thread: %s(%d)" % (id2name.get(threadId,'n/a'), threadId))

        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, e:

        pass

    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')

if __name__ == "__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"]:

        # Some module register themselves when they are loaded so we need the

        # services to be running before loading any registry.

        openerp.service.start_services()

    for m in openerp.conf.server_wide_modules:

        try:

            openerp.modules.module.load_openerp_module(m)

        except Exception:

            msg = ''

            if m == 'web':

                msg = """

The `web` module is provided by the addons found in the `openerp-web` project.

Maybe you forgot to add those addons in your addons_path configuration."""

            _logger.exception('Failed to load server-wide module `%s`.%s', m, msg)

    if config['db_name']:

        for dbname in config['db_name'].split(','):

            preload_registry(dbname)

    if config["stop_after_init"]:

        sys.exit(0)

    setup_pid_file()

    _logger.info('OpenERP server is running, waiting for connections...')

    quit_on_signals()

# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: