DownloadAndArchivePythonLibrary

Source

import shlex
import subprocess
import zipfile
from pathlib import Path
from tempfile import mkdtemp

import demistomock as demisto  # noqa: F401
from CommonServerPython import *  # noqa: F401


def installLibrary(dir_path: Path, library_name: str) -> str:
    # Install the package using pip
    demisto.debug(f"Running in {dir_path=}, {library_name=}")
    cmd = f"python3 -m pip install --target {dir_path} {library_name}"
    process = subprocess.Popen(shlex.split(cmd), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout, stderr = process.communicate()
    demisto.debug(f"returned code: {process.returncode}")
    if process.returncode != 0:
        raise DemistoException(f"Failed to install the {library_name} library: {stderr.decode('utf-8')}")

    # Create a zip file with maximum compression level
    zip_path = dir_path / (library_name + ".zip")
    with zipfile.ZipFile(zip_path, "w", compression=zipfile.ZIP_DEFLATED, compresslevel=9) as zip_file:  # pylint: disable=E1101
        for root, _dirs, files in os.walk(dir_path):
            for file in files:
                demisto.debug(f"{root=}, {file=}")
                file_path = Path(root) / file
                # Ensure the folder inside the ZIP file is named 'python'
                arcname = Path("python") / file_path.relative_to(dir_path)
                zip_file.write(file_path, arcname=arcname)
                demisto.debug("Updated file")

    # Read the zip file contents
    with open(zip_path, "rb") as zip_file:
        zip_content = zip_file.read()

    return fileResult(library_name + ".zip", zip_content)


def main():
    args = demisto.args()
    library_name = args.get("library_name")
    try:
        dir_path = Path(mkdtemp(prefix="python"))
        demisto.debug("Starting installing library.")
        result = installLibrary(dir_path, library_name)
        return_results(result)

    except Exception as e:
        return_error(f"An error occurred: {e!s}")


if __name__ in ("__builtin__", "builtins"):
    main()

README