Fix pypi token keyrinerror (#5189)

CHANGELOG.md

@@ -107,6 +107,8 @@
 - Improved documentation of the `readme` option, including multiple files and additional formats ([#5158](https://github.com/python-poetry/poetry/pull/5158))
 - Improved contributing documentation ([#5708](https://github.com/python-poetry/poetry/pull/5708))
 - Remove all references to `--dev-only` option ([#5771](https://github.com/python-poetry/poetry/pull/5771))
+- Added environment variable checking for PyPi token ([#5911](https://github.com/python-poetry/poetry/pull/5911))
+
 
 ## [1.2.0b1] - 2022-03-17
 

src/poetry/utils/password_manager.py

@@ -2,6 +2,7 @@
 
 import dataclasses
 import logging
+import os
 
 from contextlib import suppress
 from typing import TYPE_CHECKING
@@ -173,6 +174,25 @@ def set_pypi_token(self, name: str, token: str) -> None:
             self.keyring.set_password(name, "__token__", token)
 
     def get_pypi_token(self, name: str) -> str | None:
+        """Get PYPI token.
+
+        First checks the enviroment variables for a token,
+        then the configured username/password and last the
+        available keyring.
+
+
+        Parameters
+        ----------
+        name
+            PYPI username.
+
+        Returns
+        -------
+            Returns a token as a string if found, otherwise None.
+        """
+        if os.getenv("POETRY_PYPI_TOKEN_PYPI"):
+            return os.getenv("POETRY_PYPI_TOKEN_PYPI")
+
         if not self.keyring.is_available():
             token: str | None = self._config.get(f"pypi-token.{name}")
             return token

tests/utils/test_password_manager.py

@@ -3,6 +3,7 @@
 import os
 
 from typing import TYPE_CHECKING
+from unittest import mock
 
 import pytest
 
@@ -231,3 +232,39 @@ def test_get_http_auth_from_environment_variables(
 
     assert auth["username"] == "bar"
     assert auth["password"] == "baz"
+
+
+def test_get_pypi_token_with_env_var_positive(
+    config: Config, with_simple_keyring: None, dummy_keyring: DummyBackend
+):
+    sample_token = "sampletoken-1234"
+    manager = PasswordManager(config)
+
+    with mock.patch.dict(os.environ, {"POETRY_PYPI_TOKEN_PYPI": sample_token}):
+        result_token = manager.get_pypi_token("foo")
+
+    assert result_token == sample_token
+
+
+def test_get_pypi_token_with_env_var_negative(
+    config: Config, with_simple_keyring: None, dummy_keyring: DummyBackend
+):
+    sample_token = "sampletoken-1234"
+    manager = PasswordManager(config)
+
+    with mock.patch.dict(
+        os.environ, {"POETRY_PYPI_TOKEN_PYPI": sample_token + "somestuff"}
+    ):
+        result_token = manager.get_pypi_token("foo")
+
+    assert result_token != sample_token
+
+
+def test_get_pypi_token_with_env_var_not_available(
+    config: Config, with_simple_keyring: None, dummy_keyring: DummyBackend
+):
+    manager = PasswordManager(config)
+
+    result_token = manager.get_pypi_token("foo")
+
+    assert result_token is None