Merge upgrade and preupgrade script, add sync script for rhc devel wo…

…rker

misc/sync_scripts.py

@@ -0,0 +1,73 @@
+import os
+import ruamel.yaml
+
+# Scripts located in this project
+SCRIPT_PATH = "scripts/leapp_script.py"
+
+# Yaml playbooks in rhc-worker-script
+PRE_UPGRADE_YAML_PATH = os.path.join(
+    "..", "rhc-worker-script/development/nginx/data/leapp_preupgrade.yml"
+)
+UPGRADE_YAML_PATH = os.path.join(
+    "..", "rhc-worker-script/development/nginx/data/leapp_upgrade.yml"
+)
+
+DEFAULT_YAML_ENVELOPE = """
+- name: LEAPP
+  vars:
+    insights_signature: |
+      ascii_armored gpg signature
+    insights_signature_exclude: /vars/insights_signature,/vars/content_vars
+    interpreter: /usr/bin/python
+    content: |
+      placeholder
+    content_vars:
+      # variables that will be handed to the script as environment vars
+      # will be prefixed with RHC_WORKER_*
+      LEAPP_SCRIPT_TYPE: type
+"""
+
+
+def _get_updated_yaml_content(yaml_path, script_path):
+    if not os.path.exists(yaml_path):
+        yaml = ruamel.yaml.YAML()
+        config = yaml.load(DEFAULT_YAML_ENVELOPE)
+        mapping = 2
+        offset = 0
+    else:
+        config, mapping, offset = ruamel.yaml.util.load_yaml_guess_indent(
+            open(yaml_path)
+        )
+        print(mapping, offset)
+
+    with open(script_path) as script:
+        content = script.read()
+
+    script_type = "PREUPGRADE" if "preupgrade" in yaml_path else "UPGRADE"
+    config[0]["name"] = "LEAPP %s" % script_type.title()
+    config[0]["vars"]["content"] = content
+    config[0]["vars"]["content_vars"]["LEAPP_SCRIPT_TYPE"] = script_type
+    return config, mapping, offset
+
+
+def _write_content(config, path, mapping=None, offset=None):
+    yaml = ruamel.yaml.YAML()
+    if mapping and offset:
+        yaml.indent(mapping=mapping, sequence=mapping, offset=offset)
+    with open(path, "w") as handler:
+        yaml.dump(config, handler)
+
+
+def main():
+    config, mapping, offset = _get_updated_yaml_content(
+        PRE_UPGRADE_YAML_PATH, SCRIPT_PATH
+    )
+    print("Writing new content to %s" % PRE_UPGRADE_YAML_PATH)
+    _write_content(config, PRE_UPGRADE_YAML_PATH, mapping, offset)
+    config, mapping, offset = _get_updated_yaml_content(UPGRADE_YAML_PATH, SCRIPT_PATH)
+    print("Writing new content to %s" % UPGRADE_YAML_PATH)
+    _write_content(config, UPGRADE_YAML_PATH, mapping, offset)
+
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -2,3 +2,4 @@ jsonschema==3.2.0
 mock==3.0.5
 pytest==4.6.11
 pytest-cov==2.12.1
+ruamel.yaml==0.16.13