Skip to content

Commit

Permalink
feat(api): add missing API filters
Browse files Browse the repository at this point in the history
Add important API filters that enable the following:
1. Ability to list extract metadata applicable only to a given data source and/or data source version.
2. Ability to list upload metadata from a given facility and/or on a given date.
  • Loading branch information
kennedykori committed Sep 19, 2022
1 parent 179f3b2 commit be67098
Showing 1 changed file with 19 additions and 5 deletions.
24 changes: 19 additions & 5 deletions apps/sql_data/apiviews.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,15 +28,15 @@ class DataSourceVersionViewSet(AuditBaseViewSet):

queryset = DataSourceVersion.objects.all()
serializer_class = DataSourceVersionSerializer
filterset_fields = ["id", "data_source", "data_source_version"]
filterset_fields = ("id", "data_source", "data_source_version")


class SQLDatabaseViewSet(AuditBaseViewSet):
"""SQL Database Sources API."""

queryset = SQLDatabaseSource.objects.all()
serializer_class = SQLDatabaseSerializer
filterset_fields = ["id", "name", "database_name", "database_vendor"]
filterset_fields = ("id", "name", "database_name", "database_vendor")


class SQLExtractMetadataViewSet(AuditBaseViewSet):
Expand All @@ -46,13 +46,14 @@ class SQLExtractMetadataViewSet(AuditBaseViewSet):
"applicable_source_versions", "data_source"
).all()
serializer_class = SQLExtractMetadataSerializer
filterset_fields = [
filterset_fields = (
"id",
"name",
"data_source__database_name",
"data_source",
"applicable_source_versions",
"version",
"preferred_uploads_name",
]
)


class SQLUploadChunkViewSet(
Expand All @@ -65,6 +66,10 @@ class SQLUploadChunkViewSet(

queryset = SQLUploadChunk.objects.all()
serializer_class = SQLUploadChunkSerializer
filterset_fields = (
"id",
"upload_metadata",
)


class SQLUploadMetadataViewSet(AuditBaseViewSet):
Expand All @@ -74,6 +79,15 @@ class SQLUploadMetadataViewSet(AuditBaseViewSet):
"upload_chunks"
).all()
serializer_class = SQLUploadMetadataSerializer
filterset_fields = (
"id",
"extract_metadata",
"org_unit_code",
"org_unit_name",
"start_time",
"finish_time",
"is_consumed",
)

@action(
detail=True,
Expand Down

0 comments on commit be67098

Please sign in to comment.