コード例 #1
0
ファイル: __init__.py プロジェクト: wwitzel3/stockpot
    def setUp(self):
        connection = self.engine.connect()

        # begin a non-ORM transaction
        self.trans = connection.begin()

        # bind an individual Session to the connection
        DBSession.configure(bind=connection)
        self.session = self.Session(bind=connection)
        Base.session = self.session
コード例 #2
0
ファイル: initializedb.py プロジェクト: wwitzel3/stockpot
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)
    Base.metadata.create_all(engine)
    with transaction.manager:
        user = User(email='*****@*****.**', password='******', username='******')
        for ace in SITE_ACL:
            DBSession.add(Group(ace[1]))
        DBSession.flush()
        group = DBSession.query(Group).filter_by(name='role:admin').one()
        user.groups.add(group)
        DBSession.add(user)