api: fix admin group for notebook build

materialsproject · Sep 1, 2021 · ac6c6fb · ac6c6fb
1 parent bbe57ef
commit ac6c6fb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mpcontribs-api/mpcontribs/api/notebooks/views.py b/mpcontribs-api/mpcontribs/api/notebooks/views.py
@@ -30,6 +30,7 @@
 notebooks = Blueprint("notebooks", __name__, template_folder=templates)
 
 MPCONTRIBS_API_HOST = os.environ["MPCONTRIBS_API_HOST"]
+ADMIN_GROUP = os.environ.get("ADMIN_GROUP", "admin")
 
 rq = RQ()
 rq.default_queue = f"notebooks_{MPCONTRIBS_API_HOST}"
@@ -211,7 +212,7 @@ def make(projects=None, cids=None, force=False):
             nbf.new_code_cell("\n".join([
                 "if 'client' not in locals():",
                 "\tclient = Client(",
-                '\t\theaders={"X-Authenticated-Groups": "admin"},',
+                f'\t\theaders={{"X-Authenticated-Groups": "{ADMIN_GROUP}"}},',
                 f'\t\thost="{MPCONTRIBS_API_HOST}"',
                 "\t)",
                 "print(client.get_totals())",