def setup_user():
users = [
("*****@*****.**", "test", "admin"),
("*****@*****.**", "test", "user"),
("*****@*****.**", "test", "guest"),
]
groups = {}
for username, password, group_name in users:
user = UserModel(username=username)
user.password = password
DBSession.add(user)
if group_name not in groups:
group = GroupModel(group_name=group_name)
DBSession.add(group)
groups[group_name] = group
user.groups.append(groups[group_name])
def main(argv=sys.argv):
if len(argv) != 2:
usage(argv)
config_uri = argv[1]
setup_logging(config_uri)
settings = get_appsettings(config_uri)
engine = engine_from_config(settings, 'sqlalchemy.')
DBSession.configure(bind=engine)
BaseModel.metadata.create_all(engine)
with transaction.manager:
if DBSession.query(NewsModel).count() == 0:
setup_news()
if DBSession.query(UserModel).count() == 0 and \
DBSession.query(GroupModel).count() == 0:
setup_user()
def setup_news():
entity = NewsModel(title=u"Hello, world!", body=u"This is a news.")
DBSession.add(entity)