Skip to content
Snippets Groups Projects

Enable user context ingestion

Merged Deepa Kumari requested to merge dk/az-usr-ctx into master
Files
14
@@ -16,6 +16,7 @@
"""Context module."""
import dataclasses
import typing
@dataclasses.dataclass
@@ -24,7 +25,7 @@ class Context:
data_partition_id: str
app_key: str
user_id: typing.Optional[str]
@classmethod
def populate(cls, ctx: dict) -> 'Context':
"""
@@ -40,8 +41,12 @@ class Context:
except KeyError:
data_partition_id = ctx['dataPartitionId'] # to support some DAGs payload interface
user_id = ctx.get('userId')
print(f"user_id in Context Initialization is {user_id}")
ctx_obj = cls(app_key=ctx_payload['AppKey'],
data_partition_id=data_partition_id)
data_partition_id=data_partition_id,
user_id=user_id)
return ctx_obj
Loading