diff --git a/df_translation_toolkit/csv_bisect/cli.py b/df_translation_toolkit/csv_bisect/cli.py index 0414836..7a650c1 100644 --- a/df_translation_toolkit/csv_bisect/cli.py +++ b/df_translation_toolkit/csv_bisect/cli.py @@ -3,4 +3,4 @@ from df_translation_toolkit.csv_bisect import csv_bisect app = typer.Typer() -app.command(name="from_string_dump")(csv_bisect.main) +app.command(name="csv_bisect")(csv_bisect.main) diff --git a/df_translation_toolkit/csv_bisect/csv_bisect.py b/df_translation_toolkit/csv_bisect/csv_bisect.py index 3b80300..d5a49a6 100644 --- a/df_translation_toolkit/csv_bisect/csv_bisect.py +++ b/df_translation_toolkit/csv_bisect/csv_bisect.py @@ -32,7 +32,7 @@ def _bisect(start: int, end: int, bad: bool = False) -> bool: if bad: confirmed = True else: - confirmed = input("Is it bad (Y/N)? ").upper() == "Y" + confirmed = input("Is it bad (y/n)? ").upper() == "y" if not confirmed: return False @@ -41,7 +41,7 @@ def _bisect(start: int, end: int, bad: bool = False) -> bool: print(f"Found string, line number {start+1}:") print(data[start]) - confirmed = input("Exclude from csv (Y/N)? ").upper() == "Y" + confirmed = input("Exclude from csv (y/n)? ").upper() == "y" if confirmed: csv_utils.write_csv(file_path, encoding, data[:start] + data[start + 1 :]) @@ -66,6 +66,6 @@ def main(csv_file: Path, encoding: str): data = list(csv_utils.read_csv(backup_path, encoding)) bisect(csv_file, encoding, data) finally: - confirmed = input("Restore from backup (Y/N)? ").upper() == "Y" + confirmed = input("Restore from backup (y/n)? ").upper() == "y" if confirmed: shutil.copy(backup_path, csv_file)