schema wipe --force argument (#221)

* `schema wipe --force` argument

* Pretty warnings

Author: droserasprout

CHANGELOG.md

@@ -4,8 +4,13 @@ Please use [this](https://docs.gitlab.com/ee/development/changelog.html) documen
 
 ## [unreleased]
 
+### Added
+
+* cli: Added `schema wipe --force` argument to skip confirmation promt.
+
 ### Fixed
 
+* cli: Show warning about deprecated `--hashes` argument
 * cli: Ignore `SIGINT` signal when shutdown is in progress.
 * sentry: Ignore exceptions when shutdown is in progress.
 

src/dipdup/__init__.py

@@ -1,5 +1,7 @@
+import logging
 import os
 import sys
+import warnings
 
 __version__ = '4.1.1'
 __spec_version__ = '1.2'
@@ -17,3 +19,6 @@
 }
 
 sys.path.append(os.getcwd())
+logging.captureWarnings(True)
+warnings.simplefilter('always', DeprecationWarning)
+warnings.formatwarning = lambda msg, *a, **kw: str(msg)

src/dipdup/cli.py

@@ -5,6 +5,7 @@
 import signal
 import subprocess
 import sys
+import warnings
 from contextlib import AsyncExitStack
 from contextlib import suppress
 from dataclasses import dataclass
@@ -350,6 +351,9 @@ async def schema(ctx):
 @click.pass_context
 @cli_wrapper
 async def schema_approve(ctx, hashes: bool):
+    if hashes:
+        warnings.warn('`--hashes` option is deprecated and has no effect', DeprecationWarning)
+
     config: DipDupConfig = ctx.obj.config
     url = config.database.connection_string
     models = f'{config.package}.models'
@@ -371,22 +375,24 @@ async def schema_approve(ctx, hashes: bool):
 
 @schema.command(name='wipe', help='Drop all database tables, functions and views')
 @click.option('--immune', is_flag=True, help='Drop immune tables too')
+@click.option('--force', is_flag=True, help='Skip confirmation prompt')
 @click.pass_context
 @cli_wrapper
-async def schema_wipe(ctx, immune: bool):
+async def schema_wipe(ctx, immune: bool, force: bool):
     config: DipDupConfig = ctx.obj.config
     url = config.database.connection_string
     models = f'{config.package}.models'
 
-    try:
-        assert sys.__stdin__.isatty()
-        click.confirm(f'You\'re about to wipe schema `{url}`. All indexed data will be irreversibly lost, are you sure?', abort=True)
-    except AssertionError:
-        click.echo('Not in a TTY, skipping confirmation')
-    # FIXME: Can't catch asyncio.CancelledError here
-    except click.Abort:
-        click.echo('Aborted')
-        return
+    if not force:
+        try:
+            assert sys.__stdin__.isatty()
+            click.confirm(f'You\'re about to wipe schema `{url}`. All indexed data will be irreversibly lost, are you sure?', abort=True)
+        except AssertionError:
+            click.echo('Not in a TTY, skipping confirmation')
+        # FIXME: Can't catch asyncio.CancelledError here
+        except click.Abort:
+            click.echo('Aborted')
+            return
 
     _logger.info('Wiping schema `%s`', url)