[REF] openerp-server: better imports.
[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 #----------------------------------------------------------
154 # Launch Servers
155 #----------------------------------------------------------
156
157 LST_SIGNALS = ['SIGINT', 'SIGTERM']
158
159 SIGNALS = dict(
160     [(getattr(signal, sign), sign) for sign in LST_SIGNALS]
161 )
162
163 quit_signals_received = 0
164
165 def handler(signum, frame):
166     """
167     :param signum: the signal number
168     :param frame: the interrupted stack frame or None
169     """
170     global quit_signals_received
171     quit_signals_received += 1
172     if quit_signals_received > 1:
173         sys.stderr.write("Forced shutdown.\n")
174         os._exit(0)
175
176 def dumpstacks(signum, frame):
177     # code from http://stackoverflow.com/questions/132058/getting-stack-trace-from-a-running-python-application#answer-2569696
178     # modified for python 2.5 compatibility
179     thread_map = dict(threading._active, **threading._limbo)
180     id2name = dict([(threadId, thread.getName()) for threadId, thread in thread_map.items()])
181     code = []
182     for threadId, stack in sys._current_frames().items():
183         code.append("\n# Thread: %s(%d)" % (id2name[threadId], threadId))
184         for filename, lineno, name, line in traceback.extract_stack(stack):
185             code.append('File: "%s", line %d, in %s' % (filename, lineno, name))
186             if line:
187                 code.append("  %s" % (line.strip()))
188     logging.getLogger('dumpstacks').info("\n".join(code))
189
190 for signum in SIGNALS:
191     signal.signal(signum, handler)
192
193 if os.name == 'posix':
194     signal.signal(signal.SIGQUIT, dumpstacks)
195
196 def quit():
197     openerp.netsvc.Agent.quit()
198     openerp.netsvc.Server.quitAll()
199     if config['pidfile']:
200         os.unlink(config['pidfile'])
201     logger = logging.getLogger('shutdown')
202     logger.info("Initiating OpenERP Server shutdown")
203     logger.info("Hit CTRL-C again or send a second signal to immediately terminate the server...")
204     logging.shutdown()
205
206     # manually join() all threads before calling sys.exit() to allow a second signal
207     # to trigger _force_quit() in case some non-daemon threads won't exit cleanly.
208     # threading.Thread.join() should not mask signals (at least in python 2.5)
209     for thread in threading.enumerate():
210         if thread != threading.currentThread() and not thread.isDaemon():
211             while thread.isAlive():
212                 # need a busyloop here as thread.join() masks signals
213                 # and would present the forced shutdown
214                 thread.join(0.05)
215                 time.sleep(0.05)
216     sys.exit(0)
217
218 if config['pidfile']:
219     fd = open(config['pidfile'], 'w')
220     pidtext = "%d" % (os.getpid())
221     fd.write(pidtext)
222     fd.close()
223
224 openerp.netsvc.Server.startAll()
225
226 logger.info('OpenERP server is running, waiting for connections...')
227
228 while quit_signals_received == 0:
229     time.sleep(60)
230
231 quit()
232
233 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: