use SystemExit

xjunko · Nov 10, 2024 · 760b447 · 760b447
1 parent 9a5cdab
commit 760b447
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/app/objects/api.py b/app/objects/api.py
@@ -81,7 +81,7 @@ def from_file(cls, file: Path) -> Self | None:
             print(MISSING_FILE)
             print(SOLUTION_API_V1)
             print(SOLUTION_API_V2)
-            exit(1)
+            raise SystemExit(1)
 
         # Read values
         lines: list[str] = [
@@ -93,7 +93,7 @@ def from_file(cls, file: Path) -> Self | None:
             print(EMPTY_FILE)
             print(SOLUTION_API_V1)
             print(SOLUTION_API_V2)
-            exit(1)
+            raise SystemExit(1)
 
         # V1
         if len(lines) == 1:
@@ -107,4 +107,4 @@ def from_file(cls, file: Path) -> Self | None:
         print("[API] Invalid api key file.")
         print(SOLUTION_API_V1)
         print(SOLUTION_API_V2)
-        exit(1)
+        raise SystemExit(1)
diff --git a/app/objects/beatmap.py b/app/objects/beatmap.py
@@ -164,7 +164,7 @@ def from_id(cls, id: int):
                     print(
                         "[API] If this is a custom beatmap, please pass the beatmap path with `-b` param.",
                     )
-                    exit(1)
+                    raise SystemExit(1)
 
                 print(" success!")
                 beatmap_file.write_bytes(res.content)