1 files changed, 103 insertions, 0 deletions
diff --git a/openerpcommand/initialize.py b/openerpcommand/initialize.py
new file mode 100644
index 00000000000..3393c9a02d4
--- /dev/null
+++ b/openerpcommand/initialize.py
@@ -0,0 +1,103 @@
+"""
+Install OpenERP on a new (by default) database.
+"""
+import os
+import sys
+
+import common
+
+def install_openerp(database_name, create_database_flag, module_names, install_demo_data):
+    import openerp
+    config = openerp.tools.config
+
+    if create_database_flag:
+        create_database(database_name)
+
+    config['init'] = dict.fromkeys(module_names, 1)
+
+    # Install the import hook, to import openerp.addons.<module>.
+    openerp.modules.module.initialize_sys_path()
+    if hasattr(openerp.modules.loading, 'open_openerp_namespace'):
+        openerp.modules.loading.open_openerp_namespace()
+
+    registry = openerp.modules.registry.RegistryManager.get(
+        database_name, update_module=True, force_demo=install_demo_data)
+
+    return registry
+
+# TODO turn template1 in a parameter
+# This should be exposed from openerp (currently in
+# openerp/service/web_services.py).
+def create_database(database_name):
+    import openerp
+    db = openerp.sql_db.db_connect('template1')
+    cr = db.cursor() # TODO `with db as cr:`
+    try:
+        cr.autocommit(True)
+        cr.execute("""CREATE DATABASE "%s"
+            ENCODING 'unicode' TEMPLATE "template1" """ \
+            % (database_name,))
+    finally:
+        cr.close()
+
+def run(args):
+    assert args.database
+    assert not (args.module and args.all_modules)
+
+    import openerp
+
+    config = openerp.tools.config
+
+    if args.tests:
+        config['log_handler'] = [':TEST']
+        config['test_enable'] = True
+        config['without_demo'] = False
+    else:
+        config['log_handler'] = [':CRITICAL']
+        config['test_enable'] = False
+        config['without_demo'] = True
+
+    if args.addons:
+        args.addons = args.addons.split(':')
+    else:
+        args.addons = []
+    config['addons_path'] = ','.join(args.addons)
+
+    if args.all_modules:
+        module_names = common.get_addons_from_paths(args.addons, args.exclude)
+    elif args.module:
+        module_names = args.module
+    else:
+        module_names = ['base']
+
+    openerp.netsvc.init_logger()
+    registry = install_openerp(args.database, not args.no_create, module_names, not config['without_demo'])
+
+    # The `_assertion_report` attribute was added on the registry during the
+    # OpenERP 7.0 development.
+    if hasattr(registry, '_assertion_report'):
+        sys.exit(1 if registry._assertion_report.failures else 0)
+
+def add_parser(subparsers):
+    parser = subparsers.add_parser('initialize',
+        description='Create and initialize a new OpenERP database.')
+    parser.add_argument('-d', '--database', metavar='DATABASE',
+        **common.required_or_default('DATABASE', 'the database to create'))
+    common.add_addons_argument(parser)
+    parser.add_argument('--module', metavar='MODULE', action='append',
+        help='specify a module to install'
+        ' (this option can be repeated)')
+    parser.add_argument('--all-modules', action='store_true',
+        help='install all visible modules (not compatible with --module)')
+    parser.add_argument('--no-create', action='store_true',
+        help='do not create the database, only initialize it')
+    parser.add_argument('--exclude', metavar='MODULE', action='append',
+        help='exclude a module from installation'
+        ' (this option can be repeated)')
+    parser.add_argument('--tests', action='store_true',
+        help='run the tests as modules are installed'
+        ' (use the `run-tests` command to choose specific'
+        ' tests to run against an existing database).'
+        ' Demo data are installed.')
+
+    parser.set_defaults(run=run)