python: allow custom proxy for poetry

Author: morpheu

python/deploy

@@ -208,6 +208,22 @@ if [ -f "${CURRENT_DIR}/poetry.lock" ]; then
     fi
     pip install poetry${POETRY_VERSION_SPEC}
     poetry config virtualenvs.create false
+
+    # Configure custom Poetry repositories from environment variables
+    # Format: POETRY_REPOSITORIES_<NAME>_URL=<URL>
+    # Example: POETRY_REPOSITORIES_ARTIFACTORY_URL=https://artifactory.example.com/pypi/simple
+    for var in $(env | grep '^POETRY_REPOSITORIES_.*_URL=' | cut -d= -f1); do
+        # Extract repository name from variable name
+        # POETRY_REPOSITORIES_ARTIFACTORY_URL -> ARTIFACTORY
+        repo_name=$(echo "$var" | sed 's/^POETRY_REPOSITORIES_//' | sed 's/_URL$//' | tr '[:upper:]' '[:lower:]')
+        repo_url="${!var}"
+
+        if [ -n "$repo_url" ]; then
+            echo "Configuring Poetry repository: $repo_name -> $repo_url"
+            poetry source add --priority=primary "$repo_name" "$repo_url"
+        fi
+    done
+
     # Check if lock file needs updating and regenerate if needed
     if ! poetry check --lock 2>/dev/null; then
         echo "Lock file out of sync, regenerating..."

tests/python/tests.bats

@@ -468,4 +468,26 @@ EOF
 
     rm ${CURRENT_DIR}/pyproject.toml ${CURRENT_DIR}/poetry.lock
     unset PYTHON_VERSION
+}
+
+@test "install from poetry.lock with custom repository" {
+    unset PYTHON_VERSION
+    export POETRY_REPOSITORIES_ARTIFACTORY_URL=https://pypi.org/simple
+    cp pyproject.toml poetry.lock ${CURRENT_DIR}/
+
+    run /var/lib/tsuru/deploy
+    assert_success
+
+    [[ "$output" == *"poetry.lock detected"* ]]
+    [[ "$output" == *"Configuring Poetry repository: artifactory -> https://pypi.org/simple"* ]]
+
+    pushd ${CURRENT_DIR}
+    run pip freeze
+    popd
+
+    assert_success
+    [[ "$output" == *"msgpack"* ]]
+
+    rm ${CURRENT_DIR}/pyproject.toml ${CURRENT_DIR}/poetry.lock
+    unset POETRY_REPOSITORIES_ARTIFACTORY_URL
 }