yield workunit with status (#41)

Affirm · Jul 7, 2023 · 68301fb · 68301fb
1 parent 8530a3c
commit 68301fb
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/aws/dynamodb.py b/metadata-ingestion/src/datahub/ingestion/source/aws/dynamodb.py
@@ -17,6 +17,7 @@
 from datahub.ingestion.source.aws.aws_common import AwsSourceConfig
 from datahub.metadata.schema_classes import (
     ChangeTypeClass,
+    StatusClass
 )
 from datahub.metadata.com.linkedin.pegasus2avro.schema import (
     ArrayType,
@@ -112,6 +113,12 @@ def get_workunits(self) -> Iterable[MetadataWorkUnit]:
                 continue
             yield wu
 
+            # add status
+            dataset_urn = make_dataset_urn(DEFAULT_PLATFORM, table_name, self.config.env)
+            yield MetadataChangeProposalWrapper(
+                entityUrn=f"{dataset_urn}",
+                aspect=StatusClass(removed=False),
+            ).as_workunit()
 
     def make_workunit(self, table_name: str):
         table = describe_table(self.config.dynamodb_client, table_name)