73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
"""Construye un ZIP reproducible y limpio del Bot de Descarga."""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import hashlib
|
|
from pathlib import Path
|
|
from zipfile import ZIP_DEFLATED, ZipFile, ZipInfo
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
DEFAULT_OUTPUT = ROOT.parent / "bot_descarga.zip"
|
|
TOP_LEVEL_FILES = (
|
|
".env.example",
|
|
".gitignore",
|
|
"README.md",
|
|
"requirements.txt",
|
|
"run_gui.py",
|
|
"main.py",
|
|
)
|
|
EXCLUDED_PARTS = {"__pycache__", ".pytest_cache", ".venv", "data", "logs", "out"}
|
|
EXCLUDED_SUFFIXES = {".pyc", ".pyo", ".log"}
|
|
|
|
|
|
def package_files(root: Path) -> list[Path]:
|
|
files: list[Path] = []
|
|
for name in TOP_LEVEL_FILES:
|
|
path = root / name
|
|
if path.is_file():
|
|
files.append(path)
|
|
package_root = root / "bot_descarga"
|
|
for path in package_root.rglob("*"):
|
|
relative = path.relative_to(root)
|
|
if not path.is_file() or any(part in EXCLUDED_PARTS for part in relative.parts):
|
|
continue
|
|
if path.suffix.lower() in EXCLUDED_SUFFIXES:
|
|
continue
|
|
files.append(path)
|
|
return sorted(files, key=lambda path: path.relative_to(root).as_posix())
|
|
|
|
|
|
def build_package(root: Path = ROOT, output: Path = DEFAULT_OUTPUT) -> Path:
|
|
output.parent.mkdir(parents=True, exist_ok=True)
|
|
with ZipFile(output, "w", compression=ZIP_DEFLATED, compresslevel=9) as archive:
|
|
for source in package_files(root):
|
|
relative = source.relative_to(root).as_posix()
|
|
info = ZipInfo(f"bot_descarga/{relative}", date_time=(2020, 1, 1, 0, 0, 0))
|
|
info.compress_type = ZIP_DEFLATED
|
|
info.external_attr = 0o644 << 16
|
|
archive.writestr(info, source.read_bytes())
|
|
return output
|
|
|
|
|
|
def sha256(path: Path) -> str:
|
|
digest = hashlib.sha256()
|
|
with path.open("rb") as source:
|
|
for chunk in iter(lambda: source.read(1024 * 1024), b""):
|
|
digest.update(chunk)
|
|
return digest.hexdigest()
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("--output", type=Path, default=DEFAULT_OUTPUT)
|
|
args = parser.parse_args()
|
|
target = build_package(output=args.output.resolve())
|
|
print(f"Paquete: {target}")
|
|
print(f"SHA256: {sha256(target)}")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|