def do_importgroup(args, dbh, settings):
cerr('Importing group')
groups = yaml.load(open(args.importgroup))
from rhombus.models.user import UserClass, Group
Group.bulk_insert(groups, dbsession=dbh.session())
def setup( dbh ):
""" populate the database with basic, essential data """
if get_datalogger():
get_clsreg().sync()
session.commit()
dbsession = dbh.session()
EK.bulk_update( ek_initlist, dbsession=dbsession )
Group.bulk_insert( essential_groups, dbsession=dbsession )
UserClass.bulk_insert( system_userclass, dbsession=dbsession )
group_id = Group._id('_SysAdm_', dbsession)
file = File( path='/', group_id = group_id, permanent = True )
dbsession.add( file )
file.type = 'file/folder'
file.mimetype = 'application/x-directory'
cerr('INFO: root password is %s\n' % root_password)
def setup_db( *ops ):
""" setup the database (create tables, etc) and populate with basic data """
base = get_base()
# WARN: use alembic to create initial tables
#base.metadata.create_all()
with transaction.manager:
if get_datalogger():
get_clsreg().sync()
with transaction.manager:
EK.bulk_insert( ek_initlist )
Group.bulk_insert( essential_groups )
file = File( path='/', type='file/folder', mimetype='application/x-directory',
group_id = Group._id('_SysAdm_'), permanent = True )
dbsession.add( file )
for op in ops:
op()
def populate_db( dbsession, initial_groups=None, initial_userclass=None ):
if initial_groups:
Group.bulk_insert( initial_groups, dbsession=dbsession )
if initial_userclass:
UserClass.bulk_insert( initial_userclass, dbsession=dbsession )
