2 # -*- coding: utf-8 -*-
3 ##############################################################################
5 # OpenERP, Open Source Management Solution
6 # Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
8 # This program is free software: you can redistribute it and/or modify
9 # it under the terms of the GNU Affero General Public License as
10 # published by the Free Software Foundation, either version 3 of the
11 # License, or (at your option) any later version.
13 # This program is distributed in the hope that it will be useful,
14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 # GNU Affero General Public License for more details.
18 # You should have received a copy of the GNU Affero General Public License
19 # along with this program. If not, see <http://www.gnu.org/licenses/>.
21 ##############################################################################
25 OpenERP is an ERP+CRM program for small and medium businesses.
27 The whole source code is distributed under the terms of the
30 (c) 2003-TODAY, Fabien Pinckaers - OpenERP SA
42 __author__ = openerp.release.author
43 __version__ = openerp.release.version
48 def check_root_user():
49 """ Exit if the process's user is 'root' (on POSIX system)."""
50 if os.name == 'posix':
52 if pwd.getpwuid(os.getuid())[0] == 'root' :
53 sys.stderr.write("Running as user 'root' is a security risk, aborting.\n")
56 def check_postgres_user():
57 """ Exit if the configured database user is 'postgres'.
59 This function assumes the configuration has been initialized.
61 config = openerp.tools.config
62 if config['db_user'] == 'postgres':
63 sys.stderr.write("Using the database user 'postgres' is a security risk, aborting.")
66 def report_configuration():
67 """ Log the server version and some configuration values.
69 This function assumes the configuration has been initialized.
71 config = openerp.tools.config
72 logger = logging.getLogger('server')
73 logger.info("OpenERP version %s", __version__)
74 for name, value in [('addons paths', config['addons_path']),
75 ('database hostname', config['db_host'] or 'localhost'),
76 ('database port', config['db_port'] or '5432'),
77 ('database user', config['db_user'])]:
78 logger.info("%s: %s", name, value)
81 """ Create a file with the process id written in it.
83 This function assumes the configuration has been initialized.
85 config = openerp.tools.config
87 fd = open(config['pidfile'], 'w')
88 pidtext = "%d" % (os.getpid())
92 def preload_registry(dbname):
93 """ Preload a registry, and start the cron."""
95 db, registry = openerp.pooler.get_db_and_pool(dbname, update_module=config['init'] or config['update'], pooljobs=False)
97 # jobs will start to be processed later, when openerp.cron.start_master_thread() is called by openerp.service.start_services()
98 registry.schedule_cron_jobs()
100 logging.exception('Failed to initialize database `%s`.', dbname)
102 def run_test_file(dbname, test_file):
103 """ Preload a registry, possibly run a test file, and start the cron."""
105 db, registry = openerp.pooler.get_db_and_pool(dbname, update_module=config['init'] or config['update'], pooljobs=False)
107 logger = logging.getLogger('server')
108 logger.info('loading test file %s', test_file)
109 openerp.tools.convert_yaml_import(cr, 'base', file(test_file), {}, 'test', True)
113 logging.exception('Failed to initialize database `%s` and run test file `%s`.', dbname, test_file)
116 def export_translation():
117 config = openerp.tools.config
118 dbname = config['db_name']
119 logger = logging.getLogger('server')
121 if config["language"]:
122 msg = "language %s" % (config["language"],)
125 logger.info('writing translation file for %s to %s', msg,
126 config["translate_out"])
128 fileformat = os.path.splitext(config["translate_out"])[-1][1:].lower()
129 buf = file(config["translate_out"], "w")
130 cr = openerp.pooler.get_db(dbname).cursor()
131 openerp.tools.trans_export(config["language"],
132 config["translate_modules"] or ["all"], buf, fileformat, cr)
136 logger.info('translation file written successfully')
138 def import_translation():
139 config = openerp.tools.config
140 context = {'overwrite': config["overwrite_existing_translations"]}
141 dbname = config['db_name']
143 cr = openerp.pooler.get_db(dbname).cursor()
144 openerp.tools.trans_load( cr, config["translate_in"], config["language"],
146 openerp.tools.trans_update_res_ids(cr)
150 # Variable keeping track of the number of calls to the signal handler defined
151 # below. This variable is monitored by ``quit_on_signals()``.
152 quit_signals_received = 0
154 def signal_handler(sig, frame):
155 """ Signal handler: exit ungracefully on the second handled signal.
157 :param sig: the signal number
158 :param frame: the interrupted stack frame or None
160 global quit_signals_received
161 quit_signals_received += 1
162 if quit_signals_received > 1:
163 # logging.shutdown was already called at this point.
164 sys.stderr.write("Forced shutdown.\n")
167 def dumpstacks(sig, frame):
168 """ Signal handler: dump a stack trace for each existing thread."""
169 # code from http://stackoverflow.com/questions/132058/getting-stack-trace-from-a-running-python-application#answer-2569696
170 # modified for python 2.5 compatibility
171 thread_map = dict(threading._active, **threading._limbo)
172 id2name = dict([(threadId, thread.getName()) for threadId, thread in thread_map.items()])
174 for threadId, stack in sys._current_frames().items():
175 code.append("\n# Thread: %s(%d)" % (id2name[threadId], threadId))
176 for filename, lineno, name, line in traceback.extract_stack(stack):
177 code.append('File: "%s", line %d, in %s' % (filename, lineno, name))
179 code.append(" %s" % (line.strip()))
180 logging.getLogger('dumpstacks').info("\n".join(code))
182 def setup_signal_handlers():
183 """ Register the signal handler defined above. """
184 SIGNALS = map(lambda x: getattr(signal, "SIG%s" % x), "INT TERM".split())
185 if os.name == 'posix':
186 map(lambda sig: signal.signal(sig, signal_handler), SIGNALS)
187 signal.signal(signal.SIGQUIT, dumpstacks)
188 elif os.name == 'nt':
190 win32api.SetConsoleCtrlHandler(lambda sig: signal_handler(sig, None), 1)
192 def quit_on_signals():
193 """ Wait for one or two signals then shutdown the server.
195 The first SIGINT or SIGTERM signal will initiate a graceful shutdown while
196 a second one if any will force an immediate exit.
199 # Wait for a first signal to be handled. (time.sleep will be interrupted
200 # by the signal handler.) The try/except is for the win32 case.
202 while quit_signals_received == 0:
204 except KeyboardInterrupt, e:
207 if config['pidfile']:
208 os.unlink(config['pidfile'])
210 openerp.service.stop_services()
213 if __name__ == "__main__":
215 os.environ["TZ"] = "UTC"
218 openerp.tools.config.parse_config(sys.argv[1:])
221 class ImportHook(object):
223 def find_module(self, module_name, package_path):
224 module_parts = module_name.split('.')
225 if len(module_parts) == 3 and module_name.startswith('openerp.modules.'):
226 return self # We act as a loader too.
228 # TODO list of loadable modules can be cached instead of always
229 # calling get_module_path().
230 if len(module_parts) == 1 and \
231 openerp.modules.module.get_module_path(module_parts[0],
232 display_warning=False):
233 return self # We act as a loader too.
235 def load_module(self, module_name):
237 module_parts = module_name.split('.')
238 if len(module_parts) == 3 and module_name.startswith('openerp.modules.'):
239 module_part = module_parts[2]
240 if module_name in sys.modules:
241 return sys.modules[module_name]
243 if len(module_parts) == 1:
244 module_part = module_parts[0]
245 if module_part in sys.modules:
246 return sys.modules[module_part]
249 # Check if the bare module name clashes with another module.
250 f, path, descr = imp.find_module(module_part)
251 logger = logging.getLogger('init')
253 Ambiguous import: the OpenERP module `%s` is shadowing another module
254 (available at %s).""" % (module_name, path))
255 except ImportError, e:
256 # Using `import <module_name>` instead of
257 # `import openerp.modules.<module_name>` is ugly but not harmful.
260 f, path, descr = imp.find_module(module_part, openerp.modules.module.ad_paths)
261 mod = imp.load_module(module_name, f, path, descr)
262 sys.modules[module_part] = mod
263 sys.modules['openerp.modules.' + module_part] = mod
266 openerp.modules.module.initialize_sys_path()
267 sys.meta_path.append(ImportHook())
269 check_postgres_user()
270 openerp.netsvc.init_logger()
271 report_configuration()
273 config = openerp.tools.config
275 setup_signal_handlers()
277 if config["test_file"]:
278 run_test_file(config['db_name'], config['test_file'])
281 if config["translate_out"]:
285 if config["translate_in"]:
289 if not config["stop_after_init"]:
290 # Some module register themselves when they are loaded so we need the
291 # services to be running before loading any registry.
292 openerp.service.start_services()
294 for m in openerp.conf.server_wide_modules:
297 # Call any post_load hook.
298 info = openerp.modules.module.load_information_from_description_file(m)
299 if info['post_load']:
300 getattr(sys.modules[m], info['post_load'])()
305 The `web` module is provided by the addons found in the `openerp-web` project.
306 Maybe you forgot to add those addons in your addons_path configuration."""
307 logging.exception('Failed to load server-wide module `%s`.%s', m, msg)
309 if config['db_name']:
310 for dbname in config['db_name'].split(','):
311 preload_registry(dbname)
313 if config["stop_after_init"]:
317 logger = logging.getLogger('server')
318 logger.info('OpenERP server is running, waiting for connections...')
321 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: