From 7bd44800cd13edfedd692df55230bc3b6ea8ad35 Mon Sep 17 00:00:00 2001 From: Deimos Date: Sun, 14 Apr 2019 21:35:32 -0600 Subject: [PATCH] Dev version: auto-subscribe TestUser to ~testing --- tildes/scripts/initialize_db.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/tildes/scripts/initialize_db.py b/tildes/scripts/initialize_db.py index d868207..06ea6da 100644 --- a/tildes/scripts/initialize_db.py +++ b/tildes/scripts/initialize_db.py @@ -15,7 +15,7 @@ from sqlalchemy.engine import Connectable, Engine from tildes.database_models import * # noqa from tildes.lib.database import get_session_from_config from tildes.models import DatabaseModel -from tildes.models.group import Group +from tildes.models.group import Group, GroupSubscription from tildes.models.log import Log from tildes.models.user import User @@ -77,13 +77,10 @@ def insert_dev_data(config_path: str) -> None: """Load the app config and insert some "starter" data for a dev version.""" session = get_session_from_config(config_path) - session.add_all( - [ - User("TestUser", "password"), - Group( - "testing", "An automatically created group to use for testing purposes" - ), - ] - ) + user = User("TestUser", "password") + group = Group("testing", "An automatically created group to use for testing") + subscription = GroupSubscription(user, group) + + session.add_all([user, group, subscription]) session.commit()