Cédric Krier
b75d10cd20
Move workflow method in ModelWorkflow Move workflow service in workflow module Remove Service and LocalService Remove web_service and use a dispatcher in protocols New netrpc/xmlrpc syntax to be easier to introspect
250 lines
8.8 KiB
Python
250 lines
8.8 KiB
Python
#This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
#this repository contains the full copyright notices and license terms.
|
|
"""
|
|
%prog [options]
|
|
"""
|
|
import logging
|
|
format='[%(asctime)s] %(levelname)s:%(name)s:%(message)s'
|
|
datefmt='%a %b %d %H:%M:%S %Y'
|
|
logging.basicConfig(level=logging.DEBUG, format=format, datefmt=datefmt)
|
|
import sys, os, signal
|
|
import netsvc
|
|
import time
|
|
from trytond.backend import Database, DatabaseOperationalError
|
|
from config import CONFIG
|
|
from trytond.modules import register_classes
|
|
import mx.DateTime
|
|
|
|
if not hasattr(mx.DateTime, 'strptime'):
|
|
mx.DateTime.strptime = lambda x, y: mx.DateTime.mktime(
|
|
time.strptime(x, y))
|
|
|
|
from getpass import getpass
|
|
import sha
|
|
import logging
|
|
import logging.handlers
|
|
import threading
|
|
from pool import Pool
|
|
|
|
|
|
class TrytonServer(object):
|
|
|
|
def __init__(self):
|
|
|
|
if CONFIG['logfile']:
|
|
logf = CONFIG['logfile']
|
|
# test if the directories exist, else create them
|
|
try:
|
|
diff = 0
|
|
if os.path.isfile(logf):
|
|
diff = int(time.time()) - int(os.stat(logf)[-1])
|
|
handler = logging.handlers.TimedRotatingFileHandler(
|
|
logf, 'D', 1, 30)
|
|
handler.rolloverAt -= diff
|
|
except Exception, exception:
|
|
sys.stderr.write("ERROR: couldn't create the logfile directory:" \
|
|
+ str(exception))
|
|
else:
|
|
formatter = logging.Formatter(format, datefmt)
|
|
# tell the handler to use this format
|
|
handler.setFormatter(formatter)
|
|
|
|
# add the handler to the root logger
|
|
logging.getLogger().addHandler(handler)
|
|
logging.getLogger().setLevel(logging.INFO)
|
|
elif os.name != 'nt':
|
|
reverse = '\x1b[7m'
|
|
reset = '\x1b[0m'
|
|
# reverse color for error and critical messages
|
|
for level in logging.ERROR, logging.CRITICAL:
|
|
msg = reverse + logging.getLevelName(level) + reset
|
|
logging.addLevelName(level, msg)
|
|
|
|
self.logger = logging.getLogger("init")
|
|
|
|
if CONFIG.configfile:
|
|
self.logger.info('using %s as configuration file' % \
|
|
CONFIG.configfile)
|
|
self.logger.info('initialising distributed objects services')
|
|
|
|
def run(self):
|
|
"Run the server and never return"
|
|
|
|
update = False
|
|
init = {}
|
|
for db_name in CONFIG["db_name"]:
|
|
init[db_name] = False
|
|
try:
|
|
database = Database(db_name).connect()
|
|
cursor = database.cursor()
|
|
except DatabaseOperationalError:
|
|
self.logger.error("could not connect to database '%s'!" % db_name,)
|
|
continue
|
|
|
|
if CONFIG['init']:
|
|
if not cursor.test():
|
|
self.logger.info("init db")
|
|
Database.init(cursor)
|
|
init[db_name] = True
|
|
cursor.commit()
|
|
cursor.close()
|
|
elif not cursor.test():
|
|
self.logger.error("'%s' is not a Tryton database!" % db_name)
|
|
|
|
register_classes()
|
|
|
|
for db_name in CONFIG["db_name"]:
|
|
try:
|
|
cursor = Database(db_name).connect().cursor()
|
|
except DatabaseOperationalError:
|
|
self.logger.error("could not connect to database '%s'!" % db_name,)
|
|
continue
|
|
if not cursor.test():
|
|
cursor.close()
|
|
continue
|
|
cursor.execute('SELECT code FROM ir_lang ' \
|
|
'WHERE translatable = True')
|
|
lang = [x[0] for x in cursor.fetchall()]
|
|
cursor.close()
|
|
update = bool(CONFIG['init'] or CONFIG['update'])
|
|
Pool(db_name).init(update=update, lang=lang)
|
|
|
|
for kind in ('init', 'update'):
|
|
CONFIG[kind] = {}
|
|
|
|
for db_name in CONFIG['db_name']:
|
|
if init[db_name]:
|
|
while True:
|
|
password = getpass('Admin Password for %s:' % db_name)
|
|
password2 = getpass('Admin Password Confirmation:')
|
|
if password != password2:
|
|
sys.stderr.write('Admin Password Confirmation doesn\'t match ' \
|
|
'Admin Password!\n')
|
|
continue
|
|
if not password:
|
|
sys.stderr.write('Admin Password is required!\n')
|
|
continue
|
|
break
|
|
|
|
database = Database(db_name).connect()
|
|
cursor = database.cursor()
|
|
cursor.execute('UPDATE res_user ' \
|
|
'SET password = %s ' \
|
|
'WHERE login = \'admin\'',
|
|
(sha.new(password).hexdigest(),))
|
|
cursor.commit()
|
|
cursor.close()
|
|
|
|
if update:
|
|
self.logger.info('Update/Init succeed!')
|
|
logging.shutdown()
|
|
sys.exit(0)
|
|
|
|
# Launch Server
|
|
if CONFIG['xmlrpc']:
|
|
interface = CONFIG["interface"]
|
|
try:
|
|
port = int(CONFIG["xmlport"])
|
|
except:
|
|
self.logger.error("invalid port '%s'!" % (CONFIG["xmlport"],))
|
|
logging.shutdown()
|
|
sys.exit(1)
|
|
|
|
httpd = netsvc.HttpDaemon(interface, port, CONFIG['secure_xmlrpc'])
|
|
|
|
xml_gw = netsvc.XmlRpc.RpcGateway('web-service')
|
|
httpd.attach("/xmlrpc", xml_gw )
|
|
logging.getLogger("web-service").info(
|
|
"starting XML-RPC" + \
|
|
(CONFIG['secure_xmlrpc'] and ' Secure' or '') + \
|
|
" services, port " + str(port))
|
|
|
|
if CONFIG['netrpc']:
|
|
interface = CONFIG["interface"]
|
|
try:
|
|
port = int(CONFIG["netport"])
|
|
except:
|
|
self.logger.error("invalid port '%s'!" % (CONFIG["netport"],))
|
|
logging.shutdown()
|
|
sys.exit(1)
|
|
|
|
tinysocket = netsvc.TinySocketServerThread(interface, port,
|
|
CONFIG['secure_netrpc'])
|
|
logging.getLogger("web-service").info(
|
|
"starting netrpc" + \
|
|
(CONFIG['secure_netrpc'] and ' Secure' or '') + \
|
|
" service, port " + str(port))
|
|
|
|
if CONFIG['webdav']:
|
|
interface = CONFIG['interface']
|
|
try:
|
|
port = int(CONFIG['webdavport'])
|
|
except:
|
|
self.logger.error("invalid port '%s'!" % (CONFIG['webdavport'],))
|
|
logging.shutdown()
|
|
sys.exit(1)
|
|
|
|
webdavd = netsvc.WebDAVServerThread(interface, port,
|
|
CONFIG['secure_webdav'])
|
|
logging.getLogger("web-service").info(
|
|
'starting webdav' + \
|
|
(CONFIG['secure_webdav'] and ' Secure' or '') + \
|
|
' service, port ' + str(port))
|
|
|
|
def handler(signum, frame):
|
|
if hasattr(signal, 'SIGUSR1'):
|
|
if signum == signal.SIGUSR1:
|
|
for db_name in Database.connected_list():
|
|
Pool(db_name).init()
|
|
return
|
|
if CONFIG['netrpc']:
|
|
tinysocket.stop()
|
|
if CONFIG['xmlrpc']:
|
|
httpd.stop()
|
|
if CONFIG['webdav']:
|
|
webdavd.stop()
|
|
if CONFIG['pidfile']:
|
|
os.unlink(CONFIG['pidfile'])
|
|
logging.getLogger('web-service').info('stop server')
|
|
logging.shutdown()
|
|
sys.exit(0)
|
|
|
|
if CONFIG['pidfile']:
|
|
fd_pid = open(CONFIG['pidfile'], 'w')
|
|
pidtext = "%d" % (os.getpid())
|
|
fd_pid.write(pidtext)
|
|
fd_pid.close()
|
|
|
|
signal.signal(signal.SIGINT, handler)
|
|
signal.signal(signal.SIGTERM, handler)
|
|
if hasattr(signal, 'SIGQUIT'):
|
|
signal.signal(signal.SIGQUIT, handler)
|
|
if hasattr(signal, 'SIGUSR1'):
|
|
signal.signal(signal.SIGUSR1, handler)
|
|
|
|
self.logger.info('the server is running, waiting for connections...')
|
|
if CONFIG['netrpc']:
|
|
tinysocket.start()
|
|
if CONFIG['xmlrpc']:
|
|
httpd.start()
|
|
if CONFIG['webdav']:
|
|
webdavd.start()
|
|
|
|
if CONFIG['psyco']:
|
|
import psyco
|
|
psyco.full()
|
|
|
|
while True:
|
|
for dbname in Pool.database_list():
|
|
pool = Pool(dbname)
|
|
cron_obj = pool.get('ir.cron')
|
|
thread = threading.Thread(
|
|
target=cron_obj.pool_jobs,
|
|
args=(dbname,), kwargs={})
|
|
thread.start()
|
|
time.sleep(60)
|
|
|
|
if __name__ == "__main__":
|
|
SERVER = TrytonServer()
|
|
SERVER.run()
|