petekio 0.3.15

Subsurface data ingestion + structure layer: surfaces, wells, points, polygons with loading, interpolation, and statistics.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

#!/usr/bin/env python3
"""Wait until a just-published package is installable from PyPI.

PyPI's JSON/project page can become visible before pip's simple index is usable
from every runner. Release workflows use this as the final PyPI visibility gate.
"""

from __future__ import annotations

import argparse
import os
import subprocess
import sys
import tempfile
import time
from pathlib import Path


def main() -> int:
    parser = argparse.ArgumentParser()
    parser.add_argument("package")
    parser.add_argument("version")
    parser.add_argument("--attempts", type=int, default=30)
    parser.add_argument("--sleep", type=float, default=20.0)
    parser.add_argument("--import-name", default=None)
    args = parser.parse_args()

    requirement = f"{args.package}=={args.version}"
    import_name = args.import_name or args.package.replace("-", "_")

    with tempfile.TemporaryDirectory(prefix="wait-pypi-") as tmp:
        for attempt in range(1, args.attempts + 1):
            target = Path(tmp) / f"site-{attempt}"
            target.mkdir()
            print(f"PyPI visibility check {attempt}/{args.attempts}: {requirement}", flush=True)
            install = subprocess.run(
                [
                    sys.executable,
                    "-m",
                    "pip",
                    "install",
                    "--disable-pip-version-check",
                    "--no-cache-dir",
                    "--only-binary=:all:",
                    "--target",
                    str(target),
                    requirement,
                ],
                text=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.STDOUT,
            )
            if install.returncode == 0:
                env = dict(os.environ)
                env["PYTHONPATH"] = str(target) + os.pathsep + env.get("PYTHONPATH", "")
                smoke = subprocess.run(
                    [sys.executable, "-c", f"import {import_name}; print({import_name}.__name__)"],
                    env=env,
                    text=True,
                    stdout=subprocess.PIPE,
                    stderr=subprocess.STDOUT,
                )
                if smoke.returncode == 0:
                    print(smoke.stdout, end="")
                    return 0
                print(smoke.stdout, end="")
            else:
                print(install.stdout, end="")

            if attempt < args.attempts:
                time.sleep(args.sleep)

    print(f"{requirement} was not installable from PyPI in time", file=sys.stderr)
    return 1


if __name__ == "__main__":
    raise SystemExit(main())