Merge pull request #15 from YeetCode-devs/staging/prathamdby

Improve the install_hook script's logic and add the python-dotenv dependency in poetry

pyproject.toml

@@ -11,6 +11,7 @@ package-mode = false
 python = "^3.10"
 g4f = {extras = ["all"], version = "^0.2.5.4"}
 telethon = "^1.34.0"
+python-dotenv = "^1.0.1"
 
 [tool.poetry.group.dev.dependencies]
 black = "^24.3.0"

scripts/install_hook.py

@@ -21,13 +21,19 @@
 SCRIPT = r"""#!/bin/sh
 set -o noglob
 
-python3 scripts/install_hook.py
+python3 scripts/install_hook.py || python scripts/install_hook.py
 
 poetry run black .
 
-for file in $(git diff --cached --name-only); do
-    test -f $file && git add $file
-done"""
+files=$(git diff --cached --name-only)
+
+if [ -n "$files" ]; then
+    for file in $files; do
+        if [ -f "$file" ]; then
+            git add "$file"
+        fi
+    done
+fi"""
 
 
 def main() -> None:
@@ -40,4 +46,5 @@ def main() -> None:
     Path(".git/hooks/pre-commit").chmod(0o700)
 
 
-main()
+if __name__ == "__main__":
+    main()