[py] Update lock and make sed work on non-linux

Author: cgoldberg

py/requirements.txt

@@ -39,7 +39,7 @@ pyOpenSSL==25.1.0
 pyparsing==3.2.3
 pyproject-api==1.9.1
 PySocks==1.7.1
-pytest==8.3.5
+pytest==8.4.0
 pytest-instafail==0.5.0
 pytest-mock==3.14.1
 pytest-trio==0.8.0

py/requirements_lock.txt

@@ -576,6 +576,7 @@ pygments==2.19.1 \
     --hash=sha256:9ea1544ad55cecf4b8242fab6dd35a93bbce657034b0611ee383099054ab6d8c
     # via
     #   -r py/requirements.txt
+    #   pytest
     #   readme-renderer
     #   rich
 pyopenssl==25.1.0 \
@@ -596,10 +597,12 @@ pysocks==1.7.1 \
     --hash=sha256:08e69f092cc6dbe92a0fdd16eeb9b9ffbc13cadfe5ca4c7bd92ffb078b293299 \
     --hash=sha256:2725bd0a9925919b9b51739eea5f9e2bae91e83288108a9ad338b2e3a4435ee5 \
     --hash=sha256:3f8804571ebe159c380ac6de37643bb4685970655d3bba243530d6558b799aa0
-    # via -r py/requirements.txt
-pytest==8.3.5 \
-    --hash=sha256:c69214aa47deac29fad6c2a4f590b9c4a9fdb16a403176fe154b79c0b4d4d820 \
-    --hash=sha256:f4efe70cc14e511565ac476b57c279e12a855b11f48f212af1080ef2263d3845
+    # via
+    #   -r py/requirements.txt
+    #   urllib3
+pytest==8.4.0 \
+    --hash=sha256:14d920b48472ea0dbf68e45b96cd1ffda4705f33307dcc86c676c1b5104838a6 \
+    --hash=sha256:f40f825768ad76c0977cbacdf1fd37c6f7a468e460ea6a0636078f8972d4517e
     # via
     #   -r py/requirements.txt
     #   pytest-instafail
@@ -737,7 +740,7 @@ typing-extensions==4.14.0 \
     #   pyopenssl
     #   rich
     #   tox
-urllib3==2.4.0 \
+urllib3[socks]==2.4.0 \
     --hash=sha256:414bc6535b787febd7567804cc015fee39daab8ad86268f1310a9250697de466 \
     --hash=sha256:4e16665048960a0900c702d4a66415956a584919c03361cac9f1df5c5dd7e813
     # via

scripts/update_py_dependencies.sh

@@ -49,14 +49,21 @@ echo
 pip list --outdated | while read -r line; do
     if [[ ! "${line}" =~ "Version Latest" && ! "${line}" =~ "----" ]]; then
         read -ra fields <<< "${line}"
-        echo "upgrading ${fields[0]} from ${fields[1]} to ${fields[2]}"
-        pip install --upgrade "${fields[0]}==${fields[2]}" > /dev/null
+        package="${fields[0]}"
+        echo "upgrading ${package} from ${fields[1]} to ${fields[2]}"
+        pip install --upgrade "${package}==${fields[2]}" > /dev/null
     fi
 done
 
 echo
 echo "generating new ${REQUIREMENTS_FILE}"
 pip freeze > "${REQUIREMENTS_FILE}"
+# `pip freeze` doesn't show package "extras", so we explicitly add it here for urllib3
+if [[ "${OSTYPE}" == "linux"* ]]; then
+    sed -i "s/urllib3/urllib3[socks]/g" "${REQUIREMENTS_FILE}" # GNU sed
+else
+    sed -i "" "s/urllib3/urllib3[socks]/g" "${REQUIREMENTS_FILE}"
+fi
 
 echo "generating new lock file"
 bazel run //py:requirements.update