bin/tools/config.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21
  22 import ConfigParser
  23 import optparse
  24 import os
  25 import sys
  26 import netsvc
  27 import logging
  28 import release
  29
  30 def check_ssl():
  31     try:
  32         from OpenSSL import SSL
  33         import socket
  34
  35         return hasattr(socket, 'ssl')
  36     except:
  37         return False
  38
  39 class configmanager(object):
  40     def __init__(self, fname=None):
  41         self.options = {
  42             'email_from':False,
  43             'xmlrpc_interface': '',    # this will bind the server to all interfaces
  44             'xmlrpc_port': 8069,
  45             'netrpc_interface': '',
  46             'netrpc_port': 8070,
  47             'xmlrpcs_interface': '',    # this will bind the server to all interfaces
  48             'xmlrpcs_port': 8071,
  49             'db_host': False,
  50             'db_port': False,
  51             'db_name': False,
  52             'db_user': False,
  53             'db_password': False,
  54             'db_maxconn': 64,
  55             'reportgz': False,
  56             'netrpc': True,
  57             'xmlrpc': True,
  58             'xmlrpcs': True,
  59             'translate_in': None,
  60             'translate_out': None,
  61             'language': None,
  62             'pg_path': None,
  63             'admin_passwd': 'admin',
  64             'csv_internal_sep': ',',
  65             'addons_path': None,
  66             'root_path': None,
  67             'debug_mode': False,
  68             'import_partial': "",
  69             'pidfile': None,
  70             'logfile': None,
  71             'logrotate': '1',
  72             'smtp_server': 'localhost',
  73             'smtp_user': False,
  74             'smtp_port':25,
  75             'smtp_ssl':False,
  76             'smtp_password': False,
  77             'stop_after_init': False,   # this will stop the server after initialization
  78             'price_accuracy': 2,
  79             'syslog' : False,
  80             'log_level': logging.INFO,
  81             'assert_exit_level': logging.ERROR, # level above which a failed assert will be raised
  82             'cache_timeout': 100000,
  83             'login_message': False,
  84             'list_db' : True,
  85             'timezone' : False, # to override the default TZ
  86             'test-disable' : False,
  87             'test-commit' : False,
  88         }
  89
  90         self.misc = {}
  91         self.config_file = fname
  92         self.has_ssl = check_ssl()
  93
  94         self._LOGLEVELS = dict([(getattr(netsvc, 'LOG_%s' % x), getattr(logging, x))
  95                           for x in ('CRITICAL', 'ERROR', 'WARNING', 'INFO', 'TEST', 'DEBUG', 'DEBUG_RPC', 'NOTSET')])
  96
  97         version = "%s %s" % (release.description, release.version)
  98         self.parser = parser = optparse.OptionParser(version=version)
  99
 100         parser.add_option("-c", "--config", dest="config", help="specify alternate config file")
 101         parser.add_option("-s", "--save", action="store_true", dest="save", default=False,
 102                           help="save configuration to ~/.openerp_serverrc")
 103         parser.add_option("--pidfile", dest="pidfile", help="file where the server pid will be stored")
 104
 105         group = optparse.OptionGroup(parser, "XML-RPC Configuration")
 106         group.add_option("--xmlrpc-interface", dest="xmlrpc_interface", help="specify the TCP IP address for the XML-RPC protocol")
 107         group.add_option("--xmlrpc-port", dest="xmlrpc_port", help="specify the TCP port for the XML-RPC protocol", type="int")
 108         group.add_option("--no-xmlrpc", dest="xmlrpc", action="store_false", help="disable the XML-RPC protocol")
 109         parser.add_option_group(group)
 110
 111         if self.has_ssl:
 112             group = optparse.OptionGroup(parser, "XML-RPC Secure Configuration")
 113             group.add_option("--xmlrpcs-interface", dest="xmlrpcs_interface", help="specify the TCP IP address for the XML-RPC Secure protocol")
 114             group.add_option("--xmlrpcs-port", dest="xmlrpcs_port", help="specify the TCP port for the XML-RPC Secure protocol", type="int")
 115             group.add_option("--no-xmlrpcs", dest="xmlrpcs", action="store_false", help="disable the XML-RPC Secure protocol")
 116             group.add_option("--cert-file", dest="secure_cert_file", default="server.cert", help="specify the certificate file for the SSL connection")
 117             group.add_option("--pkey-file", dest="secure_pkey_file", default="server.pkey", help="specify the private key file for the SSL connection")
 118             parser.add_option_group(group)
 119
 120         group = optparse.OptionGroup(parser, "NET-RPC Configuration")
 121         group.add_option("--netrpc-interface", dest="netrpc_interface", help="specify the TCP IP address for the NETRPC protocol")
 122         group.add_option("--netrpc-port", dest="netrpc_port", help="specify the TCP port for the NETRPC protocol", type="int")
 123         group.add_option("--no-netrpc", dest="netrpc", action="store_false", help="disable the NETRPC protocol")
 124         parser.add_option_group(group)
 125
 126         parser.add_option("-i", "--init", dest="init", help="init a module (use \"all\" for all modules)")
 127         parser.add_option("--without-demo", dest="without_demo",
 128                           help="load demo data for a module (use \"all\" for all modules)", default=False)
 129         parser.add_option("-u", "--update", dest="update",
 130                           help="update a module (use \"all\" for all modules)")
 131         parser.add_option("--cache-timeout", dest="cache_timeout",
 132                           help="set the timeout for the cache system", default=100000, type="int")
 133         parser.add_option("-t", "--timezone", dest="timezone", help="specify reference timezone for the server (e.g. Europe/Brussels")
 134
 135         # stops the server from launching after initialization
 136         parser.add_option("--stop-after-init", action="store_true", dest="stop_after_init", default=False,
 137                           help="stop the server after it initializes")
 138         parser.add_option('--debug', dest='debug_mode', action='store_true', default=False, help='enable debug mode')
 139         parser.add_option("--assert-exit-level", dest='assert_exit_level', type="choice", choices=self._LOGLEVELS.keys(),
 140                           help="specify the level at which a failed assertion will stop the server. Accepted values: %s" % (self._LOGLEVELS.keys(),))
 141         parser.add_option('--price_accuracy', dest='price_accuracy', default='2', help='deprecated since v6.0, replaced by module decimal_precision')
 142
 143         # Testing Group
 144         group = optparse.OptionGroup(parser, "Testing Configuration")
 145         group.add_option("--test-disable", action="store_true", dest="test_disable",
 146                          default=False, help="Disable loading test files.")
 147         group.add_option("--test-commit", action="store_true", dest="test_commit",
 148                          default=False, help="Commit database changes performed by tests.")
 149         parser.add_option_group(group)
 150
 151         # Logging Group
 152         group = optparse.OptionGroup(parser, "Logging Configuration")
 153         group.add_option("--logfile", dest="logfile", help="file where the server log will be stored")
 154         group.add_option("--no-logrotate", dest="logrotate", action="store_false",
 155                          default=None, help="do not rotate the logfile")
 156         group.add_option("--syslog", action="store_true", dest="syslog",
 157                          default=False, help="Send the log to the syslog server")
 158         group.add_option('--log-level', dest='log_level', type='choice', choices=self._LOGLEVELS.keys(),
 159                          help='specify the level of the logging. Accepted values: ' + str(self._LOGLEVELS.keys()))
 160         parser.add_option_group(group)
 161
 162         # SMTP Group
 163         group = optparse.OptionGroup(parser, "SMTP Configuration")
 164         group.add_option('--email-from', dest='email_from', default='', help='specify the SMTP email address for sending email')
 165         group.add_option('--smtp', dest='smtp_server', default='', help='specify the SMTP server for sending email')
 166         group.add_option('--smtp-port', dest='smtp_port', default='25', help='specify the SMTP port', type="int")
 167         group.add_option('--smtp-ssl', dest='smtp_ssl', default='', help='specify the SMTP server support SSL or not')
 168         group.add_option('--smtp-user', dest='smtp_user', default='', help='specify the SMTP username for sending email')
 169         group.add_option('--smtp-password', dest='smtp_password', default='', help='specify the SMTP password for sending email')
 170         parser.add_option_group(group)
 171
 172         group = optparse.OptionGroup(parser, "Database related options")
 173         group.add_option("-d", "--database", dest="db_name", help="specify the database name")
 174         group.add_option("-r", "--db_user", dest="db_user", help="specify the database user name")
 175         group.add_option("-w", "--db_password", dest="db_password", help="specify the database password")
 176         group.add_option("--pg_path", dest="pg_path", help="specify the pg executable path")
 177         group.add_option("--db_host", dest="db_host", help="specify the database host")
 178         group.add_option("--db_port", dest="db_port", help="specify the database port", type="int")
 179         group.add_option("--db_maxconn", dest="db_maxconn", default='64',
 180                          help="specify the the maximum number of physical connections to posgresql")
 181         group.add_option("-P", "--import-partial", dest="import_partial",
 182                          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)
 183         parser.add_option_group(group)
 184
 185         group = optparse.OptionGroup(parser, "Internationalisation options",
 186             "Use these options to translate OpenERP to another language."
 187             "See i18n section of the user manual. Option '-d' is mandatory."
 188             "Option '-l' is mandatory in case of importation"
 189             )
 190
 191         group.add_option('-l', "--language", dest="language",
 192                          help="specify the language of the translation file. Use it with --i18n-export or --i18n-import")
 193         group.add_option("--i18n-export", dest="translate_out",
 194                          help="export all sentences to be translated to a CSV file, a PO file or a TGZ archive and exit")
 195         group.add_option("--i18n-import", dest="translate_in",
 196                          help="import a CSV or a PO file with translations and exit. The '-l' option is required.")
 197         group.add_option("--modules", dest="translate_modules",
 198                          help="specify modules to export. Use in combination with --i18n-export")
 199         group.add_option("--addons-path", dest="addons_path",
 200                          help="specify an alternative addons path.",
 201                          action="callback", callback=self._check_addons_path, nargs=1, type="string")
 202         parser.add_option_group(group)
 203
 204         security = optparse.OptionGroup(parser, 'Security-related options')
 205         security.add_option('--no-database-list', action="store_false", dest='list_db', default=True, help="disable the ability to return the list of databases")
 206         security.add_option('--enable-code-actions', action='store_true',
 207                             dest='server_actions_allow_code', default=False,
 208                             help='Enables server actions of state "code". Warning, this is a security risk.')
 209         parser.add_option_group(security)
 210
 211     def parse_config(self):
 212         (opt, args) = self.parser.parse_args()
 213
 214         def die(cond, msg):
 215             if cond:
 216                 print msg
 217                 sys.exit(1)
 218
 219         die(bool(opt.syslog) and bool(opt.logfile),
 220             "the syslog and logfile options are exclusive")
 221
 222         die(opt.translate_in and (not opt.language or not opt.db_name),
 223             "the i18n-import option cannot be used without the language (-l) and the database (-d) options")
 224
 225         die(opt.translate_out and (not opt.db_name),
 226             "the i18n-export option cannot be used without the database (-d) option")
 227
 228         # Check if the config file exists (-c used, but not -s)
 229         die(not opt.save and opt.config and not os.path.exists(opt.config),
 230             "The config file '%s' selected with -c/--config doesn't exist, "\
 231             "use -s/--save if you want to generate it"%(opt.config))
 232
 233         # place/search the config file on Win32 near the server installation
 234         # (../etc from the server)
 235         # 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,
 236         # else he won't be able to save the configurations, or even to start the server...
 237         if os.name == 'nt':
 238             rcfilepath = os.path.join(os.path.abspath(os.path.dirname(sys.argv[0])), 'openerp-server.conf')
 239         else:
 240             rcfilepath = os.path.expanduser('~/.openerp_serverrc')
 241
 242         self.rcfile = os.path.abspath(
 243             self.config_file or opt.config \
 244                 or os.environ.get('OPENERP_SERVER') or rcfilepath)
 245         self.load()
 246
 247
 248         # Verify that we want to log or not, if not the output will go to stdout
 249         if self.options['logfile'] in ('None', 'False'):
 250             self.options['logfile'] = False
 251         # the same for the pidfile
 252         if self.options['pidfile'] in ('None', 'False'):
 253             self.options['pidfile'] = False
 254
 255         keys = ['xmlrpc_interface', 'xmlrpc_port', 'db_name', 'db_user', 'db_password', 'db_host',
 256                 'db_port', 'list_db', 'logfile', 'pidfile', 'smtp_port', 'cache_timeout','smtp_ssl',
 257                 'email_from', 'smtp_server', 'smtp_user', 'smtp_password', 'price_accuracy',
 258                 'netrpc_interface', 'netrpc_port', 'db_maxconn', 'import_partial', 'addons_path',
 259                 'netrpc', 'xmlrpc', 'syslog', 'without_demo', 'timezone',]
 260
 261         if self.has_ssl:
 262             keys.extend([
 263                 'xmlrpcs_interface',
 264                 'xmlrpcs_port',
 265                 'xmlrpcs',
 266                 'secure_cert_file',
 267                 'secure_pkey_file']
 268             )
 269
 270         for arg in keys:
 271             if getattr(opt, arg):
 272                 self.options[arg] = getattr(opt, arg)
 273
 274         keys = ['language', 'translate_out', 'translate_in', 'debug_mode',
 275                 'stop_after_init', 'logrotate', 'without_demo', 'netrpc', 'xmlrpc', 'xmlrpcs', 'syslog',
 276                 'list_db', 'server_actions_allow_code']
 277
 278         for arg in keys:
 279             if getattr(opt, arg) is not None:
 280                 self.options[arg] = getattr(opt, arg)
 281
 282         if opt.assert_exit_level:
 283             self.options['assert_exit_level'] = self._LOGLEVELS[opt.assert_exit_level]
 284         else:
 285             self.options['assert_exit_level'] = self._LOGLEVELS.get(self.options['assert_exit_level']) or int(self.options['assert_exit_level'])
 286
 287         if opt.log_level:
 288             self.options['log_level'] = self._LOGLEVELS[opt.log_level]
 289         else:
 290             self.options['log_level'] = self._LOGLEVELS.get(self.options['log_level']) or int(self.options['log_level'])
 291
 292         if not self.options['root_path'] or self.options['root_path']=='None':
 293             self.options['root_path'] = os.path.abspath(os.path.dirname(sys.argv[0]))
 294         if not self.options['addons_path'] or self.options['addons_path']=='None':
 295             self.options['addons_path'] = os.path.join(self.options['root_path'], 'addons')
 296
 297         self.options['init'] = opt.init and dict.fromkeys(opt.init.split(','), 1) or {}
 298         self.options["demo"] = not opt.without_demo and self.options['init'] or {}
 299         self.options["test-disable"] =  opt.test_disable
 300         self.options["test-commit"] =  opt.test_commit
 301         self.options['update'] = opt.update and dict.fromkeys(opt.update.split(','), 1) or {}
 302
 303         self.options['translate_modules'] = opt.translate_modules and map(lambda m: m.strip(), opt.translate_modules.split(',')) or ['all']
 304         self.options['translate_modules'].sort()
 305
 306         if self.options['timezone']:
 307             # If an explicit TZ was provided in the config, make sure it is known
 308             try:
 309                 import pytz
 310                 tz = pytz.timezone(self.options['timezone'])
 311             except pytz.UnknownTimeZoneError:
 312                 die(True, "The specified timezone (%s) is invalid" % self.options['timezone'])
 313             except:
 314                 # If pytz is missing, don't check the provided TZ, it will be ignored anyway.
 315                 pass
 316
 317         if opt.pg_path:
 318             self.options['pg_path'] = opt.pg_path
 319
 320         if self.options.get('language', False):
 321             if len(self.options['language']) > 5:
 322                 raise Exception('ERROR: The Lang name must take max 5 chars, Eg: -lfr_BE')
 323
 324         if not self.options['db_user']:
 325             try:
 326                 import getpass
 327                 self.options['db_user'] = getpass.getuser()
 328             except:
 329                 self.options['db_user'] = None
 330
 331         die(not self.options['db_user'], 'ERROR: No user specified for the connection to the database')
 332
 333         if self.options['db_password']:
 334             if sys.platform == 'win32' and not self.options['db_host']:
 335                 self.options['db_host'] = 'localhost'
 336             #if self.options['db_host']:
 337             #    self._generate_pgpassfile()
 338
 339         if opt.save:
 340             self.save()
 341
 342     def _generate_pgpassfile(self):
 343         """
 344         Generate the pgpass file with the parameters from the command line (db_host, db_user,
 345         db_password)
 346
 347         Used because pg_dump and pg_restore can not accept the password on the command line.
 348         """
 349         is_win32 = sys.platform == 'win32'
 350         if is_win32:
 351             filename = os.path.join(os.environ['APPDATA'], 'pgpass.conf')
 352         else:
 353             filename = os.path.join(os.environ['HOME'], '.pgpass')
 354
 355         text_to_add = "%(db_host)s:*:*:%(db_user)s:%(db_password)s" % self.options
 356
 357         if os.path.exists(filename):
 358             content = [x.strip() for x in file(filename, 'r').readlines()]
 359             if text_to_add in content:
 360                 return
 361
 362         fp = file(filename, 'a+')
 363         fp.write(text_to_add + "\n")
 364         fp.close()
 365
 366         if is_win32:
 367             import _winreg
 368             x=_winreg.ConnectRegistry(None,_winreg.HKEY_LOCAL_MACHINE)
 369             y = _winreg.OpenKey(x, r"SYSTEM\CurrentControlSet\Control\Session Manager\Environment", 0,_winreg.KEY_ALL_ACCESS)
 370             _winreg.SetValueEx(y,"PGPASSFILE", 0, _winreg.REG_EXPAND_SZ, filename )
 371             _winreg.CloseKey(y)
 372             _winreg.CloseKey(x)
 373         else:
 374             import stat
 375             os.chmod(filename, stat.S_IRUSR + stat.S_IWUSR)
 376
 377     def _check_addons_path(self, option, opt, value, parser):
 378         res = os.path.abspath(os.path.expanduser(value))
 379         if not os.path.exists(res):
 380             raise optparse.OptionValueError("option %s: no such directory: %r" % (opt, value))
 381
 382         contains_addons = False
 383         for f in os.listdir(res):
 384             modpath = os.path.join(res, f)
 385             if os.path.isdir(modpath) and \
 386                os.path.exists(os.path.join(modpath, '__init__.py')) and \
 387                (os.path.exists(os.path.join(modpath, '__openerp__.py')) or \
 388                 os.path.exists(os.path.join(modpath, '__terp__.py'))):
 389
 390                 contains_addons = True
 391                 break
 392
 393         if not contains_addons:
 394             raise optparse.OptionValueError("option %s: The addons-path %r does not seem to a be a valid Addons Directory!" % (opt, value))
 395
 396         setattr(parser.values, option.dest, res)
 397
 398     def load(self):
 399         p = ConfigParser.ConfigParser()
 400         try:
 401             p.read([self.rcfile])
 402             for (name,value) in p.items('options'):
 403                 if value=='True' or value=='true':
 404                     value = True
 405                 if value=='False' or value=='false':
 406                     value = False
 407                 self.options[name] = value
 408             #parse the other sections, as well
 409             for sec in p.sections():
 410                 if sec == 'options':
 411                     continue
 412                 if not self.misc.has_key(sec):
 413                     self.misc[sec]= {}
 414                 for (name, value) in p.items(sec):
 415                     if value=='True' or value=='true':
 416                         value = True
 417                     if value=='False' or value=='false':
 418                         value = False
 419                     self.misc[sec][name] = value
 420         except IOError:
 421             pass
 422         except ConfigParser.NoSectionError:
 423             pass
 424
 425     def save(self):
 426         p = ConfigParser.ConfigParser()
 427         loglevelnames = dict(zip(self._LOGLEVELS.values(), self._LOGLEVELS.keys()))
 428         p.add_section('options')
 429         for opt in self.options.keys():
 430             if opt in ('version', 'language', 'translate_out', 'translate_in', 'init', 'update'):
 431                 continue
 432             if opt in ('log_level', 'assert_exit_level'):
 433                 p.set('options', opt, loglevelnames.get(self.options[opt], self.options[opt]))
 434             else:
 435                 p.set('options', opt, self.options[opt])
 436
 437         for sec in self.misc.keys():
 438             for opt in self.misc[sec].keys():
 439                 p.set(sec,opt,self.misc[sec][opt])
 440
 441         # try to create the directories and write the file
 442         try:
 443             if not os.path.exists(os.path.dirname(self.rcfile)):
 444                 os.makedirs(os.path.dirname(self.rcfile))
 445             try:
 446                 p.write(file(self.rcfile, 'w'))
 447                 os.chmod(self.rcfile, 0600)
 448             except IOError:
 449                 sys.stderr.write("ERROR: couldn't write the config file\n")
 450
 451         except OSError:
 452             # what to do if impossible?
 453             sys.stderr.write("ERROR: couldn't create the config directory\n")
 454
 455     def get(self, key, default=None):
 456         return self.options.get(key, default)
 457
 458     def get_misc(self, sect, key, default=None):
 459         return self.misc.get(sect,{}).get(key, default)
 460
 461     def __setitem__(self, key, value):
 462         self.options[key] = value
 463
 464     def __getitem__(self, key):
 465         return self.options[key]
 466
 467 config = configmanager()
 468
 469 # FIXME:following line should be called explicitly by the server
 470 # when it starts, to allow doing 'import tools.config' from
 471 # other python executables without parsing *their* args.
 472 config.parse_config()
 473
 474 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 475