yeet_api: Add support for windows and guard against unsupported OSSes

YeetCodeDevs · Jan 25, 2024 · c8f6bbb · c8f6bbb
1 parent e132dca
commit c8f6bbb
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/yeet_api/__init__.py b/yeet_api/__init__.py
@@ -5,6 +5,15 @@
 
 from pathlib import Path
 
+_home_dir: str
+
+if os.name == "posix":
+    _home_dir = os.getenv("HOME", "")
+elif os.name == "nt":
+    _home_dir = os.getenv("USERPROFILE", "")
+else:
+    raise OSError("Unsupported OS: yeet-api only supports Windows and unix-like OSses.")
+
 if not shutil.which("git"):
     raise FileNotFoundError("Git executable is not found."
                             "\nFor Windows, install it from https://git-scm.com/downloads"