[pkg] some extra branding refactor

1 files changed, 38 insertions, 0 deletions

diff --git a/branding/scripts/generate-snap b/branding/scripts/generate-snap
new file mode 100755
index 0000000..3271d9e
--- /dev/null
+++ b/branding/scripts/generate-snap
@@ -0,0 +1,38 @@
+#!/usr/bin/env python3
+
+import json
+import os
+import sys
+
+import configparser
+
+from provider import getDefaultProvider
+from provider import getProviderData
+
+
+VERSION = os.environ.get('VERSION', 'unknown')
+SCRIPT = 'generate-snap.py'
+
+
+def writeOutput(data, outfile):
+
+    with open(outfile, 'w') as outf:
+        outf.write(json.dumps(data))
+
+
+if __name__ == "__main__":
+    VENDOR_PATH = os.environ.get('VENDOR_PATH')
+    configFile = os.path.join(VENDOR_PATH, 'vendor.conf')
+    config = configparser.ConfigParser()
+    config.read(configFile)
+    provider = getDefaultProvider(config)
+    data = getProviderData(provider, config)
+
+    if len(sys.argv) != 2:
+        print('Usage: {} <output_file>'.format(SCRIPT))
+        sys.exit(1)
+
+    outputf = sys.argv[1]
+
+    data['version'] = VERSION
+    writeOutput(data, outputf)