add command to delete database, usefull for testing

doc/cli.md

@@ -16,7 +16,17 @@ and active the Tracim virtualenv:
 
 ### Create database
 
-    tracim db init
+    tracimcli db init
+
+### Create database with some default test data (many users, workspaces, etc…)
+
+    tracimcli db init --test-data
+
+### Delete database /!\
+
+This will drop all your database, be carefull !
+
+    tracimcli db delete --force
 
 ## User ##
    

setup.py

@@ -98,7 +98,8 @@ setup(
             'test = tracim.command:TestTracimCommand',
             'user_create = tracim.command.user:CreateUserCommand',
             'user_update = tracim.command.user:UpdateUserCommand',
-            'db_init = tracim.command.initializedb:InitializeDBCommand',
+            'db_init = tracim.command.database:InitializeDBCommand',
+            'db_delete = tracim.command.database:DeleteDBCommand',
             'webdav start = tracim.command.webdav:WebdavRunnerCommand',
         ]
     },

tracim/command/initializedb.py → tracim/command/database.py

@@ -11,6 +11,7 @@ from pyramid.paster import (
 from tracim import CFG
 from tracim.fixtures import FixturesLoader
 from tracim.fixtures.users_and_groups import Base as BaseFixture
+from tracim.fixtures.content import Content as ContentFixture
 from sqlalchemy.exc import IntegrityError
 from tracim.command import AppContextCommand
 from tracim.models.meta import DeclarativeBase
@@ -25,10 +26,18 @@ class InitializeDBCommand(AppContextCommand):
     auto_setup_context = False
 
     def get_description(self) -> str:
-        return "Initialize DB"
+        return "Initialize database"
 
     def get_parser(self, prog_name: str) -> argparse.ArgumentParser:
         parser = super().get_parser(prog_name)
+        parser.add_argument(
+            "--test-data",
+            help='Add some default data to database to make test',
+            dest='test_data',
+            required=False,
+            action='store_true',
+            default=False,
+        )
         return parser
 
     def take_action(self, parsed_args: argparse.Namespace) -> None:
@@ -37,7 +46,7 @@ class InitializeDBCommand(AppContextCommand):
         setup_logging(config_uri)
         settings = get_appsettings(config_uri)
         self._create_schema(settings)
-        self._populate_database(settings)
+        self._populate_database(settings, add_test_data=parsed_args.test_data)
 
     @classmethod
     def _create_schema(
@@ -51,7 +60,8 @@ class InitializeDBCommand(AppContextCommand):
     @classmethod
     def _populate_database(
             cls,
-            settings: plaster_pastedeploy.ConfigDict
+            settings: plaster_pastedeploy.ConfigDict,
+            add_test_data: bool
     ) -> None:
         engine = get_engine(settings)
         session_factory = get_session_factory(engine)
@@ -60,8 +70,14 @@ class InitializeDBCommand(AppContextCommand):
         with transaction.manager:
             dbsession = get_tm_session(session_factory, transaction.manager)
             try:
+                fixtures = [BaseFixture]
                 fixtures_loader = FixturesLoader(dbsession, app_config)
-                fixtures_loader.loads([BaseFixture])
+                fixtures_loader.loads(fixtures)
+                transaction.commit()
+                if add_test_data:
+                    app_config.configure_filedepot()
+                    fixtures = [ContentFixture]
+                    fixtures_loader.loads(fixtures)
                 transaction.commit()
                 print("Database initialized.")
             except IntegrityError:
@@ -71,3 +87,36 @@ class InitializeDBCommand(AppContextCommand):
                 print(traceback.format_exc())
                 transaction.abort()
                 print('Database initialization failed')
+
+
+class DeleteDBCommand(AppContextCommand):
+    auto_setup_context = False
+
+    def get_description(self) -> str:
+        return "Delete database"
+
+    def get_parser(self, prog_name: str) -> argparse.ArgumentParser:
+        parser = super().get_parser(prog_name)
+        parser.add_argument(
+            "--force",
+            help='force delete of database',
+            dest='force',
+            required=False,
+            action='store_true',
+            default=False,
+        )
+        return parser
+
+    def take_action(self, parsed_args: argparse.Namespace) -> None:
+        super(DeleteDBCommand, self).take_action(parsed_args)
+        config_uri = parsed_args.config_file
+        setup_logging(config_uri)
+        settings = get_appsettings(config_uri)
+        engine = get_engine(settings)
+        if parsed_args.force:
+            print('Database deletion begin')
+            DeclarativeBase.metadata.drop_all(engine)
+            print('Database deletion done')
+        else:
+            print('Warning, You should use --force if you really want to'
+                  ' delete database.')

tracim/tests/__init__.py

@@ -5,7 +5,7 @@ from depot.manager import DepotManager
 from pyramid import testing
 from sqlalchemy.exc import IntegrityError
 
-from tracim.command.initializedb import InitializeDBCommand
+from tracim.command.database import InitializeDBCommand
 from tracim.lib.core.content import ContentApi
 from tracim.lib.core.workspace import WorkspaceApi
 from tracim.models import get_engine, DeclarativeBase, get_session_factory, \