add get project

chuongmep · Sep 10, 2024 · 7758d5e · 7758d5e
1 parent 13ec84b
commit 7758d5e
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 20 deletions.
diff --git a/APSToolkitPython/src/aps_toolkit/AECDataModel.py b/APSToolkitPython/src/aps_toolkit/AECDataModel.py
@@ -48,3 +48,30 @@ def get_hubs(self) -> pd.DataFrame:
         result = self.execute_query(data)
         hubs = result['data']['hubs']['results']
         return pd.json_normalize(hubs)
+
+    def get_projects(self, hub_id: str) -> pd.DataFrame:
+        data = {
+            "query": """
+                query GetProjects($hubId: ID!) {
+                    projects(hubId: $hubId) {
+                        results {
+                            id
+                            name
+                            hub {
+                                id
+                                name
+                            }
+                            alternativeIdentifiers{
+                             dataManagementAPIProjectId
+                            }
+                        }
+                    }
+                }
+            """,
+            "variables": {
+                "hubId": hub_id
+            }
+        }
+        result = self.execute_query(data)
+        projects = result['data']['projects']['results']
+        return pd.json_normalize(projects)
diff --git a/APSToolkitPython/src/test/test_aec_data_model.py b/APSToolkitPython/src/test/test_aec_data_model.py
@@ -27,26 +27,7 @@ def test_get_hubs(self):
         self.assertIsNotNone(df_result)
 
     def test_get_projects(self):
-        data = {
-            "query": """
-                query GetProjects($hubId: ID!) {
-                    projects(hubId: $hubId) {
-                        results {
-                                id
-                                name
-                                hub {
-                                    id
-                                    name
-                                            }
-                                        }
-                                }
-                }
-                    """,
-            "variables": {
-                "hubId": f"{self.hub_id}"
-            }
-        }
-        result = self.aec_data_model.execute_query(data)
+        result = self.aec_data_model.get_projects(self.hub_id)
         self.assertIsNotNone(result)
 
     def test_version_group_by_project(self):