Files
homelab/scripts/run_updates.py
Alex Frantz 77504eeffd
All checks were successful
Deploy Containers / Prepare (push) Successful in 4s
fix update script
2025-11-21 21:52:00 -05:00

54 lines
1.8 KiB
Python

import requests
import subprocess
import os
import yaml
def search_for_image(image_name):
tasks_folder = os.path.realpath(os.path.join('./', 'tasks'))
if "library/" in image_name:
image_name = image_name.replace("library/", "")
for task in os.listdir(tasks_folder):
with open(os.path.join(tasks_folder, task), 'r') as file:
data = yaml.safe_load(file)
for key in data:
if "vars" in key:
if image_name in key["vars"]["image"]["name"]:
return f"{task.split(".")[0]}_deploy"
def main():
update_list = requests.get("https://cup.fntz.net/api/v3/json")
update_list.raise_for_status()
update_list = update_list.json()
deployable_tags = []
with open("main.yml", 'r') as file:
data = yaml.safe_load(file)
for host in data:
for task in host['tasks']:
deployable_tags.append(task['tags'])
if len(update_list["images"]) <= 0:
print("No images to update!")
else:
print(f"Updating {update_list["metrics"]["updates_available"]} image(s)..\n")
already_deployed = []
for image in update_list["images"]:
if image['result']['has_update']:
if "remote_digest" in image["result"]["info"]:
image_name = image["parts"]["repository"]
ansible_tag = search_for_image(image_name)
if ansible_tag and ansible_tag in deployable_tags and ansible_tag not in already_deployed:
print(f"Updating '{image_name}' ({ansible_tag})..")
subprocess.run(f'ANSIBLE_CONFIG=ansible.cfg ansible-playbook main.yml --tags {ansible_tag} --vault-password-file=~/.vault_pass.txt', shell=True)
already_deployed.append(ansible_tag)
print("\nAll images updated, refreshing Cup")
requests.get("https://cup.fntz.net/api/v3/refresh")
if __name__ == "__main__":
main()