From 2cb8072ad8711d2111a015084cace4b40fe99ea5 Mon Sep 17 00:00:00 2001
From: Pavel Chechetin <pchechetin@mirantis.com>
Date: Fri, 7 Oct 2016 06:11:52 +0300
Subject: [PATCH] Remove old repos from compute on preupgrade stage

Change-Id: I93dcf51e1592cde0e72936c4bb7cf16314a7ad5d
---
 .../octane_tasks/files/delete_section.py       | 18 ++++++++++++++++++
 .../graphs/orig/preupgrade-compute.yaml        | 12 +++++++++++-
 2 files changed, 29 insertions(+), 1 deletion(-)
 create mode 100644 deployment/puppet/octane_tasks/files/delete_section.py

diff --git a/deployment/puppet/octane_tasks/files/delete_section.py b/deployment/puppet/octane_tasks/files/delete_section.py
new file mode 100644
index 00000000..3e55875d
--- /dev/null
+++ b/deployment/puppet/octane_tasks/files/delete_section.py
@@ -0,0 +1,18 @@
+#!/usr/bin/python
+
+import yaml
+import sys
+
+target_file = sys.argv[1]
+section = sys.argv[2]
+subsection = sys.argv[3]
+
+try:
+    with open(target_file,'r+') as f:
+        data = yaml.load(f)
+        del data[section][subsection]
+
+    with open(target_file,'w+') as f:
+        yaml.dump(data,f,default_flow_style=False)
+except KeyError as e:
+    print "Failed to find key: {0}".format(e)
diff --git a/deployment/puppet/octane_tasks/graphs/orig/preupgrade-compute.yaml b/deployment/puppet/octane_tasks/graphs/orig/preupgrade-compute.yaml
index ef8aed27..e348a8e2 100644
--- a/deployment/puppet/octane_tasks/graphs/orig/preupgrade-compute.yaml
+++ b/deployment/puppet/octane_tasks/graphs/orig/preupgrade-compute.yaml
@@ -56,7 +56,7 @@
   type: puppet
   version: 2.1.0
   groups: [compute]
-  requires: [cleanup_existing_repos, rsync_latest_puppet, override_repos_in_hiera]
+  requires: [cleanup_existing_repos, rsync_latest_puppet, override_repos_in_hiera, remove_hiera_section_repo_setup]
   required_for: []
   parameters:
     puppet_manifest: /etc/fuel/octane/latest_modules/osnailyfacter/modular/fuel_pkgs/setup_repositories.pp
@@ -85,6 +85,16 @@
     puppet_modules:  /etc/fuel/octane/puppet:/etc/puppet/modules
     timeout: 360
 
+- id: remove_hiera_section_repo_setup
+  type: shell
+  version: 2.1.0
+  groups: [compute]
+  requires: [rsync_octane]
+  required_for: []
+  parameters:
+    cmd: python /etc/fuel/octane/puppet/octane_tasks/files/delete_section.py /etc/astute.yaml repo_setup repos
+    timeout: 60
+
 - id: remove_hiera_override
   type: shell
   version: 2.1.0