diff --git a/aerich/cli.py b/aerich/cli.py index 6e8a82e..63bc522 100644 --- a/aerich/cli.py +++ b/aerich/cli.py @@ -48,7 +48,11 @@ def coro(f): @click.group(context_settings={"help_option_names": ["-h", "--help"]}) @click.version_option(__version__, "-V", "--version") @click.option( - "-c", "--config", default="aerich.ini", show_default=True, help="Config file.", + "-c", + "--config", + default="aerich.ini", + show_default=True, + help="Config file.", ) @click.option("--app", required=False, help="Tortoise-ORM app name.") @click.option( @@ -121,7 +125,9 @@ async def upgrade(ctx: Context): for upgrade_query in upgrade_query_list: await conn.execute_script(upgrade_query) await Aerich.create( - version=version_file, app=app, content=get_models_describe(app), + version=version_file, + app=app, + content=get_models_describe(app), ) click.secho(f"Success upgrade {version_file}", fg=Color.green) migrated = True @@ -215,17 +221,21 @@ async def history(ctx: Context): help="Tortoise-ORM config module dict variable, like settings.TORTOISE_ORM.", ) @click.option( - "--location", default="./migrations", show_default=True, help="Migrate store location.", + "--location", + default="./migrations", + show_default=True, + help="Migrate store location.", ) @click.option( "-s", - "--src_folder", default=".", show_default=False, help="Folder of the source, relative to the project root." + "--src_folder", + default=".", + show_default=False, + help="Folder of the source, relative to the project root.", ) @click.pass_context @coro -async def init( - ctx: Context, tortoise_orm, location, src_folder -): +async def init(ctx: Context, tortoise_orm, location, src_folder): config_file = ctx.obj["config_file"] name = ctx.obj["name"] if Path(config_file).exists(): @@ -234,8 +244,8 @@ async def init( if os.path.isabs(src_folder): src_folder = os.path.relpath(os.getcwd(), src_folder) # Add ./ so it's clear that this is relative path - if not src_folder.startswith('./'): - src_folder = './' + src_folder + if not src_folder.startswith("./"): + src_folder = "./" + src_folder # check that we can find the configuration, if not we can fail before the config file gets created add_src_path(src_folder) @@ -287,7 +297,9 @@ async def init_db(ctx: Context, safe): version = await Migrate.generate_version() await Aerich.create( - version=version, app=app, content=get_models_describe(app), + version=version, + app=app, + content=get_models_describe(app), ) content = { "upgrade": [schema], @@ -298,7 +310,11 @@ async def init_db(ctx: Context, safe): @cli.command(help="Introspects the database tables to standard output as TortoiseORM model.") @click.option( - "-t", "--table", help="Which tables to inspect.", multiple=True, required=False, + "-t", + "--table", + help="Which tables to inspect.", + multiple=True, + required=False, ) @click.pass_context @coro diff --git a/aerich/utils.py b/aerich/utils.py index e289b77..25547f1 100644 --- a/aerich/utils.py +++ b/aerich/utils.py @@ -36,7 +36,8 @@ def get_app_connection_name(config, app_name: str) -> str: if app: return app.get("default_connection", "default") raise BadOptionUsage( - option_name="--app", message=f'Can\'t get app named "{app_name}"', + option_name="--app", + message=f'Can\'t get app named "{app_name}"', ) @@ -64,7 +65,7 @@ def get_tortoise_config(ctx: Context, tortoise_orm: str) -> dict: try: config_module = importlib.import_module(config_path) except ModuleNotFoundError as e: - raise ClickException(f'Error while importing configuration module: {e}') from None + raise ClickException(f"Error while importing configuration module: {e}") from None config = getattr(config_module, tortoise_config, None) if not config: