Refactor generate-sku-list.py

pull/4/head
WeirdTreeThing 2023-08-23 21:23:17 -04:00
parent 5935e5f43c
commit b84259e08f
No known key found for this signature in database
GPG Key ID: 1F56A4B52998B851
1 changed files with 40 additions and 27 deletions

View File

@ -1,37 +1,50 @@
#!/usr/bin/env python3
import os
import sys
import json
variants = []
sku_ids = {}
def format_skus(sku_ids, suffix):
sku_ids_formatted = ""
for sku in sku_ids[suffix]:
if sku_ids_formatted == "":
sku_ids_formatted = sku
else:
sku_ids_formatted = f"{sku_ids_formatted}|{sku}"
return sku_ids_formatted
def load_project_config():
with open("project-config.json", "r") as project:
p_json = json.load(project)
configs = p_json['chromeos']['configs']
project_json = json.load(project)
return project_json['chromeos']['configs']
def select_variant(configs):
variants = []
print("Variants:")
for config in configs:
variant = config['name']
if not variant in variants:
variants.append(variant)
print(variant)
variant = input("Choose a variant: ")
if config['name'] in variants:
continue
variants.append(config['name'])
print(config['name'])
return input("Choose a variant: ")
def parse_sku_ids(configs, variant):
sku_ids = {}
for config in configs:
if config['audio']['main']['cras-config-dir'] == variant:
if not config['audio']['main']['ucm-suffix'] in sku_ids:
if variant not in config['audio']['main']['cras-config-dir']:
continue
if config['audio']['main']['ucm-suffix'] not in sku_ids:
sku_ids[config['audio']['main']['ucm-suffix']] = []
sku_ids[config['audio']['main']['ucm-suffix']].append(config['identity']['sku-id'])
return sku_ids
def print_sku_id_list(sku_ids):
for suffix in sku_ids:
suffixes_formatted = ""
for sku in sku_ids[suffix]:
if suffixes_formatted == "":
suffixes_formatted = str(sku)
else:
suffixes_formatted = f"{suffixes_formatted}|{str(sku)}"
print(f"{suffix}: {suffixes_formatted}")
sku_ids_formatted = format_skus(sku_ids, suffix)
print(f"{suffix}: {sku_ids_formatted}")
def main():
configs = load_project_config()
variant = select_variant(configs)
sku_ids = parse_sku_ids(configs, variant)
print_sku_id_list(sku_ids)
main()