Merge pull request bongtrop#1 from utpk/patch-1

improve yes or no input handling
Kirlif · Jan 11, 2021 · b21b4bd · b21b4bd
2 parents 1bd94f6 + 932eaca
commit b21b4bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hbctool/hasm.py b/hbctool/hasm.py
@@ -33,8 +33,8 @@ def write_func(f, func, i, hbc):
 def dump(hbc, path, force=False):
 
     if os.path.exists(path) and not force:
-        c = input(f"'{path}' exists. Do you want to remove it ? (y/n): ").strip()
-        if c == "y":
+        c = input(f"'{path}' exists. Do you want to remove it ? (y/n): ").lower().strip()
+        if c[:1] == "y":
             shutil.rmtree(path)
         else:
             exit(1337)