Use legacycrypt where crypt isn't available

Since Python 3.13 has removed the crypt module, add legacycrypt to the requirements if we're using that version of Python, importing it if the crypt module isn't found. Fixes networktocode#594
s-t-e-v-e-n-k · Dec 3, 2024 · 3517bfc · 3517bfc
1 parent b35bad3
commit 3517bfc
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/netutils/password.py b/netutils/password.py
@@ -1,7 +1,9 @@
 """Functions for working with Passwords."""
 
-# TODO: Swap out crypt prior to py3.13
-import crypt  # pylint: disable=deprecated-module
+try:
+    import crypt  # pylint: disable=deprecated-module
+except ModuleNotFoundError:
+    import legacycrypt as crypt
 import random
 import secrets
 import string

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -29,6 +29,7 @@ include = [
 python = "^3.8"
 napalm = {version = "^4.0.0", optional = true}
 jsonschema = {version = "^4.17.3", optional = true}
+legacycrypt = {markers = "python_version >= \"3.13\"", version = "^0.3"}
 
 [tool.poetry.extras]
 optionals = ["jsonschema", "napalm"]