From 9063588e899e3693a928902f58e2d69b53478334 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jakub=20K=C3=A1kona?= <kaklik@mlab.cz>
Date: Sat, 16 Sep 2023 18:38:44 +0200
Subject: [PATCH] Adding script which updates module status according to JSON
 file.

---
 src/mlabtools/update_github_metadata.py | 70 +++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 src/mlabtools/update_github_metadata.py

diff --git a/src/mlabtools/update_github_metadata.py b/src/mlabtools/update_github_metadata.py
new file mode 100644
index 0000000..5981ff6
--- /dev/null
+++ b/src/mlabtools/update_github_metadata.py
@@ -0,0 +1,70 @@
+
+import requests
+import base64
+import yaml
+import json
+import sys
+
+# Your GitHub authentication token is passed as a command-line argument
+if len(sys.argv) < 2:
+    print("Usage: python script.py YOUR_GITHUB_TOKEN")
+    sys.exit(1)
+
+GITHUB_TOKEN = sys.argv[1]
+
+# Common GitHub API headers
+HEADERS = {
+    "Authorization": f"token {GITHUB_TOKEN}",
+    "Accept": "application/vnd.github.v3+json"
+}
+
+# Load modules.json data (this should be read from an actual JSON file in a real implementation)
+modules_json_data = json.load(open('modules.json', 'r'))
+#modules_json_data = [{"name": "USB232R02", "status": 2}]  # Simplified example data
+
+def get_yaml_from_repo(module_name):
+    repo_url = f"https://api.github.com/repos/mlab-modules/{module_name}/contents/doc/metadata.yaml"
+    response = requests.get(repo_url, headers=HEADERS)
+    if response.status_code == 200:
+        file_content_base64 = response.json()['content']
+        file_sha = response.json()['sha']
+        return base64.b64decode(file_content_base64).decode('utf-8'), file_sha
+    else:
+        print(f"Failed to get metadata.yaml for {module_name}")
+        return None, None
+
+def update_yaml_in_repo(module_name, updated_yaml_content, sha):
+    repo_url = f"https://api.github.com/repos/mlab-modules/{module_name}/contents/doc/metadata.yaml"
+    data = {
+        "message": "Update status",
+        "content": base64.b64encode(updated_yaml_content.encode('utf-8')).decode('utf-8'),
+        "sha": sha
+    }
+    response = requests.put(repo_url, headers=HEADERS, json=data)
+    if response.status_code == 200:
+        print(f"Successfully updated metadata.yaml for {module_name}")
+    else:
+        print(f"Failed to update metadata.yaml for {module_name}")
+
+for module in modules_json_data:
+    module_name = module['name']
+    new_status = module['status']
+    
+    yaml_content, yaml_sha = get_yaml_from_repo(module_name)
+    if yaml_content:
+        yaml_data = yaml.safe_load(yaml_content)
+        if not 'status' in yaml_data:
+            yaml_data['status'] = new_status
+            updated_yaml_content = yaml.safe_dump(yaml_data)
+            update_yaml_in_repo(module_name, updated_yaml_content, yaml_sha)
+            print(f"Creating status key in metadata.yaml for {module_name}")
+
+        elif yaml_data['status'] != new_status:
+            yaml_data['status'] = new_status
+            updated_yaml_content = yaml.safe_dump(yaml_data)
+            update_yaml_in_repo(module_name, updated_yaml_content, yaml_sha)
+            print(f"Updating status key in metadata.yaml for {module_name}")
+        else: 
+            print(f"Status key in metadata.yaml for {module_name} is already the same")
+
+