Upload a package to Jamf's JCDS2

pkg_upload_jcds2.py

#!/usr/bin/env python3

"""
A script to upload a package to the Jamf Cloud Distribution Point S3 bucket (JCDS 2.0)

Requirements from pip
boto3
requests
"""

import boto3
from botocore.exceptions import ClientError
import os.path
import requests
from requests.exceptions import HTTPError
import sys
import threading


# REQUIRED AUTHENTICATION VARIABLES
jamfProUser = ""
jamfProPassword = ""
jamfProBaseURL = ""

# path to package - UPDATE AS APPROPRIATE
pkg_path = os.path.join(
    "/Users/gpugh/sourcecode",
    "erase-install/pkg/erase-install/build/erase-install-30.0.pkg",
)


class ProgressPercentage(object):
    """display upload progress"""

    def __init__(self, filename):
        self._filename = filename
        self._size = float(os.path.getsize(filename))
        self._seen_so_far = 0
        self._lock = threading.Lock()

    def __call__(self, bytes_amount):
        # To simplify, assume this is hooked up to a single filename
        with self._lock:
            self._seen_so_far += bytes_amount
            percentage = (self._seen_so_far / self._size) * 100
            sys.stdout.write(
                "\r%s  %s / %s  (%.2f%%)"
                % (self._filename, self._seen_so_far, self._size, percentage)
            )
            sys.stdout.flush()


try:
    pkg = os.path.basename(pkg_path)

    response = requests.post(
        jamfProBaseURL + "/api/v1/auth/token",
        auth=(jamfProUser, jamfProPassword),
        data="",
    )
    response.raise_for_status()

    jsonResponse = response.json()
    print("Retreived Token: ")
    jamfProToken = jsonResponse["token"]
    print(jamfProToken)

    # Initiate Upload via Jamf Pro API
    headers = {"Accept": "application/json", "Authorization": "Bearer " + jamfProToken}

    print(headers)

    response = requests.post(
        jamfProBaseURL + "/api/v1/jcds/files", headers=headers, data=""
    )
    response.raise_for_status()

    credentials = response.json()
    print("Retreived Credentials Object: ")
    print(credentials)


except HTTPError as http_err:
    print(f"HTTP error occurred: {http_err}")
except Exception as err:
    print(f"Other error occurred: {err}")


# Upload File To AWS S3
s3_client = boto3.client(
    "s3",
    aws_access_key_id=credentials["accessKeyID"],
    aws_secret_access_key=credentials["secretAccessKey"],
    aws_session_token=credentials["sessionToken"],
)
try:
    response = s3_client.upload_file(
        pkg_path,
        credentials["bucketName"],
        credentials["path"] + pkg,
        Callback=ProgressPercentage(pkg_path),
    )
    print(response)
except ClientError as e:
    print(f"Failure uploading to S3: {e}")

Also, updated for API Clients. Because of the /JSSResource/packages/id/ is needed, the API client would also need access to Packages, as well as Jamf Content Distribution Server Files

#!/usr/bin/env python3

"""
A script to upload a package to the Jamf Cloud Distribution Point S3 bucket (JCDS 2.0)

Requirements from pip
boto3
requests

Updated from @grahampugh's original script to use API Clients, instead of user accounts

"""

import boto3
from botocore.exceptions import ClientError
import os.path
import requests
from requests.exceptions import HTTPError
import sys
import threading


# REQUIRED AUTHENTICATION VARIABLES
JAMF_PRO_CLIENT_ID = "client_id"
JAMF_PRO_CLIENT_SECRET = "client_secret"
JAMF_URL = "https://jss.jamfcloud.com"

# path to package - UPDATE AS APPROPRIATE
pkg_path = os.path.join(
    "/home/ubuntu/jcds",
    "InstallAssistant-14.3-23D56.pkg",
)


class ProgressPercentage(object):
    """display upload progress"""

    def __init__(self, filename):
        self._filename = filename
        self._size = float(os.path.getsize(filename))
        self._seen_so_far = 0
        self._lock = threading.Lock()

    def __call__(self, bytes_amount):
        # To simplify, assume this is hooked up to a single filename
        with self._lock:
            self._seen_so_far += bytes_amount
            percentage = (self._seen_so_far / self._size) * 100
            sys.stdout.write(
                "\r%s  %s / %s  (%.2f%%)"
                % (self._filename, self._seen_so_far, self._size, percentage)
            )
            sys.stdout.flush()


try:
    pkg = os.path.basename(pkg_path)

    # Get Jamf Pro API Token using API Clients
    URL = JAMF_URL + "/api/oauth/token"
    headers = {"Content-Type": "application/x-www-form-urlencoded"}
    data = {
        "grant_type": "client_credentials",
        "client_id": JAMF_PRO_CLIENT_ID,
        "client_secret": JAMF_PRO_CLIENT_SECRET,
    }
    response = requests.post(URL, headers=headers, data=data)
    
    response.raise_for_status()
    jsonResponse = response.json()
    print("Retreived Token: ")
    jamfProToken = jsonResponse["access_token"]
    print(jamfProToken)

    # Initiate Upload via Jamf Pro API
    headers = {"Accept": "application/json", "Authorization": "Bearer " + jamfProToken}

    print(headers)

    response = requests.post(
        JAMF_URL + "/api/v1/jcds/files", headers=headers, data=""
    )
    response.raise_for_status()

    credentials = response.json()
    print("Retreived Credentials Object: ")
    print(credentials)


except HTTPError as http_err:
    print(f"HTTP error occurred: {http_err}")
except Exception as err:
    print(f"Other error occurred: {err}")

# Upload File To AWS S3
s3_client = boto3.client(
    "s3",
    region_name=credentials['region'],
    aws_access_key_id=credentials["accessKeyID"],
    aws_secret_access_key=credentials["secretAccessKey"],
    aws_session_token=credentials["sessionToken"],
)
try:
    response = s3_client.upload_file(
        pkg_path,
        credentials["bucketName"],
        credentials["path"] + pkg,
        Callback=ProgressPercentage(pkg_path),
    )
    print(response)
except ClientError as e:
    print(f"Failure uploading to S3: {e}")


# Tell Jamf Pro about the package
pkg_data = f"<package><name>{pkg}</name><filename>{pkg}</filename></package>"
try:
    headers = {"Content-Type": "application/xml", "Authorization": "Bearer " + jamfProToken}
    response = requests.post(
        JAMF_URL + "/JSSResource/packages/id/0", headers=headers, data=pkg_data
    )
    response.raise_for_status()
    # if response includes 201, then package was created, if 409, then package already exists
    if response.status_code == 201:
        print("Package created")
    if response.status_code == 409:
        print("Package already exists")
except HTTPError as http_err:
    print(f"HTTP error occurred: {http_err}")
except Exception as err:
    print(f"Other error occurred: {err}")