Preserve cache (#114)

* Preserve wheels from cache

* bump version

builder/__main__.py

@@ -11,9 +11,9 @@
 
 from builder.apk import install_apks
 from builder.infra import (
+    check_available_binary,
     create_wheels_folder,
     create_wheels_index,
-    check_available_binary,
 )
 from builder.pip import (
     build_wheels_local,
@@ -24,7 +24,7 @@
     write_requirement,
 )
 from builder.upload import run_upload
-from builder.utils import check_url, fix_wheels_name
+from builder.utils import check_url, copy_wheels_from_cache, fix_wheels_name
 
 
 @click.command("builder")
@@ -126,6 +126,7 @@ def builder(
                     exit_code = 109
                 except TimeoutExpired:
                     exit_code = 80
+                    copy_wheels_from_cache(Path("/root/.cache/pip/wheels"), wheels_dir)
         else:
             # Build all needed wheels at once
             packages = extract_packages(requirement, requirement_diff)
@@ -146,6 +147,7 @@ def builder(
                 exit_code = 109
             except TimeoutExpired:
                 exit_code = 80
+                copy_wheels_from_cache(Path("/root/.cache/pip/wheels"), wheels_dir)
 
         fix_wheels_name(wheels_dir)
         run_upload(upload, output, remote)

builder/utils.py

@@ -1,10 +1,12 @@
 """Some utils for builder."""
-from pathlib import Path
+from contextlib import suppress
 import os
+from pathlib import Path
 import re
+import shutil
 import subprocess
 import sys
-from typing import Optional, Dict
+from typing import Dict, Optional
 
 import requests
 
@@ -38,6 +40,13 @@ def fix_wheels_name(wheels_folder: Path) -> None:
         package.rename(Path(package.parent, f"{match.group('name')}none-any.whl"))
 
 
+def copy_wheels_from_cache(cache_folder: Path, wheels_folder: Path) -> None:
+    """Preserve wheels from cache on timeout error."""
+    for wheel_file in cache_folder.glob("**/*.whl"):
+        with suppress(OSError):
+            shutil.copy(wheel_file, wheels_folder)
+
+
 def run_command(
     cmd: str, env: Optional[Dict[str, str]] = None, timeout: Optional[int] = None
 ) -> None:

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup
 
-VERSION = "1.12.3"
+VERSION = "1.12.4"
 
 setup(
     name="builder",