Skip to content

Commit 2d54409

Browse files
authored
Set pagination on appropriate streams (#12)
1 parent fb69a94 commit 2d54409

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

tap_freshdesk/client.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,7 @@ def get_url_params(
211211
Returns:
212212
A dictionary of URL query parameters.
213213
"""
214+
context = context or {}
214215
params = super().get_url_params(context, next_page_token)
215216
params['per_page'] = 100
216217
if next_page_token:

tap_freshdesk/streams.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,29 +13,29 @@
1313
SCHEMAS_DIR = Path(__file__).parent / Path("./schemas")
1414

1515

16-
class AgentsStream(FreshdeskStream):
16+
class AgentsStream(PagedFreshdeskStream):
1717
name = "agents"
1818

1919

20-
class CompaniesStream(FreshdeskStream):
20+
class CompaniesStream(PagedFreshdeskStream):
2121
name = "companies"
2222

2323

2424
class TicketFieldsStream(FreshdeskStream):
2525
name = "ticket_fields"
2626

2727

28-
class GroupsStream(FreshdeskStream):
28+
class GroupsStream(PagedFreshdeskStream):
2929
name = "groups"
3030

3131

32-
class ContactsStream(FreshdeskStream):
32+
class ContactsStream(PagedFreshdeskStream):
3333
name = "contacts"
3434

35-
class EmailConfigsStream(FreshdeskStream):
35+
class EmailConfigsStream(PagedFreshdeskStream):
3636
name = "email_configs"
3737

38-
class SlaPoliciesStream(FreshdeskStream):
38+
class SlaPoliciesStream(PagedFreshdeskStream):
3939
name = "sla_policies"
4040

4141
class TicketsAbridgedStream(PagedFreshdeskStream):
@@ -138,7 +138,7 @@ def request_records(self, context: dict | None) -> Iterable[dict]:
138138

139139
paginator.advance(resp)
140140

141-
class ConversationsStream(FreshdeskStream):
141+
class ConversationsStream(PagedFreshdeskStream):
142142

143143
name = "conversations"
144144

0 commit comments

Comments
 (0)