def database(): admin_db_config = load_config( BASE_DIR / 'config' / 'admin_config.toml')['database'] test_db_config = load_config( BASE_DIR / 'config' / 'test_config.toml')['database'] setup_db(executor_config=admin_db_config, target_config=test_db_config) yield teardown_db(executor_config=admin_db_config, target_config=test_db_config)
def tables_and_data(database): test_db_config = load_config( BASE_DIR / 'config' / 'test_config.toml')['database'] create_tables(target_config=test_db_config) create_sample_data(target_config=test_db_config) yield drop_tables(target_config=test_db_config)
def main(configpath): config = load_config(configpath) logging.basicConfig(level=logging.DEBUG) app = init_app(config) web.run_app(app)
async def client(aiohttp_client): config = load_config(BASE_DIR / 'config' / 'test_config.toml') app = await init_app(config) return await aiohttp_client(app)
'user_id': 1, 'body': 'Lovely day' }, { 'user_id': 2, 'body': 'Roses are red' }, { 'user_id': 2, 'body': 'Lorem ipsum' }, ]) if __name__ == '__main__': user_db_config = load_config('config/user_config.toml')['database'] admin_db_config = load_config('config/admin_config.toml')['database'] import argparse parser = argparse.ArgumentParser(description='DB related shortcuts') parser.add_argument("-c", "--create", help="Create empty database and user with permissions", action='store_true') parser.add_argument("-d", "--drop", help="Drop database and user role", action='store_true') parser.add_argument("-r", "--recreate", help="Drop and recreate database and user",