portal: names arg for maintenance

materialsproject · Sep 23, 2021 · 980a13e · 980a13e
1 parent 8cc4938
commit 980a13e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mpcontribs-portal/maintenance.py b/mpcontribs-portal/maintenance.py
@@ -9,10 +9,11 @@
 HEADERS = {"X-Authenticated-Groups": os.environ["ADMIN_GROUP"]}
 
 
-def generate_downloads():
+def generate_downloads(names=None):
+    q = {"name__in": names} if names else {}
     client = Client(host=os.environ["MPCONTRIBS_API_HOST"], headers=HEADERS)
     projects = client.projects.get_entries(
-        _fields=["name", "stats"]
+        _fields=["name", "stats"], **q
     ).result().get("data", [])
     skip = {"columns", "contributions"}
     print("PROJECTS", len(projects))