Skip to content

Commit 154fd45

Browse files
fix: Correct response model for create entitlement (#92)
Updated the response model for create entitlement route. Also added logic to conditionally set result params in parse_list_response.
1 parent 5a4c3f6 commit 154fd45

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

chargebee/models/entitlement/responses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,4 +30,4 @@ class ListResponse(Response):
3030
@dataclass
3131
class CreateResponse(Response):
3232
is_idempotency_replayed: bool
33-
entitlement: EntitlementResponse
33+
list: List[ListEntitlementResponse]

chargebee/responses.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,9 @@ def parse_list_response(self) -> T:
9494
field_name = name
9595
field_type = type
9696

97+
if field_name == "is_idempotency_replayed":
98+
result["is_idempotency_replayed"] = self.is_idempotency_replayed()
99+
97100
if hasattr(field_type, "__origin__") and field_type.__origin__ == list:
98101
list_data = []
99102
for response in self._response:
@@ -118,7 +121,10 @@ def parse_list_response(self) -> T:
118121
list_data.append(field_type.__args__[0](**data))
119122

120123
result[field_name] = list_data
121-
result["next_offset"] = self._next_offset
122124
result["headers"] = self._response_header
123125
result["http_status_code"] = self._response_status_code
126+
127+
if hasattr(self._response_type, 'next_offset'):
128+
result["next_offset"] = self._next_offset
129+
124130
return self._response_type(**result)

0 commit comments

Comments
 (0)