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",