Skip to content

Commit 708204d

Browse files
committed
Pass env details from main to helpers
1 parent 7a4a6f1 commit 708204d

File tree

1 file changed

+24
-5
lines changed

1 file changed

+24
-5
lines changed

admin/create_secrets_files.py

Lines changed: 24 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@ def _create_and_get_vumark_details(
8686
def _generate_secrets_file_content(
8787
database_details: "DatabaseDict",
8888
vumark_details: "VuMarkDatabaseDict",
89+
inactive_database_details: dict[str, str],
8990
) -> str:
9091
"""Generate the content of a secrets file."""
9192
return textwrap.dedent(
@@ -96,11 +97,11 @@ def _generate_secrets_file_content(
9697
VUFORIA_CLIENT_ACCESS_KEY={database_details["client_access_key"]}
9798
VUFORIA_CLIENT_SECRET_KEY={database_details["client_secret_key"]}
9899
99-
INACTIVE_VUFORIA_TARGET_MANAGER_DATABASE_NAME={os.environ["INACTIVE_VUFORIA_TARGET_MANAGER_DATABASE_NAME"]}
100-
INACTIVE_VUFORIA_SERVER_ACCESS_KEY={os.environ["INACTIVE_VUFORIA_SERVER_ACCESS_KEY"]}
101-
INACTIVE_VUFORIA_SERVER_SECRET_KEY={os.environ["INACTIVE_VUFORIA_SERVER_SECRET_KEY"]}
102-
INACTIVE_VUFORIA_CLIENT_ACCESS_KEY={os.environ["INACTIVE_VUFORIA_CLIENT_ACCESS_KEY"]}
103-
INACTIVE_VUFORIA_CLIENT_SECRET_KEY={os.environ["INACTIVE_VUFORIA_CLIENT_SECRET_KEY"]}
100+
INACTIVE_VUFORIA_TARGET_MANAGER_DATABASE_NAME={inactive_database_details["INACTIVE_VUFORIA_TARGET_MANAGER_DATABASE_NAME"]}
101+
INACTIVE_VUFORIA_SERVER_ACCESS_KEY={inactive_database_details["INACTIVE_VUFORIA_SERVER_ACCESS_KEY"]}
102+
INACTIVE_VUFORIA_SERVER_SECRET_KEY={inactive_database_details["INACTIVE_VUFORIA_SERVER_SECRET_KEY"]}
103+
INACTIVE_VUFORIA_CLIENT_ACCESS_KEY={inactive_database_details["INACTIVE_VUFORIA_CLIENT_ACCESS_KEY"]}
104+
INACTIVE_VUFORIA_CLIENT_SECRET_KEY={inactive_database_details["INACTIVE_VUFORIA_CLIENT_SECRET_KEY"]}
104105
105106
VUMARK_VUFORIA_TARGET_MANAGER_DATABASE_NAME={vumark_details["database_name"]}
106107
VUMARK_VUFORIA_SERVER_ACCESS_KEY={vumark_details["server_access_key"]}
@@ -121,6 +122,23 @@ def main() -> None:
121122
msg = f"Existing secrets file does not exist: {existing_secrets_file}"
122123
raise FileNotFoundError(msg)
123124
load_dotenv(dotenv_path=existing_secrets_file)
125+
inactive_database_details = {
126+
"INACTIVE_VUFORIA_TARGET_MANAGER_DATABASE_NAME": os.environ[
127+
"INACTIVE_VUFORIA_TARGET_MANAGER_DATABASE_NAME"
128+
],
129+
"INACTIVE_VUFORIA_SERVER_ACCESS_KEY": os.environ[
130+
"INACTIVE_VUFORIA_SERVER_ACCESS_KEY"
131+
],
132+
"INACTIVE_VUFORIA_SERVER_SECRET_KEY": os.environ[
133+
"INACTIVE_VUFORIA_SERVER_SECRET_KEY"
134+
],
135+
"INACTIVE_VUFORIA_CLIENT_ACCESS_KEY": os.environ[
136+
"INACTIVE_VUFORIA_CLIENT_ACCESS_KEY"
137+
],
138+
"INACTIVE_VUFORIA_CLIENT_SECRET_KEY": os.environ[
139+
"INACTIVE_VUFORIA_CLIENT_SECRET_KEY"
140+
],
141+
}
124142
new_secrets_dir.mkdir(exist_ok=True)
125143

126144
num_databases = 100
@@ -178,6 +196,7 @@ def main() -> None:
178196
file_contents = _generate_secrets_file_content(
179197
database_details=database_details,
180198
vumark_details=vumark_details,
199+
inactive_database_details=inactive_database_details,
181200
)
182201
file.write_text(data=file_contents)
183202
sys.stdout.write(f"Created database {file.name}\n")

0 commit comments

Comments
 (0)