finally write a get cookies script

Author: NotFish232

.gitignore

@@ -1,3 +1,4 @@
-/target
+target/
+venv/
 .direnv/
 .envrc

flake.nix

@@ -9,6 +9,8 @@
     let
       system = "x86_64-linux";
       pkgs = import nixpkgs { inherit system; };
+      python = pkgs.python312;
+      venv_dir = "./venv/";
     in
     {
       devShells.${system}.default = pkgs.mkShell {
@@ -17,12 +19,19 @@
           cargo
           clippy
           rust-analyzer
-          (python312.withPackages (
-            ps: with ps; [
-              natsort
-            ]
-          ))
+          python.pkgs.venvShellHook
         ];
+        venvDir = venv_dir;
+        postShellHook = ''
+          SENTINEL="${venv_dir}/.installed"
+          REQUIREMENTS="requirements.txt"
+
+          if [ ! -f "$SENTINEL" ] || [ "$REQUIREMENTS" -nt "$SENTINEL" ]; then
+            pip install --upgrade pip
+            pip install -r "$REQUIREMENTS"
+            touch "$SENTINEL"
+          fi
+        '';
       };
     };
 }

requirements.txt

@@ -0,0 +1,2 @@
+natsort==8.4.0
+browser_cookie3==0.20.1

scripts/get_cookies.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+
+import browser_cookie3  # type: ignore
+import os
+import re
+
+
+LEETCODE_URL = "https://leetcode.com"
+
+CSRFTOKEN_KEY = "csrftoken"
+LEETCODE_SESSION_KEY = "LEETCODE_SESSION"
+
+
+CONFIG_LOCATION = "~/.leetcode/leetcode.toml"
+CONFIG_LOCATION = os.path.expanduser(CONFIG_LOCATION)
+
+CSRF_CONFIG_REGEX = re.compile(r"^csrf = \".*\"$", re.MULTILINE)
+SESSION_CONFIG_REGEX = re.compile(r"^session = \".*\"$", re.MULTILINE)
+
+
+def main() -> None:
+    cj = browser_cookie3.chrome(domain_name="leetcode.com")
+    cookies = {c.name: c.value for c in cj}
+
+    csrftoken = cookies[CSRFTOKEN_KEY]
+    leetcode_session = cookies[LEETCODE_SESSION_KEY]
+
+    with open(CONFIG_LOCATION, "r") as f:
+        content = f.read()
+
+    content = CSRF_CONFIG_REGEX.sub(f'csrf = "{csrftoken}"', content)
+    content = SESSION_CONFIG_REGEX.sub(f'session = "{leetcode_session}"', content)
+
+    with open(CONFIG_LOCATION, "w") as f:
+        f.write(content)
+
+
+if __name__ == "__main__":
+    main()