mirror of
https://github.com/fleetdm/fleet
synced 2026-04-21 21:47:20 +00:00
2a5629248c
[Github issue: 13323](https://github.com/fleetdm/confidential/issues/13323#event-21368586679)
110 lines
3 KiB
Python
110 lines
3 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import requests
|
|
|
|
# First make sure the environment variable exists
|
|
# > export FLEET_API_TOKEN="your_token"
|
|
api_token = os.getenv("FLEET_API_TOKEN")
|
|
if not api_token:
|
|
print("No token found in the environment")
|
|
raise
|
|
|
|
#############################
|
|
# ENTER YOUR INFORMATION HERE
|
|
#############################
|
|
base_url = "https://your-Fleet-server.com" # no trailing slash
|
|
api_path = "api/v1/fleet"
|
|
script_path = "script.sh" # relative path to this file (example is a script in the same directory)
|
|
#
|
|
# END
|
|
|
|
|
|
def get_all_results(endpoint, key, headers=None, params=None, per_page=10):
|
|
"""generic GET request that handles pagination"""
|
|
all_results = []
|
|
page = 0
|
|
|
|
if params is None:
|
|
params = {}
|
|
|
|
endpoint = f"{base_url}/{api_path}/{endpoint}"
|
|
headers = {"Authorization": f"Bearer {api_token}"}
|
|
params["per_page"] = per_page
|
|
|
|
while True:
|
|
params["page"] = page
|
|
|
|
response = requests.get(
|
|
endpoint,
|
|
headers=headers,
|
|
params=params,
|
|
)
|
|
response.raise_for_status()
|
|
|
|
results = response.json().get(key, [])
|
|
|
|
if not results:
|
|
break
|
|
|
|
all_results.extend(results)
|
|
|
|
if len(results) < per_page:
|
|
break
|
|
|
|
page += 1
|
|
|
|
return all_results
|
|
|
|
|
|
def upload_script(script_path, team_id):
|
|
"""
|
|
Upload a script to team if not already present.
|
|
Patch it if present
|
|
"""
|
|
headers = {"Authorization": f"Bearer {api_token}"}
|
|
script_name = os.path.basename(script_path)
|
|
|
|
# Check if script already exists in this team
|
|
existing_scripts = get_all_results(
|
|
"scripts", "scripts", params={"team_id": team_id}
|
|
)
|
|
|
|
existing_script_id = None
|
|
for script in existing_scripts:
|
|
if script.get("name") == script_name:
|
|
existing_script_id = script.get("id")
|
|
break
|
|
|
|
with open(script_path, "rb") as script_file:
|
|
files = {
|
|
"script": (
|
|
script_name,
|
|
script_file,
|
|
"application/octet-stream",
|
|
)
|
|
}
|
|
|
|
if existing_script_id:
|
|
# Script exists - PATCH to update
|
|
endpoint = f"{base_url}/{api_path}/scripts/{existing_script_id}"
|
|
response = requests.patch(endpoint, headers=headers, files=files)
|
|
action = "updated"
|
|
else:
|
|
# Script doesn't exist - POST to create
|
|
endpoint = f"{base_url}/{api_path}/scripts"
|
|
files["team_id"] = (None, str(team_id)) # Add team_id to form data
|
|
response = requests.post(endpoint, headers=headers, files=files)
|
|
action = "created"
|
|
|
|
response.raise_for_status()
|
|
print(f"Script '{script_name}' {action} for team_id {team_id}")
|
|
|
|
return response
|
|
|
|
|
|
if __name__ == "__main__":
|
|
all_teams = get_all_results("teams", "teams")
|
|
all_team_ids = {team["id"]: team["name"] for team in all_teams}
|
|
for team_id in all_team_ids:
|
|
response = upload_script(script_path, team_id)
|