Skip to content

Commit 42da9d3

Browse files
committed
refactor(appsync): use model param for consistency
1 parent bc250f3 commit 42da9d3

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

aws_lambda_powertools/event_handler/appsync.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def common_field() -> str:
3737
# Would match all fieldNames matching 'commonField'
3838
return str(uuid.uuid4())
3939
"""
40-
40+
4141
current_event: AppSyncResolverEvent
4242
lambda_context: LambdaContext
4343

@@ -62,7 +62,9 @@ def register_resolver(func):
6262

6363
return register_resolver
6464

65-
def resolve(self, event: dict, context: LambdaContext, current_event_data_class: Type[AppSyncResolverEvent] = AppSyncResolverEvent) -> Any:
65+
def resolve(
66+
self, event: dict, context: LambdaContext, model: Type[AppSyncResolverEvent] = AppSyncResolverEvent
67+
) -> Any:
6668
"""Resolve field_name
6769
6870
Parameters
@@ -71,8 +73,8 @@ def resolve(self, event: dict, context: LambdaContext, current_event_data_class:
7173
Lambda event
7274
context : LambdaContext
7375
Lambda context
74-
current_event_data_class:
75-
Decode instance of event to class or subclass of AppSyncResolverEvent
76+
model:
77+
Your data model to decode AppSync event, by default AppSyncResolverEvent
7678
7779
Returns
7880
-------
@@ -84,7 +86,7 @@ def resolve(self, event: dict, context: LambdaContext, current_event_data_class:
8486
ValueError
8587
If we could not find a field resolver
8688
"""
87-
self.current_event = current_event_data_class(event)
89+
self.current_event = model(event)
8890
self.lambda_context = context
8991
resolver = self._get_resolver(self.current_event.type_name, self.current_event.field_name)
9092
return resolver(**self.current_event.arguments)

0 commit comments

Comments
 (0)