From b7a8eba4de0e814ea0a2b37d12facd76476d8986 Mon Sep 17 00:00:00 2001 From: Alek Westover Date: Tue, 18 Jul 2023 15:14:29 -0400 Subject: [PATCH] revert to python script --- scripts/combine_control_files.py | 17 +++++++++ scripts/combine_control_files.sh | 65 -------------------------------- 2 files changed, 17 insertions(+), 65 deletions(-) create mode 100644 scripts/combine_control_files.py delete mode 100644 scripts/combine_control_files.sh diff --git a/scripts/combine_control_files.py b/scripts/combine_control_files.py new file mode 100644 index 0000000000..cf91e3e284 --- /dev/null +++ b/scripts/combine_control_files.py @@ -0,0 +1,17 @@ +import json +import os + +# enable custom extensions for specific tenants +enabled_extensions = {"123454321": ["anon"], "public": ["embedding"]} + +control_data = {} +os.chdir("control_files") +for control_file in os.listdir("."): + ext_name = control_file.replace(".control", "") + with open(control_file, "r") as f: + control_data[ext_name] = f.read() + +all_data = {"enabled_extensions": enabled_extensions, "control_data": control_data} + +with open("ext_index.json", "w") as f: + json.dump(all_data, f) diff --git a/scripts/combine_control_files.sh b/scripts/combine_control_files.sh deleted file mode 100644 index fab3bd1b57..0000000000 --- a/scripts/combine_control_files.sh +++ /dev/null @@ -1,65 +0,0 @@ -#!/bin/bash - -# Enable custom extensions for specific tenants -declare -A enabled_extensions -enabled_extensions["123454321"]=("anon") -enabled_extensions["public"]=("embedding") - -# Initialize control data -declare -A control_data - -# Read control files -for control_file in control_files/*.control; do - ext_name=$(basename "$control_file" .control) - control_data["$ext_name"]=$(cat "$control_file") -done - -# Construct all data -all_data='{"enabled_extensions":'"$(declare -p enabled_extensions)"', "control_data":'"$(declare -p control_data)"'}' - -# Write all data to ext_index.json -echo "$all_data" > ext_index.json - -# const fs = require('fs'); - -# // enable custom extensions for specific tenants -# const enabledExtensions = { -# "123454321": ["anon"], -# "public": ["embedding"] -# }; - -# const controlData = {}; -# const controlFiles = fs.readdirSync("control_files"); - -# controlFiles.forEach(controlFile => { -# const extName = controlFile.replace(".control", ""); -# const controlFilePath = `control_files/${controlFile}`; -# const fileData = fs.readFileSync(controlFilePath, "utf8"); -# controlData[extName] = fileData; -# }); - -# const allData = { -# enabledExtensions: enabledExtensions, -# controlData: controlData -# }; - -# const jsonData = JSON.stringify(allData, null, 2); -# fs.writeFileSync("ext_index.json", jsonData); - - -# // import json -# // import os - -# // # enable custom extensions for specific tenants -# // enabled_extensions = {"123454321": ["anon"], "public": ["embedding"]} - -# // control_data = {} -# // for control_file in os.listdir("control_files"): -# // ext_name = control_file.replace(".control", "") -# // with open(control_file, "r") as f: -# // control_data[ext_name] = f.read() - -# // all_data = {"enabled_extensions": enabled_extensions, "control_data": control_data} - -# // with open("ext_index.json", "w") as f: -# // json.dump(all_data, f)