From f12f274f278db9ad4728ff5fbcf3b0cea0903174 Mon Sep 17 00:00:00 2001
From: Adam Miller <maxamillion@fedoraproject.org>
Date: Oct 19 2016 22:11:15 +0000
Subject: Add special case for latest target w/ Atomic ISO, it's not in fedmsg data


Signed-off-by: Adam Miller <maxamillion@fedoraproject.org>

---

diff --git a/getfedora.org/build/atomic_vars.py b/getfedora.org/build/atomic_vars.py
index e99da9a..89a47aa 100755
--- a/getfedora.org/build/atomic_vars.py
+++ b/getfedora.org/build/atomic_vars.py
@@ -16,6 +16,7 @@ import functools
 import json
 import logging
 import os
+import sys
 import socket
 
 from datetime import datetime, timedelta
@@ -27,6 +28,14 @@ import requests
 
 log = logging.getLogger("atomic_vars")
 
+try:
+    sys.path.append('../build.d')
+
+    import globalvar
+except ImportError:
+    log.error("Unable to import globalvar")
+    sys.exit(1)
+
 base_url = 'https://apps.fedoraproject.org/datagrepper/raw'
 topic = "org.fedoraproject.prod.releng.atomic.twoweek.complete"
 
@@ -185,6 +194,20 @@ def collect(curr_id, next_id):
             iso_size_key = iso_size_prefix + mapping[key]
             results['iso_size'][iso_size_key] = str(length)
 
+    # Special case for Atomic ISO latest redirect rule mapping because it's
+    # not included in fedmsg data
+    atomic_iso_filename = "Fedora-Atomic-dvd-x86_64-{}-{}.iso".format(
+        globalvar.release['curr_id'],
+        results['release'][composedate_prefix + 'atomic_composedate']
+    )
+    results['release']['redir_map']['atomic_iso'] = {}
+    results['release']['redir_map']['atomic_iso']['redirect'] = \
+        globalvar.path['download_atomic'] + "/stable/Fedora-Atomic-" + \
+        globalvar.release['curr_id'] + '-' + \
+        results['release'][composedate_prefix + 'atomic_composedate'] + \
+        "/Atomic/x86_64/iso/{}".format(atomic_iso_filename)
+    results['release']['redir_map']['atomic_iso']['filename'] = atomic_iso_filename
+
     return results