bin/tools/config.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
   6 #    $Id$
   7 #
   8 #    This program is free software: you can redistribute it and/or modify
   9 #    it under the terms of the GNU General Public License as published by
  10 #    the Free Software Foundation, either version 3 of the License, or
  11 #    (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 General Public License for more details.
  17 #
  18 #    You should have received a copy of the GNU General Public License
  19 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 #
  21 ##############################################################################
  22
  23 import ConfigParser
  24 import optparse
  25 import os
  26 import sys
  27 import netsvc
  28 import logging
  29 import release
  30
  31 def check_ssl():
  32     try:
  33         from OpenSSL import SSL
  34         import socket
  35
  36         return hasattr(socket, 'ssl')
  37     except:
  38         return False
  39
  40 class configmanager(object):
  41     def __init__(self, fname=None):
  42         self.options = {
  43             'email_from':False,
  44             'interface': '',    # this will bind the server to all interfaces
  45             'port': 8069,
  46             'netinterface': '',
  47             'netport': 8070,
  48             'db_host': False,
  49             'db_port': False,
  50             'db_name': False,
  51             'db_user': False,
  52             'db_password': False,
  53             'db_maxconn': 64,
  54             'reportgz': False,
  55             'netrpc': True,
  56             'xmlrpc': True,
  57             'soap': False,
  58             'translate_in': None,
  59             'translate_out': None,
  60             'language': None,
  61             'pg_path': None,
  62             'admin_passwd': 'admin',
  63             'addons_path': None,
  64             'root_path': None,
  65             'debug_mode': False,
  66             'import_partial': "",
  67             'pidfile': None,
  68             'logfile': None,
  69             'smtp_server': 'localhost',
  70             'smtp_user': False,
  71             'smtp_port':25,
  72             'smtp_password': False,
  73             'stop_after_init': False,   # this will stop the server after initialization
  74             'price_accuracy': 2,
  75             'secure' : False,
  76             'syslog' : False,
  77             'log_level': logging.INFO,
  78             'assert_exit_level': logging.WARNING, # level above which a failed assert will be raise
  79             'cache_timeout': 100000,
  80         }
  81
  82         hasSSL = check_ssl()
  83
  84         loglevels = dict([(getattr(netsvc, 'LOG_%s' % x), getattr(logging, x))
  85                           for x in ('CRITICAL', 'ERROR', 'WARNING', 'INFO', 'DEBUG', 'DEBUG_RPC', 'NOTSET')])
  86
  87         version = "%s %s" % (release.description, release.version)
  88         parser = optparse.OptionParser(version=version)
  89
  90         parser.add_option("-c", "--config", dest="config", help="specify alternate config file")
  91         parser.add_option("-s", "--save", action="store_true", dest="save", default=False,
  92                           help="save configuration to ~/.openerp_serverrc")
  93         parser.add_option("--pidfile", dest="pidfile", help="file where the server pid will be stored")
  94
  95         parser.add_option("-n", "--interface", dest="interface", help="specify the TCP IP address")
  96         parser.add_option("-p", "--port", dest="port", help="specify the TCP port", type="int")
  97         parser.add_option("--net_interface", dest="netinterface", help="specify the TCP IP address for netrpc")
  98         parser.add_option("--net_port", dest="netport", help="specify the TCP port for netrpc", type="int")
  99         parser.add_option("--no-netrpc", dest="netrpc", action="store_false", default=True, help="disable netrpc")
 100         parser.add_option("--no-xmlrpc", dest="xmlrpc", action="store_false", default=True, help="disable xmlrpc")
 101         parser.add_option("-i", "--init", dest="init", help="init a module (use \"all\" for all modules)")
 102         parser.add_option("--without-demo", dest="without_demo",
 103                           help="load demo data for a module (use \"all\" for all modules)", default=False)
 104         parser.add_option("-u", "--update", dest="update",
 105                           help="update a module (use \"all\" for all modules)")
 106         parser.add_option("--cache-timeout", dest="cache_timeout",
 107                           help="set the timeout for the cache system", default=100000, type="int")
 108
 109         # stops the server from launching after initialization
 110         parser.add_option("--stop-after-init", action="store_true", dest="stop_after_init", default=False,
 111                           help="stop the server after it initializes")
 112         parser.add_option('--debug', dest='debug_mode', action='store_true', default=False, help='enable debug mode')
 113         parser.add_option("--assert-exit-level", dest='assert_exit_level', type="choice", choices=loglevels.keys(),
 114                           help="specify the level at which a failed assertion will stop the server. Accepted values: %s" % (loglevels.keys(),))
 115         if hasSSL:
 116             group = optparse.OptionGroup(parser, "SSL Configuration")
 117             group.add_option("-S", "--secure", dest="secure", action="store_true",
 118                              help="launch server over https instead of http", default=False)
 119             group.add_option("--cert-file", dest="secure_cert_file",
 120                               default="server.cert",
 121                               help="specify the certificate file for the SSL connection")
 122             group.add_option("--pkey-file", dest="secure_pkey_file",
 123                               default="server.pkey",
 124                               help="specify the private key file for the SSL connection")
 125             parser.add_option_group(group)
 126
 127         # Logging Group
 128         group = optparse.OptionGroup(parser, "Logging Configuration")
 129         group.add_option("--logfile", dest="logfile", help="file where the server log will be stored")
 130         group.add_option("--syslog", action="store_true", dest="syslog",
 131                          default=False, help="Send the log to the syslog server")
 132         group.add_option('--log-level', dest='log_level', type='choice', choices=loglevels.keys(),
 133                          help='specify the level of the logging. Accepted values: ' + str(loglevels.keys()))
 134         parser.add_option_group(group)
 135
 136         # SMTP Group
 137         group = optparse.OptionGroup(parser, "SMTP Configuration")
 138         group.add_option('--email-from', dest='email_from', default='', help='specify the SMTP email address for sending email')
 139         group.add_option('--smtp', dest='smtp_server', default='', help='specify the SMTP server for sending email')
 140         group.add_option('--smtp-port', dest='smtp_port', default='25', help='specify the SMTP port', type="int")
 141         if hasSSL:
 142             group.add_option('--smtp-ssl', dest='smtp_ssl', default='', help='specify the SMTP server support SSL or not')
 143         group.add_option('--smtp-user', dest='smtp_user', default='', help='specify the SMTP username for sending email')
 144         group.add_option('--smtp-password', dest='smtp_password', default='', help='specify the SMTP password for sending email')
 145         group.add_option('--price_accuracy', dest='price_accuracy', default='2', help='specify the price accuracy')
 146         parser.add_option_group(group)
 147
 148         group = optparse.OptionGroup(parser, "Database related options")
 149         group.add_option("-d", "--database", dest="db_name", help="specify the database name")
 150         group.add_option("-r", "--db_user", dest="db_user", help="specify the database user name")
 151         group.add_option("-w", "--db_password", dest="db_password", help="specify the database password")
 152         group.add_option("--pg_path", dest="pg_path", help="specify the pg executable path")
 153         group.add_option("--db_host", dest="db_host", help="specify the database host")
 154         group.add_option("--db_port", dest="db_port", help="specify the database port", type="int")
 155         group.add_option("--db_maxconn", dest="db_maxconn", default='64',
 156                          help="specify the the maximum number of physical connections to posgresql")
 157         group.add_option("-P", "--import-partial", dest="import_partial",
 158                          help="Use this for big data importation, if it crashes you will be able to continue at the current state. Provide a filename to store intermediate importation states.", default=False)
 159         parser.add_option_group(group)
 160
 161         group = optparse.OptionGroup(parser, "Internationalisation options",
 162             "Use these options to translate OpenERP to another language."
 163             "See i18n section of the user manual. Option '-d' is mandatory."
 164             "Option '-l' is mandatory in case of importation"
 165             )
 166
 167         group.add_option('-l', "--language", dest="language",
 168                          help="specify the language of the translation file. Use it with --i18n-export or --i18n-import")
 169         group.add_option("--i18n-export", dest="translate_out",
 170                          help="export all sentences to be translated to a CSV file, a PO file or a TGZ archive and exit")
 171         group.add_option("--i18n-import", dest="translate_in",
 172                          help="import a CSV or a PO file with translations and exit. The '-l' option is required.")
 173         group.add_option("--modules", dest="translate_modules",
 174                          help="specify modules to export. Use in combination with --i18n-export")
 175         group.add_option("--addons-path", dest="addons_path",
 176                          help="specify an alternative addons path.",
 177                          action="callback", callback=self._check_addons_path, nargs=1, type="string")
 178         parser.add_option_group(group)
 179
 180         (opt, args) = parser.parse_args()
 181
 182         assert not (opt.translate_in and (not opt.language or not opt.db_name)), "the i18n-import option cannot be used without the language (-l) and the database (-d) options"
 183         assert not (opt.translate_out and (not opt.db_name)), "the i18n-export option cannot be used without the database (-d) option"
 184
 185         # place/search the config file on Win32 near the server installation
 186         # (../etc from the server)
 187         # if the server is run by an unprivileged user, he has to specify location of a config file where he has the rights to write,
 188         # else he won't be able to save the configurations, or even to start the server...
 189         if os.name == 'nt':
 190             rcfilepath = os.path.join(os.path.abspath(os.path.dirname(sys.argv[0])), 'openerp-server.conf')
 191         else:
 192             rcfilepath = os.path.expanduser('~/.openerp_serverrc')
 193
 194         self.rcfile = fname or opt.config or os.environ.get('OPENERP_SERVER') or rcfilepath
 195         self.load()
 196
 197
 198         # Verify that we want to log or not, if not the output will go to stdout
 199         if self.options['logfile'] in ('None', 'False'):
 200             self.options['logfile'] = False
 201         # the same for the pidfile
 202         if self.options['pidfile'] in ('None', 'False'):
 203             self.options['pidfile'] = False
 204
 205         keys = ['interface', 'port', 'db_name', 'db_user', 'db_password', 'db_host',
 206                 'db_port', 'logfile', 'pidfile', 'smtp_port', 'cache_timeout',
 207                 'email_from', 'smtp_server', 'smtp_user', 'smtp_password', 'price_accuracy',
 208                 'netinterface', 'netport', 'db_maxconn', 'import_partial', 'addons_path',
 209                 'netrpc', 'xmlrpc', 'syslog', 'without_demo']
 210
 211         if hasSSL:
 212             keys.extend(['smtp_ssl', 'secure_cert_file', 'secure_pkey_file'])
 213             keys.append('secure')
 214
 215         for arg in keys:
 216             if getattr(opt, arg):
 217                 self.options[arg] = getattr(opt, arg)
 218
 219         keys = ['language', 'translate_out', 'translate_in', 'debug_mode',
 220                 'stop_after_init']
 221
 222         for arg in keys:
 223             self.options[arg] = getattr(opt, arg)
 224
 225         if opt.assert_exit_level:
 226             self.options['assert_exit_level'] = loglevels[opt.assert_exit_level]
 227
 228         if opt.log_level:
 229             self.options['log_level'] = loglevels[opt.log_level]
 230
 231         if not self.options['root_path'] or self.options['root_path']=='None':
 232             self.options['root_path'] = os.path.abspath(os.path.dirname(sys.argv[0]))
 233         if not self.options['addons_path'] or self.options['addons_path']=='None':
 234             self.options['addons_path'] = os.path.join(self.options['root_path'], 'addons')
 235
 236         self.options['init'] = opt.init and dict.fromkeys(opt.init.split(','), 1) or {}
 237         self.options["demo"] = not opt.without_demo and self.options['init'] or {}
 238         self.options['update'] = opt.update and dict.fromkeys(opt.update.split(','), 1) or {}
 239
 240         self.options['translate_modules'] = opt.translate_modules and map(lambda m: m.strip(), opt.translate_modules.split(',')) or ['all']
 241         self.options['translate_modules'].sort()
 242
 243         if opt.pg_path:
 244             self.options['pg_path'] = opt.pg_path
 245
 246         if self.options.get('language', False):
 247             assert len(self.options['language'])<=5, 'ERROR: The Lang name must take max 5 chars, Eg: -lfr_BE'
 248         if opt.save:
 249             self.save()
 250
 251     def _check_addons_path(self, option, opt, value, parser):
 252         res = os.path.abspath(os.path.expanduser(value))
 253         if not os.path.exists(res):
 254             raise optparse.OptionValueError("option %s: no such directory: %r" % (opt, value))
 255         setattr(parser.values, option.dest, res)
 256
 257     def load(self):
 258         p = ConfigParser.ConfigParser()
 259         try:
 260             p.read([self.rcfile])
 261             for (name,value) in p.items('options'):
 262                 if value=='True' or value=='true':
 263                     value = True
 264                 if value=='False' or value=='false':
 265                     value = False
 266                 self.options[name] = value
 267         except IOError:
 268             pass
 269         except ConfigParser.NoSectionError:
 270             pass
 271
 272     def save(self):
 273         p = ConfigParser.ConfigParser()
 274         p.add_section('options')
 275         for o in [opt for opt in self.options.keys() if opt not in ('version','language','translate_out','translate_in','init','update')]:
 276             p.set('options', o, self.options[o])
 277
 278         # try to create the directories and write the file
 279         try:
 280             if not os.path.exists(os.path.dirname(self.rcfile)):
 281                 os.makedirs(os.path.dirname(self.rcfile))
 282             try:
 283                 p.write(file(self.rcfile, 'w'))
 284             except IOError:
 285                 sys.stderr.write("ERROR: couldn't write the config file\n")
 286
 287         except OSError:
 288             # what to do if impossible?
 289             sys.stderr.write("ERROR: couldn't create the config directory\n")
 290
 291     def get(self, key, default=None):
 292         return self.options.get(key, default)
 293
 294     def __setitem__(self, key, value):
 295         self.options[key] = value
 296
 297     def __getitem__(self, key):
 298         return self.options[key]
 299
 300 config = configmanager()
 301
 302
 303
 304 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 305