API Storage cửa hàng: tải tiếp theo phần

"""Python 3.11+; pip install requests. Set WHYPIK_STORAGE_KEY in the environment.

Upload an independent Store file. Reuse the same UUID/file/key to resume for 48h.
This does not publish a software release or bypass its scan/license policy.
"""

import argparse
import hashlib
import os
import uuid
from pathlib import Path

import requests


def main():
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--shop", required=True, type=uuid.UUID)
    parser.add_argument("--file", required=True, type=Path)
    parser.add_argument("--idempotency-key", required=True, type=uuid.UUID)
    args = parser.parse_args()
    base = f"https://app.whypik.com/api/v1/storage/{args.shop}"
    with requests.Session() as session:
        session.headers["Authorization"] = "Bearer " + os.environ["WHYPIK_STORAGE_KEY"]

        def call(method, suffix, **kwargs):
            response = session.request(
                method, base + suffix, timeout=(10, 190), allow_redirects=False, **kwargs
            )
            if response.status_code not in {200, 201}:
                raise RuntimeError(f"HTTP {response.status_code}; keep the same UUID to retry.")
            return response.json()

        with args.file.open("rb") as stream:
            sha = hashlib.file_digest(stream, "sha256").hexdigest()
        row = call("POST", "/files", headers={"Idempotency-Key": str(args.idempotency_key)}, json={
            "filename": args.file.name, "size": args.file.stat().st_size,
            "sha256": sha, "transport": "chunks-v1",
        })["upload"]
        path = f"/uploads/{uuid.UUID(row['id'])}"
        if row["state"] == "completed":
            print("Already received; inspect scan state:", row["release_id"])
            return
        if row["state"] not in {"pending", "receiving"}:
            raise RuntimeError("Reservation closed; inspect it before creating another.")
        size = row["chunk_size"]
        if type(size) is not int or not 1 <= size <= 8 * 1024**2:
            raise RuntimeError("Invalid server chunk size")
        if row["state"] == "pending":
            saved = {part["index"]: part for part in row["parts"]}
            with args.file.open("rb") as stream:
                index = 0
                while body := stream.read(size):
                    digest = hashlib.sha256(body).hexdigest()
                    if index in saved:
                        if saved[index]["size"] != len(body) or saved[index]["sha256"] != digest:
                            raise RuntimeError("File differs from accepted bytes")
                    else:
                        call("PUT", path + f"/parts/{index}", data=body, headers={
                            "Content-Type": "application/octet-stream", "X-Chunk-SHA256": digest,
                        })
                    index += 1
        result = call("POST", path + "/complete", json={})["file"]
        print("Received; still requires a clean scan:", result["id"])


if __name__ == "__main__":
    main()