Add previously removed columns to agencies data source subquery

Police-Data-Accessibility-Project · Nov 18, 2024 · 77eefd2 · 77eefd2
1 parent 3356f27
commit 77eefd2
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/database_client/subquery_logic.py b/database_client/subquery_logic.py
@@ -54,7 +54,18 @@ def get_subquery_params(
         get_subquery_params,
         relation=Relations.AGENCIES_EXPANDED,
         linking_column="agencies",
-        columns=["id", "name"],
+        columns=[
+            "id",
+            "name",
+            "submitted_name",
+            "state_name",
+            "locality_name",
+            "state_iso",
+            "county_name",
+            "agency_type",
+            "jurisdiction_type",
+            "homepage_url"
+        ],
     )
 
     data_requests = partialmethod(

diff --git a/middleware/schema_and_dto_logic/primary_resource_schemas/data_sources_advanced_schemas.py b/middleware/schema_and_dto_logic/primary_resource_schemas/data_sources_advanced_schemas.py
@@ -28,7 +28,18 @@ class DataSourceGetSchema(DataSourceExpandedSchema):
     agencies = fields.List(
         fields.Nested(
             AgenciesExpandedSchema(
-                only=["id", "name"],
+                only=[
+                    "id",
+                    "name",
+                    "submitted_name",
+                    "state_name",
+                    "locality_name",
+                    "state_iso",
+                    "county_name",
+                    "agency_type",
+                    "jurisdiction_type",
+                    "homepage_url"
+                ],
             ),
             metadata=get_json_metadata("The agencies associated with the data source."),
         ),