[REF] osv_pool cleaning:
[odoo/odoo.git] / openerp-server
1 #!/usr/bin/env python
2 # -*- coding: utf-8 -*-
3 ##############################################################################
4 #
5 #    OpenERP, Open Source Management Solution
6 #    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
7 #
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.
12 #
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.
17 #
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/>.
20 #
21 ##############################################################################
22
23 """
24 OpenERP - Server
25 OpenERP is an ERP+CRM program for small and medium businesses.
26
27 The whole source code is distributed under the terms of the
28 GNU Public Licence.
29
30 (c) 2003-TODAY, Fabien Pinckaers - OpenERP s.a.
31 """
32
33 #----------------------------------------------------------
34 # python imports
35 #----------------------------------------------------------
36 import logging
37 import os
38 import signal
39 import sys
40 import threading
41 import traceback
42 import time
43
44 import openerp
45 __author__ = openerp.release.author
46 __version__ = openerp.release.version
47
48 if os.name == 'posix':
49     import pwd
50     # We DON't log this using the standard logger, because we might mess
51     # with the logfile's permissions. Just do a quick exit here.
52     if pwd.getpwuid(os.getuid())[0] == 'root' :
53         sys.stderr.write("Attempted to run OpenERP server as root. This is not good, aborting.\n")
54         sys.exit(1)
55
56 #-----------------------------------------------------------------------
57 # parse the command line
58 #-----------------------------------------------------------------------
59 openerp.tools.config.parse_config(sys.argv[1:])
60 config = openerp.tools.config
61
62 #----------------------------------------------------------
63 # get logger
64 #----------------------------------------------------------
65 openerp.netsvc.init_logger()
66 logger = logging.getLogger('server')
67
68 logger.info("OpenERP version - %s", __version__)
69 for name, value in [('addons_path', config['addons_path']),
70                     ('database hostname', config['db_host'] or 'localhost'),
71                     ('database port', config['db_port'] or '5432'),
72                     ('database user', config['db_user'])]:
73     logger.info("%s - %s", name, value)
74
75 # Don't allow if the connection to PostgreSQL done by postgres user
76 if config['db_user'] == 'postgres':
77     logger.error("Connecting to the database as 'postgres' user is forbidden, as it present major security issues. Shutting down.")
78     sys.exit(1)
79
80 #----------------------------------------------------------
81 # init net service
82 #----------------------------------------------------------
83 logger.info('initialising distributed objects services')
84
85 #----------------------------------------------------------
86 # Load and update databases if requested
87 #----------------------------------------------------------
88
89 if not ( config["stop_after_init"] or \
90     config["translate_in"] or \
91     config["translate_out"] ):
92     http_server = openerp.service.http_server
93     netrpc_server = openerp.service.netrpc_server
94     http_server.init_servers()
95     http_server.init_xmlrpc()
96     http_server.init_static_http()
97     netrpc_server.init_servers()
98
99 if config['db_name']:
100     for dbname in config['db_name'].split(','):
101         db, pool = openerp.pooler.get_db_and_pool(dbname, update_module=config['init'] or config['update'], pooljobs=False)
102         cr = db.cursor()
103
104         if config["test_file"]:
105             logger.info('loading test file %s', config["test_file"])
106             openerp.tools.convert_yaml_import(cr, 'base', file(config["test_file"]), {}, 'test', True)
107             cr.rollback()
108
109         pool.get('ir.cron')._poolJobs(db.dbname)
110
111         cr.close()
112
113 #----------------------------------------------------------
114 # translation stuff
115 #----------------------------------------------------------
116 if config["translate_out"]:
117     if config["language"]:
118         msg = "language %s" % (config["language"],)
119     else:
120         msg = "new language"
121     logger.info('writing translation file for %s to %s', msg, config["translate_out"])
122
123     fileformat = os.path.splitext(config["translate_out"])[-1][1:].lower()
124     buf = file(config["translate_out"], "w")
125     dbname = config['db_name']
126     cr = openerp.pooler.get_db(dbname).cursor()
127     openerp.tools.trans_export(config["language"], config["translate_modules"] or ["all"], buf, fileformat, cr)
128     cr.close()
129     buf.close()
130
131     logger.info('translation file written successfully')
132     sys.exit(0)
133
134 if config["translate_in"]:
135     context = {'overwrite': config["overwrite_existing_translations"]}
136     dbname = config['db_name']
137     cr = openerp.pooler.get_db(dbname).cursor()
138     openerp.tools.trans_load(cr,
139                      config["translate_in"], 
140                      config["language"],
141                      context=context)
142     openerp.tools.trans_update_res_ids(cr)
143     cr.commit()
144     cr.close()
145     sys.exit(0)
146
147 #----------------------------------------------------------------------------------
148 # if we don't want the server to continue to run after initialization, we quit here
149 #----------------------------------------------------------------------------------
150 if config["stop_after_init"]:
151     sys.exit(0)
152
153 openerp.netsvc.start_agent()
154
155 #----------------------------------------------------------
156 # Launch Servers
157 #----------------------------------------------------------
158
159 LST_SIGNALS = ['SIGINT', 'SIGTERM']
160
161 SIGNALS = dict(
162     [(getattr(signal, sign), sign) for sign in LST_SIGNALS]
163 )
164
165 quit_signals_received = 0
166
167 def handler(signum, frame):
168     """
169     :param signum: the signal number
170     :param frame: the interrupted stack frame or None
171     """
172     global quit_signals_received
173     quit_signals_received += 1
174     if quit_signals_received > 1:
175         sys.stderr.write("Forced shutdown.\n")
176         os._exit(0)
177
178 def dumpstacks(signum, frame):
179     # code from http://stackoverflow.com/questions/132058/getting-stack-trace-from-a-running-python-application#answer-2569696
180     # modified for python 2.5 compatibility
181     thread_map = dict(threading._active, **threading._limbo)
182     id2name = dict([(threadId, thread.getName()) for threadId, thread in thread_map.items()])
183     code = []
184     for threadId, stack in sys._current_frames().items():
185         code.append("\n# Thread: %s(%d)" % (id2name[threadId], threadId))
186         for filename, lineno, name, line in traceback.extract_stack(stack):
187             code.append('File: "%s", line %d, in %s' % (filename, lineno, name))
188             if line:
189                 code.append("  %s" % (line.strip()))
190     logging.getLogger('dumpstacks').info("\n".join(code))
191
192 for signum in SIGNALS:
193     signal.signal(signum, handler)
194
195 if os.name == 'posix':
196     signal.signal(signal.SIGQUIT, dumpstacks)
197
198 def quit():
199     openerp.netsvc.Agent.quit()
200     openerp.netsvc.Server.quitAll()
201     if config['pidfile']:
202         os.unlink(config['pidfile'])
203     logger = logging.getLogger('shutdown')
204     logger.info("Initiating OpenERP Server shutdown")
205     logger.info("Hit CTRL-C again or send a second signal to immediately terminate the server...")
206     logging.shutdown()
207
208     # manually join() all threads before calling sys.exit() to allow a second signal
209     # to trigger _force_quit() in case some non-daemon threads won't exit cleanly.
210     # threading.Thread.join() should not mask signals (at least in python 2.5)
211     for thread in threading.enumerate():
212         if thread != threading.currentThread() and not thread.isDaemon():
213             while thread.isAlive():
214                 # need a busyloop here as thread.join() masks signals
215                 # and would present the forced shutdown
216                 thread.join(0.05)
217                 time.sleep(0.05)
218     sys.exit(0)
219
220 if config['pidfile']:
221     fd = open(config['pidfile'], 'w')
222     pidtext = "%d" % (os.getpid())
223     fd.write(pidtext)
224     fd.close()
225
226 openerp.netsvc.Server.startAll()
227
228 logger.info('OpenERP server is running, waiting for connections...')
229
230 while quit_signals_received == 0:
231     time.sleep(60)
232
233 quit()
234
235 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: